Merge remote-tracking branch 'origin/dev' into zustand/io/migration

tests/test_custom_component.py

@@ -21,7 +21,7 @@ from langflow.interface.custom.custom_component import CustomComponent
 from langchain.llms.base import BaseLLM
 from langchain.chains import LLMChain
 from langchain.prompts import PromptTemplate
-from langchain.schema import Document
+from langchain_core.documents import Document
 
 import requests
 
@@ -228,6 +228,7 @@ def test_custom_component_get_function_entrypoint_return_type():
     Test the get_function_entrypoint_return_type
     property of the CustomComponent class.
     """
+
     custom_component = CustomComponent(code=code_default, function_entrypoint_name="build")
     return_type = custom_component.get_function_entrypoint_return_type
     assert return_type == [Document]

tests/test_endpoints.py

@@ -72,11 +72,12 @@ PROMPT_REQUEST = {
                             "text-ada-001",
                         ],
                         "ChatOpenAI": [
+                            "gpt-4-turbo-preview",
+                            "gpt-4-0125-preview",
                             "gpt-4-1106-preview",
-                            "gpt-4",
-                            "gpt-4-32k",
-                            "gpt-3.5-turbo",
-                            "gpt-3.5-turbo-16k",
+                            "gpt-4-vision-preview",
+                            "gpt-3.5-turbo-0125",
+                            "gpt-3.5-turbo-1106",
                         ],
                         "Anthropic": [
                             "claude-v1",

tests/test_template.py

@@ -235,7 +235,7 @@ def test_format_dict():
             "password": False,
             "multiline": False,
             "options": CHAT_OPENAI_MODELS,
-            "value": "gpt-4-1106-preview",
+            "value": "gpt-4-turbo-preview",
         },
     }
     assert format_dict(input_dict, "OpenAI") == expected_output_openai