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

core[patch]: update LanguageModelInput from List to Sequence (#14405)

Browse Source 
Co-authored-by: Erick Friis <erick@langchain.dev>
This commit is contained in:
Li-Lun Lin 2024-01-03 10:49:01 +08:00 committed by GitHub
parent 4dab37741a
commit cda68d717c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
libs/core/langchain_core/language_models/base.py
View File

@ -48,7 +48,7 @@ def _get_token_ids_default_method(text: str) -> List[int]:
return tokenizer.encode(text) return tokenizer.encode(text)
LanguageModelInput = Union[PromptValue, str, List[BaseMessage]] LanguageModelInput = Union[PromptValue, str, Sequence[BaseMessage]]
LanguageModelOutput = Union[BaseMessage, str] LanguageModelOutput = Union[BaseMessage, str]
LanguageModelLike = Runnable[LanguageModelInput, LanguageModelOutput] LanguageModelLike = Runnable[LanguageModelInput, LanguageModelOutput]
LanguageModelOutputVar = TypeVar("LanguageModelOutputVar", BaseMessage, str) LanguageModelOutputVar = TypeVar("LanguageModelOutputVar", BaseMessage, str)

2
libs/core/langchain_core/language_models/chat_models.py
View File

@ -142,7 +142,7 @@ class BaseChatModel(BaseLanguageModel[BaseMessage], ABC):
return input return input
elif isinstance(input, str): elif isinstance(input, str):
return StringPromptValue(text=input) return StringPromptValue(text=input)
elif isinstance(input, list): elif isinstance(input, Sequence):
return ChatPromptValue(messages=input) return ChatPromptValue(messages=input)
else: else:
raise ValueError( raise ValueError(

2
libs/core/langchain_core/language_models/llms.py
View File

@ -205,7 +205,7 @@ class BaseLLM(BaseLanguageModel[str], ABC):
return input return input
elif isinstance(input, str): elif isinstance(input, str):
return StringPromptValue(text=input) return StringPromptValue(text=input)
elif isinstance(input, list): elif isinstance(input, Sequence):
return ChatPromptValue(messages=input) return ChatPromptValue(messages=input)
else: else:
raise ValueError( raise ValueError(