github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-28 06:48:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Harrison Chase
2023-03-11 15:04:57 -08:00
committed by GitHub
parent 15de3e8137
commit 5903a93f3d
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
langchain/chat_models/base.py
View File

@@ -12,6 +12,7 @@ from langchain.schema import (
BaseMessage, BaseMessage,
ChatGeneration, ChatGeneration,
ChatResult, ChatResult,
HumanMessage,
LLMResult, LLMResult,
PromptValue, PromptValue,
) )
@@ -116,6 +117,10 @@ class BaseChatModel(BaseLanguageModel, BaseModel, ABC):
) -> BaseMessage: ) -> BaseMessage:
return self._generate(messages, stop=stop).generations[0].message 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): class SimpleChatModel(BaseChatModel):
def _generate( def _generate(