From 1f3d162e86505956c7a7a113ceb6b06deae703f0 Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Mon, 26 Feb 2024 21:50:51 -0300
Subject: [PATCH] Remove unused ChatDefinition class

---
 src/backend/langflow/utils/chat.py | 34 ------------------------------
 1 file changed, 34 deletions(-)
 delete mode 100644 src/backend/langflow/utils/chat.py

diff --git a/src/backend/langflow/utils/chat.py b/src/backend/langflow/utils/chat.py
deleted file mode 100644
index e1621bdab..000000000
--- a/src/backend/langflow/utils/chat.py
+++ /dev/null
@@ -1,34 +0,0 @@
-from typing import Any, Callable, Optional, Union
-
-from langchain_core.prompts import PromptTemplate as LCPromptTemplate
-from langflow.utils.prompt import GenericPromptTemplate
-from llama_index.prompts import PromptTemplate as LIPromptTemplate
-
-PromptTemplate = Union[LCPromptTemplate, LIPromptTemplate]
-
-
-class ChatDefinition:
-    def __init__(
-        self,
-        func: Callable,
-        inputs: list[str],
-        output_key: Optional[str] = None,
-        prompt_template: Optional[PromptTemplate] = None,
-    ):
-        self.func = func
-        self.input_keys = inputs
-        self.output_key = output_key
-        self.prompt_template = prompt_template
-
-    @classmethod
-    def from_prompt_template(cls, prompt_template: PromptTemplate, func: Callable, output_key: Optional[str] = None):
-        prompt = GenericPromptTemplate(prompt_template)
-        return cls(
-            func=func,
-            inputs=prompt.input_keys,
-            output_key=output_key,
-            prompt_template=prompt_template,
-        )
-
-    def __call__(self, inputs: dict, callbacks: Optional[Any] = None) -> dict:
-        return self.func(inputs, callbacks)