From 628855c50d553df4b07196cc7a1dc7afe44a864d Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Sat, 9 Dec 2023 23:23:14 -0300
Subject: [PATCH] Update template files with new fileTypes field

---
 tests/test_agents_template.py  | 23 ++++++++++++++++++++++-
 tests/test_chains_template.py  | 24 ++++++++++++++++++++++++
 tests/test_llms_template.py    | 29 +++++++++++++++++++++++++++++
 tests/test_prompts_template.py |  6 ++++++
 4 files changed, 81 insertions(+), 1 deletion(-)

diff --git a/tests/test_agents_template.py b/tests/test_agents_template.py
index 6b0ccce8d..a7919faac 100644
--- a/tests/test_agents_template.py
+++ b/tests/test_agents_template.py
@@ -28,6 +28,8 @@ def test_zero_shot_agent(client: TestClient, logged_in_headers):
         "list": True,
         "advanced": False,
         "info": "",
+        "fileTypes": [],
+        "value": None,
     }
 
     # Additional assertions for other template variables
@@ -43,6 +45,8 @@ def test_zero_shot_agent(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": False,
         "info": "",
+        "fileTypes": [],
+        "value": None,
     }
     assert template["llm"] == {
         "required": True,
@@ -56,6 +60,8 @@ def test_zero_shot_agent(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": False,
         "info": "",
+        "fileTypes": [],
+        "value": None,
     }
     assert template["output_parser"] == {
         "required": False,
@@ -69,6 +75,8 @@ def test_zero_shot_agent(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": False,
         "info": "",
+        "fileTypes": [],
+        "value": None,
     }
     assert template["input_variables"] == {
         "required": False,
@@ -82,6 +90,8 @@ def test_zero_shot_agent(client: TestClient, logged_in_headers):
         "list": True,
         "advanced": False,
         "info": "",
+        "fileTypes": [],
+        "value": None,
     }
     assert template["prefix"] == {
         "required": False,
@@ -96,6 +106,7 @@ def test_zero_shot_agent(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": False,
         "info": "",
+        "fileTypes": [],
     }
     assert template["suffix"] == {
         "required": False,
@@ -110,6 +121,7 @@ def test_zero_shot_agent(client: TestClient, logged_in_headers):
         "list": False,
         "advanced": False,
         "info": "",
+        "fileTypes": [],
     }
 
 
@@ -135,6 +147,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 +164,9 @@ def test_json_agent(client: TestClient, logged_in_headers):
         "advanced": False,
         "display_name": "LLM",
         "info": "",
+        "file_path": "",
+        "fileTypes": [],
+        "value": "",
     }
 
 
@@ -174,7 +192,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 +209,7 @@ def test_csv_agent(client: TestClient, logged_in_headers):
         "advanced": False,
         "display_name": "LLM",
         "info": "",
+        "file_path": "",
+        "fileTypes": [],
+        "value": "",
     }
diff --git a/tests/test_chains_template.py b/tests/test_chains_template.py
index 6627fb26c..2e705ac00 100644
--- a/tests/test_chains_template.py
+++ b/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."
diff --git a/tests/test_llms_template.py b/tests/test_llms_template.py
index 494c1cc90..352056de7 100644
--- a/tests/test_llms_template.py
+++ b/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 (
diff --git a/tests/test_prompts_template.py b/tests/test_prompts_template.py
index b9e55ce77..e1f679738 100644
--- a/tests/test_prompts_template.py
+++ b/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": [],
     }