From c2da021cac784f3c68fd46d03fdca0f4b2886abb Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Tue, 2 Jan 2024 23:15:33 -0300
Subject: [PATCH] Refactor ChatAdapter to ChatDefinition

---
 src/backend/langflow/utils/chat.py | 15 +++++++++++----
 1 file changed, 11 insertions(+), 4 deletions(-)

diff --git a/src/backend/langflow/utils/chat.py b/src/backend/langflow/utils/chat.py
index 84405e38e..35b4f22a6 100644
--- a/src/backend/langflow/utils/chat.py
+++ b/src/backend/langflow/utils/chat.py
@@ -1,17 +1,24 @@
 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
 
+from langflow.utils.prompt import GenericPromptTemplate
+
 PromptTemplate = Union[LCPromptTemplate, LIPromptTemplate]
 
 
-class ChatAdapter:
-    def __init__(self, func: Callable, inputs: list[str], output_key: str, prompt: Optional[PromptTemplate] = None):
+class ChatDefinition:
+    def __init__(
+        self,
+        func: Callable,
+        inputs: list[str],
+        output_key: Optional[str] = None,
+        prompt: Optional[PromptTemplate] = None,
+    ):
         self.func = func
         self.input_keys = inputs
-        self.output_keys = output_key
+        self.output_key = output_key
         self.prompt = prompt
 
     @classmethod