Merge branch 'dev' into build_config_with_templatefield

tests/test_agents_template.py

@@ -28,6 +28,7 @@ def test_zero_shot_agent(client: TestClient, logged_in_headers):
         "list": True,
         "advanced": False,
         "info": "",
+        "fileTypes": [],
     }
 
     # Additional assertions for other template variables
@@ -43,6 +44,7 @@ def test_zero_shot_agent(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": False,
         "info": "",
+        "fileTypes": [],
     }
     assert template["llm"] == {
         "required": True,
@@ -56,6 +58,7 @@ def test_zero_shot_agent(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": False,
         "info": "",
+        "fileTypes": [],
     }
     assert template["output_parser"] == {
         "required": False,
@@ -69,6 +72,7 @@ def test_zero_shot_agent(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": False,
         "info": "",
+        "fileTypes": [],
     }
     assert template["input_variables"] == {
         "required": False,
@@ -82,6 +86,7 @@ def test_zero_shot_agent(client: TestClient, logged_in_headers):
         "list": True,
         "advanced": False,
         "info": "",
+        "fileTypes": [],
     }
     assert template["prefix"] == {
         "required": False,
@@ -96,6 +101,7 @@ def test_zero_shot_agent(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": False,
         "info": "",
+        "fileTypes": [],
     }
     assert template["suffix"] == {
         "required": False,
@@ -110,6 +116,7 @@ def test_zero_shot_agent(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": False,
         "info": "",
+        "fileTypes": [],
     }
 
 
@@ -135,6 +142,9 @@ def test_json_agent(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": False,
         "info": "",
+        "file_path": "",
+        "fileTypes": [],
+        "value": "",
     }
     assert template["llm"] == {
         "required": True,
@@ -149,6 +159,9 @@ def test_json_agent(client: TestClient, logged_in_headers):
         "advanced": False,
         "display_name": "LLM",
         "info": "",
+        "file_path": "",
+        "fileTypes": [],
+        "value": "",
     }
 
 
@@ -174,7 +187,7 @@ def test_csv_agent(client: TestClient, logged_in_headers):
         "name": "path",
         "type": "file",
         "list": False,
-        "file_path": None,
+        "file_path": "",
         "advanced": False,
         "info": "",
     }
@@ -191,4 +204,7 @@ def test_csv_agent(client: TestClient, logged_in_headers):
         "advanced": False,
         "display_name": "LLM",
         "info": "",
+        "file_path": "",
+        "fileTypes": [],
+        "value": "",
     }

tests/test_chains_template.py

@@ -35,6 +35,7 @@ def test_llm_checker_chain(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": False,
         "info": "",
+        "fileTypes": [],
     }
     assert template["_type"] == "LLMCheckerChain"
 
@@ -69,6 +70,7 @@ def test_llm_math_chain(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": False,
         "info": "",
+        "fileTypes": [],
     }
     assert template["verbose"] == {
         "required": False,
@@ -83,6 +85,7 @@ def test_llm_math_chain(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": True,
         "info": "",
+        "fileTypes": [],
     }
     assert template["llm"] == {
         "required": True,
@@ -96,6 +99,7 @@ def test_llm_math_chain(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": False,
         "info": "",
+        "fileTypes": [],
     }
     assert template["input_key"] == {
         "required": True,
@@ -110,6 +114,7 @@ def test_llm_math_chain(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": True,
         "info": "",
+        "fileTypes": [],
     }
     assert template["output_key"] == {
         "required": True,
@@ -124,6 +129,7 @@ def test_llm_math_chain(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": True,
         "info": "",
+        "fileTypes": [],
     }
     assert template["_type"] == "LLMMathChain"
 
@@ -163,6 +169,9 @@ def test_series_character_chain(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": False,
         "info": "",
+        "fileTypes": [],
+        "file_path": "",
+        "value": "",
     }
     assert template["character"] == {
         "required": True,
@@ -176,6 +185,9 @@ def test_series_character_chain(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": False,
         "info": "",
+        "fileTypes": [],
+        "file_path": "",
+        "value": "",
     }
     assert template["series"] == {
         "required": True,
@@ -189,6 +201,9 @@ def test_series_character_chain(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": False,
         "info": "",
+        "fileTypes": [],
+        "file_path": "",
+        "value": "",
     }
     assert template["_type"] == "SeriesCharacterChain"
 
@@ -232,6 +247,9 @@ def test_mid_journey_prompt_chain(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": False,
         "info": "",
+        "file_path": "",
+        "fileTypes": [],
+        "value": "",
     }
     # Test the description object
     assert chain["description"] == "MidJourneyPromptChain is a chain you can use to generate new MidJourney prompts."
@@ -270,6 +288,9 @@ def test_time_travel_guide_chain(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": False,
         "info": "",
+        "file_path": "",
+        "fileTypes": [],
+        "value": "",
     }
     assert template["memory"] == {
         "required": False,
@@ -283,6 +304,9 @@ def test_time_travel_guide_chain(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": False,
         "info": "",
+        "file_path": "",
+        "fileTypes": [],
+        "value": "",
     }
 
     assert chain["description"] == "Time travel guide chain."

tests/test_frontend_nodes.py

@@ -31,9 +31,9 @@ def test_template_field_defaults(sample_template_field: TemplateField):
     assert sample_template_field.is_list is False
     assert sample_template_field.show is True
     assert sample_template_field.multiline is False
-    assert sample_template_field.value is None
+    assert sample_template_field.value == ""
     assert sample_template_field.file_types == []
-    assert sample_template_field.file_path is None
+    assert sample_template_field.file_path == ""
     assert sample_template_field.password is False
     assert sample_template_field.name == "test_field"
 

tests/test_llms_template.py

@@ -22,6 +22,7 @@ def test_openai(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": False,
         "info": "",
+        "fileTypes": [],
     }
     assert template["verbose"] == {
         "required": False,
@@ -35,6 +36,7 @@ def test_openai(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": False,
         "info": "",
+        "fileTypes": [],
     }
     assert template["client"] == {
         "required": False,
@@ -48,6 +50,7 @@ def test_openai(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": False,
         "info": "",
+        "fileTypes": [],
     }
     assert template["model_name"] == {
         "required": False,
@@ -69,6 +72,7 @@ def test_openai(client: TestClient, logged_in_headers):
         "list": True,
         "advanced": False,
         "info": "",
+        "fileTypes": [],
     }
     # Add more assertions for other properties here
     assert template["temperature"] == {
@@ -84,6 +88,7 @@ def test_openai(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": False,
         "info": "",
+        "fileTypes": [],
     }
     assert template["max_tokens"] == {
         "required": False,
@@ -98,6 +103,7 @@ def test_openai(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": False,
         "info": "",
+        "fileTypes": [],
     }
     assert template["top_p"] == {
         "required": False,
@@ -112,6 +118,7 @@ def test_openai(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": False,
         "info": "",
+        "fileTypes": [],
     }
     assert template["frequency_penalty"] == {
         "required": False,
@@ -126,6 +133,7 @@ def test_openai(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": False,
         "info": "",
+        "fileTypes": [],
     }
     assert template["presence_penalty"] == {
         "required": False,
@@ -140,6 +148,7 @@ def test_openai(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": False,
         "info": "",
+        "fileTypes": [],
     }
     assert template["n"] == {
         "required": False,
@@ -154,6 +163,7 @@ def test_openai(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": False,
         "info": "",
+        "fileTypes": [],
     }
     assert template["best_of"] == {
         "required": False,
@@ -168,6 +178,7 @@ def test_openai(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": False,
         "info": "",
+        "fileTypes": [],
     }
     assert template["model_kwargs"] == {
         "required": False,
@@ -181,6 +192,7 @@ def test_openai(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": True,
         "info": "",
+        "fileTypes": [],
     }
     assert template["openai_api_key"] == {
         "required": False,
@@ -196,6 +208,7 @@ def test_openai(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": False,
         "info": "",
+        "fileTypes": [],
     }
     assert template["batch_size"] == {
         "required": False,
@@ -210,6 +223,7 @@ def test_openai(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": False,
         "info": "",
+        "fileTypes": [],
     }
     assert template["request_timeout"] == {
         "required": False,
@@ -223,6 +237,7 @@ def test_openai(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": False,
         "info": "",
+        "fileTypes": [],
     }
     assert template["logit_bias"] == {
         "required": False,
@@ -236,6 +251,7 @@ def test_openai(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": False,
         "info": "",
+        "fileTypes": [],
     }
     assert template["max_retries"] == {
         "required": False,
@@ -250,6 +266,7 @@ def test_openai(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": False,
         "info": "",
+        "fileTypes": [],
     }
     assert template["streaming"] == {
         "required": False,
@@ -264,6 +281,7 @@ def test_openai(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": False,
         "info": "",
+        "fileTypes": [],
     }
 
 
@@ -289,6 +307,7 @@ def test_chat_open_ai(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": False,
         "info": "",
+        "fileTypes": [],
     }
     assert template["client"] == {
         "required": False,
@@ -302,6 +321,7 @@ def test_chat_open_ai(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": False,
         "info": "",
+        "fileTypes": [],
     }
     assert template["model_name"] == {
         "required": False,
@@ -324,6 +344,7 @@ def test_chat_open_ai(client: TestClient, logged_in_headers):
         "list": True,
         "advanced": False,
         "info": "",
+        "fileTypes": [],
     }
     assert template["temperature"] == {
         "required": False,
@@ -338,6 +359,7 @@ def test_chat_open_ai(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": False,
         "info": "",
+        "fileTypes": [],
     }
     assert template["model_kwargs"] == {
         "required": False,
@@ -351,6 +373,7 @@ def test_chat_open_ai(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": True,
         "info": "",
+        "fileTypes": [],
     }
     assert template["openai_api_key"] == {
         "required": False,
@@ -366,6 +389,7 @@ def test_chat_open_ai(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": False,
         "info": "",
+        "fileTypes": [],
     }
     assert template["request_timeout"] == {
         "required": False,
@@ -379,6 +403,7 @@ def test_chat_open_ai(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": False,
         "info": "",
+        "fileTypes": [],
     }
     assert template["max_retries"] == {
         "required": False,
@@ -393,6 +418,7 @@ def test_chat_open_ai(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": False,
         "info": "",
+        "fileTypes": [],
     }
     assert template["streaming"] == {
         "required": False,
@@ -407,6 +433,7 @@ def test_chat_open_ai(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": False,
         "info": "",
+        "fileTypes": [],
     }
     assert template["n"] == {
         "required": False,
@@ -421,6 +448,7 @@ def test_chat_open_ai(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": False,
         "info": "",
+        "fileTypes": [],
     }
 
     assert template["max_tokens"] == {
@@ -435,6 +463,7 @@ def test_chat_open_ai(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": False,
         "info": "",
+        "fileTypes": [],
     }
     assert template["_type"] == "ChatOpenAI"
     assert (

tests/test_prompts_template.py

@@ -31,6 +31,7 @@ def test_prompt_template(client: TestClient, logged_in_headers):
         "list": True,
         "advanced": False,
         "info": "",
+        "fileTypes": [],
     }
 
     assert template["output_parser"] == {
@@ -45,6 +46,7 @@ def test_prompt_template(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": False,
         "info": "",
+        "fileTypes": [],
     }
 
     assert template["partial_variables"] == {
@@ -59,6 +61,7 @@ def test_prompt_template(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": False,
         "info": "",
+        "fileTypes": [],
     }
 
     assert template["template"] == {
@@ -73,6 +76,7 @@ def test_prompt_template(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": False,
         "info": "",
+        "fileTypes": [],
     }
 
     assert template["template_format"] == {
@@ -88,6 +92,7 @@ def test_prompt_template(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": False,
         "info": "",
+        "fileTypes": [],
     }
 
     assert template["validate_template"] == {
@@ -103,4 +108,5 @@ def test_prompt_template(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": False,
         "info": "",
+        "fileTypes": [],
     }