Create Langchain LLM converter to quickly supply it to Langchain's chain (#102)

* Create Langchain LLM converter to quickly supply it to Langchain's chain

* Clean up
This commit is contained in:
Duc Nguyen (john)
2023-12-11 14:55:56 +07:00
committed by GitHub
parent da0ac1d69f
commit 0e30dcbb06
10 changed files with 24 additions and 15 deletions

View File

@@ -3,11 +3,12 @@ from __future__ import annotations
import logging
from kotaemon.base import BaseComponent
from kotaemon.llms.base import BaseLLM
logger = logging.getLogger(__name__)
class ChatLLM(BaseComponent):
class ChatLLM(BaseLLM):
def flow(self):
if self.inflow is None:
raise ValueError("No inflow provided.")

View File

@@ -68,6 +68,9 @@ class LCChatMixin:
logits=[],
)
def to_langchain_format(self):
return self._obj
def __repr__(self):
kwargs = []
for key, value_obj in self._kwargs.items():