From d0f10e107ec2c79e32b17dee2f467822d3ec90a6 Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Wed, 28 Feb 2024 13:17:40 -0300
Subject: [PATCH] Fix tests

---
 src/backend/langflow/api/v1/callback.py     |   5 +
 src/backend/langflow/services/chat/utils.py |  18 +---
 tests/test_prompts_template.py              | 102 --------------------
 3 files changed, 6 insertions(+), 119 deletions(-)

diff --git a/src/backend/langflow/api/v1/callback.py b/src/backend/langflow/api/v1/callback.py
index 86409c767..f9be5d56c 100644
--- a/src/backend/langflow/api/v1/callback.py
+++ b/src/backend/langflow/api/v1/callback.py
@@ -14,6 +14,11 @@ from loguru import logger
 class AsyncStreamingLLMCallbackHandler(AsyncCallbackHandler):
     """Callback handler for streaming LLM responses."""
 
+    @property
+    def ignore_chain(self) -> bool:
+        """Whether to ignore chain callbacks."""
+        return False
+
     def __init__(self, client_id: str):
         self.chat_service = get_chat_service()
         self.client_id = client_id
diff --git a/src/backend/langflow/services/chat/utils.py b/src/backend/langflow/services/chat/utils.py
index 85e6cdcd5..0562b9df2 100644
--- a/src/backend/langflow/services/chat/utils.py
+++ b/src/backend/langflow/services/chat/utils.py
@@ -3,12 +3,10 @@ from typing import Any
 from langchain.agents import AgentExecutor
 from langchain.chains.base import Chain
 from langchain_core.runnables import Runnable
-from loguru import logger
-
 from langflow.api.v1.schemas import ChatMessage
 from langflow.interface.utils import try_setting_streaming_options
 from langflow.processing.base import get_result_and_steps
-from langflow.utils.chat import ChatDefinition
+from loguru import logger
 
 LANGCHAIN_RUNNABLES = (Chain, Runnable, AgentExecutor)
 
@@ -40,20 +38,6 @@ async def process_graph(
                 client_id=client_id,
                 session_id=session_id,
             )
-        elif isinstance(build_result, ChatDefinition):
-            raw_output = await run_build_result(
-                build_result,
-                chat_inputs,
-                client_id=client_id,
-                session_id=session_id,
-            )
-            if isinstance(raw_output, dict):
-                if not build_result.output_key:
-                    raise ValueError("No output key provided to ChatDefinition when returning a dict.")
-                result = raw_output[build_result.output_key]
-            else:
-                result = raw_output
-            intermediate_steps = []
         else:
             raise TypeError(f"Unknown type {type(build_result)}")
         logger.debug("Generated result and intermediate_steps")
diff --git a/tests/test_prompts_template.py b/tests/test_prompts_template.py
index ca2bedb13..505b9db2f 100644
--- a/tests/test_prompts_template.py
+++ b/tests/test_prompts_template.py
@@ -1,5 +1,4 @@
 from fastapi.testclient import TestClient
-
 from langflow.services.deps import get_settings_service
 
 
@@ -10,104 +9,3 @@ def test_prompts_settings(client: TestClient, logged_in_headers):
     json_response = response.json()
     prompts = json_response["prompts"]
     assert set(prompts.keys()) == set(settings_service.settings.PROMPTS)
-
-
-def test_prompt_template(client: TestClient, logged_in_headers):
-    response = client.get("api/v1/all", headers=logged_in_headers)
-    assert response.status_code == 200
-    json_response = response.json()
-    prompts = json_response["prompts"]
-
-    prompt = prompts["PromptTemplate"]
-    template = prompt["template"]
-    assert template["input_variables"] == {
-        "required": True,
-        "dynamic": True,
-        "placeholder": "",
-        "show": False,
-        "multiline": False,
-        "password": False,
-        "name": "input_variables",
-        "type": "str",
-        "list": True,
-        "advanced": False,
-        "info": "",
-        "fileTypes": [],
-    }
-
-    assert template["output_parser"] == {
-        "required": False,
-        "dynamic": True,
-        "placeholder": "",
-        "show": False,
-        "multiline": False,
-        "password": False,
-        "name": "output_parser",
-        "type": "BaseOutputParser",
-        "list": False,
-        "advanced": False,
-        "info": "",
-        "fileTypes": [],
-    }
-
-    assert template["partial_variables"] == {
-        "required": False,
-        "dynamic": True,
-        "placeholder": "",
-        "show": False,
-        "multiline": False,
-        "password": False,
-        "name": "partial_variables",
-        "type": "dict",
-        "list": False,
-        "advanced": False,
-        "info": "",
-        "fileTypes": [],
-    }
-
-    assert template["template"] == {
-        "required": True,
-        "dynamic": True,
-        "placeholder": "",
-        "show": True,
-        "multiline": True,
-        "password": False,
-        "name": "template",
-        "type": "prompt",
-        "list": False,
-        "advanced": False,
-        "info": "",
-        "fileTypes": [],
-    }
-
-    assert template["template_format"] == {
-        "required": False,
-        "dynamic": True,
-        "placeholder": "",
-        "show": False,
-        "multiline": False,
-        "value": "f-string",
-        "password": False,
-        "name": "template_format",
-        "type": "str",
-        "list": False,
-        "advanced": False,
-        "info": "",
-        "fileTypes": [],
-    }
-
-    assert template["validate_template"] == {
-        "required": False,
-        "dynamic": True,
-        "placeholder": "",
-        "show": False,
-        "multiline": False,
-        "value": False,
-        "password": False,
-        "name": "validate_template",
-        "type": "bool",
-        "list": False,
-        "advanced": False,
-        "info": "",
-        "fileTypes": [],
-    }