From 5903a93f3d4167cfab38a51b23e99272ced1f63a Mon Sep 17 00:00:00 2001 From: Harrison Chase Date: Sat, 11 Mar 2023 15:04:57 -0800 Subject: [PATCH] add convinence method to call chat model as an llm (#1604) --- langchain/chat_models/base.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/langchain/chat_models/base.py b/langchain/chat_models/base.py index 62ffa57a..fc8daba6 100644 --- a/langchain/chat_models/base.py +++ b/langchain/chat_models/base.py @@ -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(