add convinence method to call chat model as an llm (#1604)

tool-patch
Harrison Chase 1 year ago committed by GitHub
parent 15de3e8137
commit 5903a93f3d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -12,6 +12,7 @@ from langchain.schema import (
BaseMessage,
ChatGeneration,
ChatResult,
HumanMessage,
LLMResult,
PromptValue,
)
@ -116,6 +117,10 @@ class BaseChatModel(BaseLanguageModel, BaseModel, ABC):
) -> BaseMessage:
return self._generate(messages, stop=stop).generations[0].message
def call_as_llm(self, message: str, stop: Optional[List[str]] = None) -> str:
result = self([HumanMessage(content=message)], stop=stop)
return result.content
class SimpleChatModel(BaseChatModel):
def _generate(

Loading…
Cancel
Save