Commit ·
5ba3489
1
Parent(s): abe7873
Update app.py
Browse files
app.py
CHANGED
|
@@ -91,9 +91,9 @@ def predict(message, model_type):
|
|
| 91 |
# conversational retrieval agent component construction - memory, prompt template, agent, agent executor
|
| 92 |
# specifying LLM to use
|
| 93 |
if (model_type==1):
|
| 94 |
-
llm = ChatOpenAI(temperature = 0.1, model_name="gpt-3.5-turbo-
|
| 95 |
else:
|
| 96 |
-
llm = ChatOpenAI(temperature = 0.1, model_name="gpt-4-
|
| 97 |
|
| 98 |
# This is needed for both the memory and the prompt
|
| 99 |
memory_key = "history"
|
|
|
|
| 91 |
# conversational retrieval agent component construction - memory, prompt template, agent, agent executor
|
| 92 |
# specifying LLM to use
|
| 93 |
if (model_type==1):
|
| 94 |
+
llm = ChatOpenAI(temperature = 0.1, model_name="gpt-3.5-turbo-1106", streaming=True, callbacks=[QueueCallback(q)])
|
| 95 |
else:
|
| 96 |
+
llm = ChatOpenAI(temperature = 0.1, model_name="gpt-4-1106-preview", streaming=True, callbacks=[QueueCallback(q)])
|
| 97 |
|
| 98 |
# This is needed for both the memory and the prompt
|
| 99 |
memory_key = "history"
|