moved default number of web results to prompt engineering file

hanlde_form_submit.py

@@ -34,12 +34,12 @@ def generate_text(
 ) -> str:
     """
     Generates text using the given pipeline.
-    :param pipeline: The pipeline to use.
-    :param prompt: The prompt to use.
-    :param output_length: The size of the groups to use.
-    :return: The generated text.
+    :param pipeline: The pipeline to use. GroupedSamplingPipeLine.
+    :param prompt: The prompt to use. str.
+    :param output_length: The size of the text to generate in tokens. int > 0.
+    :return: The generated text. str.
     """
-    better_prompt = rewrite_prompt(prompt, 5)
+    better_prompt = rewrite_prompt(prompt)
     return pipeline(
         prompt_s=better_prompt,
         max_new_tokens=output_length,

prompt_engeneering.py

@@ -49,10 +49,12 @@ def format_search_result(search_result: Generator[SearchResult, None, None]) ->
 
 def rewrite_prompt(
         prompt: str,
-        num_results: int,
 ) -> str:
     """Rewrites the prompt by adding web search results to it."""
-    raw_results = get_web_search_results(prompt, num_results)
+    raw_results = get_web_search_results(
+        prompt=prompt,
+        num_results=5,
+    )
     formatted_results = "Web search results: " + format_search_result(raw_results)
     formatted_date = "Current date: " + datetime.now().strftime("%d/%m/%Y")
     formatted_prompt = f"Query: {prompt}"