Refactor ChatAdapter to ChatDefinition
This commit is contained in:
parent
8421e60f10
commit
c2da021cac
1 changed files with 11 additions and 4 deletions
|
|
@ -1,17 +1,24 @@
|
||||||
from typing import Any, Callable, Optional, Union
|
from typing import Any, Callable, Optional, Union
|
||||||
|
|
||||||
from langchain_core.prompts import PromptTemplate as LCPromptTemplate
|
from langchain_core.prompts import PromptTemplate as LCPromptTemplate
|
||||||
from langflow.utils.prompt import GenericPromptTemplate
|
|
||||||
from llama_index.prompts import PromptTemplate as LIPromptTemplate
|
from llama_index.prompts import PromptTemplate as LIPromptTemplate
|
||||||
|
|
||||||
|
from langflow.utils.prompt import GenericPromptTemplate
|
||||||
|
|
||||||
PromptTemplate = Union[LCPromptTemplate, LIPromptTemplate]
|
PromptTemplate = Union[LCPromptTemplate, LIPromptTemplate]
|
||||||
|
|
||||||
|
|
||||||
class ChatAdapter:
|
class ChatDefinition:
|
||||||
def __init__(self, func: Callable, inputs: list[str], output_key: str, prompt: Optional[PromptTemplate] = None):
|
def __init__(
|
||||||
|
self,
|
||||||
|
func: Callable,
|
||||||
|
inputs: list[str],
|
||||||
|
output_key: Optional[str] = None,
|
||||||
|
prompt: Optional[PromptTemplate] = None,
|
||||||
|
):
|
||||||
self.func = func
|
self.func = func
|
||||||
self.input_keys = inputs
|
self.input_keys = inputs
|
||||||
self.output_keys = output_key
|
self.output_key = output_key
|
||||||
self.prompt = prompt
|
self.prompt = prompt
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue