jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

🔧 fix(OpenAIConversationalAgent.py): update OpenAI conversational agent models to use the latest versions for better performance and accuracy

Browse source 
🔧 fix(constants.py): update OpenAI chat models to use the latest versions for better performance and accuracy
🔧 fix(test_llms_template.py): update test case to use the latest OpenAI chat model for accurate testing
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-11-08 13:48:17 -03:00
commit 8dfcd06d46
3 changed files with 13 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

9
src/backend/langflow/components/agents/OpenAIConversationalAgent.py
View file

@ -20,10 +20,11 @@ class ConversationalAgent(CustomComponent):
def build_config(self): def build_config(self):
openai_function_models = [ openai_function_models = [
"gpt-3.5-turbo-0613", "gpt-4-1106-preview",
"gpt-3.5-turbo-16k-0613", "gpt-3.5-turbo",
"gpt-4-0613", "gpt-3.5-turbo-16k",
"gpt-4-32k-0613", "gpt-4",
"gpt-4-32k",
] ]
return { return {
"tools": {"is_list": True, "display_name": "Tools"}, "tools": {"is_list": True, "display_name": "Tools"},

10
src/backend/langflow/utils/constants.py
View file

@ -6,16 +6,14 @@ OPENAI_MODELS = [
"text-ada-001", "text-ada-001",
] ]
CHAT_OPENAI_MODELS = [ CHAT_OPENAI_MODELS = [
"gpt-3.5-turbo-0613", "gpt-4-1106-preview",
"gpt-3.5-turbo",
"gpt-3.5-turbo-16k-0613",
"gpt-3.5-turbo-16k",
"gpt-4-0613",
"gpt-4-32k-0613",
"gpt-4", "gpt-4",
"gpt-4-32k", "gpt-4-32k",
"gpt-3.5-turbo",
"gpt-3.5-turbo-16k",
] ]
ANTHROPIC_MODELS = [ ANTHROPIC_MODELS = [
# largest model, ideal for a wide range of more complex tasks. # largest model, ideal for a wide range of more complex tasks.
"claude-v1", "claude-v1",

11
tests/test_llms_template.py
View file

@ -309,17 +309,14 @@ def test_chat_open_ai(client: TestClient, logged_in_headers):
"placeholder": "", "placeholder": "",
"show": True, "show": True,
"multiline": False, "multiline": False,
"value": "gpt-3.5-turbo-0613", "value": "gpt-4-1106-preview",
"password": False, "password": False,
"options": [ "options": [
"gpt-3.5-turbo-0613", "gpt-4-1106-preview",
"gpt-3.5-turbo",
"gpt-3.5-turbo-16k-0613",
"gpt-3.5-turbo-16k",
"gpt-4-0613",
"gpt-4-32k-0613",
"gpt-4", "gpt-4",
"gpt-4-32k", "gpt-4-32k",
"gpt-3.5-turbo",
"gpt-3.5-turbo-16k",
], ],
"name": "model_name", "name": "model_name",
"type": "str", "type": "str",