From 907c3ca9f7074b8ab7f088e6fed83f4f1513694b Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Mon, 4 Dec 2023 16:18:31 -0300
Subject: [PATCH] Refactor test_agents_template.py by removing unused test
 cases and initializing agent template

---
 tests/test_agents_template.py | 74 -----------------------------------
 1 file changed, 74 deletions(-)

diff --git a/tests/test_agents_template.py b/tests/test_agents_template.py
index 60dc266ad..6b0ccce8d 100644
--- a/tests/test_agents_template.py
+++ b/tests/test_agents_template.py
@@ -192,77 +192,3 @@ def test_csv_agent(client: TestClient, logged_in_headers):
         "display_name": "LLM",
         "info": "",
     }
-
-
-def test_initialize_agent(client: TestClient, logged_in_headers):
-    response = client.get("api/v1/all", headers=logged_in_headers)
-    assert response.status_code == 200
-    json_response = response.json()
-    agents = json_response["agents"]
-
-    initialize_agent = agents["AgentInitializer"]
-    assert initialize_agent["base_classes"] == ["AgentExecutor", "Chain", "Callable"]
-    template = initialize_agent["template"]
-
-    assert template["agent"] == {
-        "required": True,
-        "dynamic": False,
-        "placeholder": "",
-        "show": True,
-        "multiline": False,
-        "value": "zero-shot-react-description",
-        "password": False,
-        "options": [
-            "zero-shot-react-description",
-            "react-docstore",
-            "self-ask-with-search",
-            "conversational-react-description",
-            "openai-functions",
-            "openai-multi-functions",
-        ],
-        "name": "agent",
-        "type": "str",
-        "list": True,
-        "advanced": False,
-        "info": "",
-    }
-    assert template["memory"] == {
-        "required": False,
-        "dynamic": False,
-        "placeholder": "",
-        "show": True,
-        "multiline": False,
-        "password": False,
-        "name": "memory",
-        "type": "BaseChatMemory",
-        "list": False,
-        "advanced": False,
-        "info": "",
-    }
-    assert template["tools"] == {
-        "required": True,
-        "dynamic": False,
-        "placeholder": "",
-        "show": True,
-        "multiline": False,
-        "password": False,
-        "name": "tools",
-        "type": "Tool",
-        "list": True,
-        "advanced": False,
-        "info": "",
-    }
-    assert template["llm"] == {
-        "required": True,
-        "dynamic": False,
-        "placeholder": "",
-        "show": True,
-        "multiline": False,
-        "password": False,
-        "name": "llm",
-        "type": "BaseLanguageModel",
-        "list": False,
-        "advanced": False,
-        "display_name": "LLM",
-        "info": "",
-    }