Update search_model.py

Signed-off-by: Kaveen Kumarasinghe <k5kumara@uwaterloo.ca>
2 years ago · b7b7a5f5c3
parent fccda65f43
commit b7b7a5f5c3
1 changed files with 1 additions and 1 deletions
--- a/models/search_model.py
+++ b/models/search_model.py
@ -201,7 +201,7 @@ class Search:
        llm_predictor = LLMPredictor(llm=OpenAI(model_name="text-davinci-003"))
        try:
            llm_predictor_presearch = OpenAI(
-                max_tokens=40, temperature=0, model_name="text-davinci-003"
+                max_tokens=50, temperature=0.25, presence_penalty=0.65, model_name="text-davinci-003"
            )
            # Refine a query to send to google custom search API