From 5dfaa9a81a6268c419ecccb04282e222083a74f6 Mon Sep 17 00:00:00 2001
From: Gabriel Almeida <gabriel@logspace.ai>
Date: Tue, 23 May 2023 11:51:31 -0300
Subject: [PATCH] =?UTF-8?q?=F0=9F=8E=A8=20style(tests):=20add=20display=5F?=
 =?UTF-8?q?name=20field=20to=20LLM=20in=20test=5Fagents=5Ftemplate.py=20an?=
 =?UTF-8?q?d=20test=5Fchains=5Ftemplate.py=20The=20display=5Fname=20field?=
 =?UTF-8?q?=20was=20added=20to=20the=20LLM=20object=20in=20the=20test=5Fag?=
 =?UTF-8?q?ents=5Ftemplate.py=20and=20test=5Fchains=5Ftemplate.py=20files.?=
 =?UTF-8?q?=20This=20field=20is=20used=20to=20display=20the=20name=20of=20?=
 =?UTF-8?q?the=20LLM=20in=20the=20UI.?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 tests/test_agents_template.py | 3 +++
 tests/test_chains_template.py | 3 +++
 2 files changed, 6 insertions(+)

diff --git a/tests/test_agents_template.py b/tests/test_agents_template.py
index 750050c25..7aa8de176 100644
--- a/tests/test_agents_template.py
+++ b/tests/test_agents_template.py
@@ -82,6 +82,7 @@ def test_json_agent(client: TestClient):
         "type": "BaseLanguageModel",
         "list": False,
         "advanced": False,
+        "display_name": "LLM",
     }
 
 
@@ -120,6 +121,7 @@ def test_csv_agent(client: TestClient):
         "type": "BaseLanguageModel",
         "list": False,
         "advanced": False,
+        "display_name": "LLM",
     }
 
 
@@ -183,4 +185,5 @@ def test_initialize_agent(client: TestClient):
         "type": "BaseLanguageModel",
         "list": False,
         "advanced": False,
+        "display_name": "LLM",
     }
diff --git a/tests/test_chains_template.py b/tests/test_chains_template.py
index f1154a556..c958cf64d 100644
--- a/tests/test_chains_template.py
+++ b/tests/test_chains_template.py
@@ -326,6 +326,7 @@ def test_series_character_chain(client: TestClient):
 
     assert template["llm"] == {
         "required": True,
+        "display_name": "LLM",
         "placeholder": "",
         "show": True,
         "multiline": False,
@@ -388,6 +389,7 @@ def test_mid_journey_prompt_chain(client: TestClient):
 
     assert template["llm"] == {
         "required": True,
+        "display_name": "LLM",
         "placeholder": "",
         "show": True,
         "multiline": False,
@@ -427,6 +429,7 @@ def test_time_travel_guide_chain(client: TestClient):
     assert template["llm"] == {
         "required": True,
         "placeholder": "",
+        "display_name": "LLM",
         "show": True,
         "multiline": False,
         "password": False,