Refactor ChatAdapter to ChatDefinition

This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-01-02 23:15:33 -03:00
commit c2da021cac

View file

@ -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