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

core[patch]: Message content as positional arg (#16921)

Browse Source
This commit is contained in:
Bagatur 2024-02-02 10:24:02 -08:00 committed by GitHub
parent c29e9b6412
commit d80c612c92
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
libs/core/langchain_core/messages/base.py
View File

@ -28,6 +28,12 @@ class BaseMessage(Serializable):
class Config: class Config:
extra = Extra.allow extra = Extra.allow
def __init__(
self, content: Union[str, List[Union[str, Dict]]], **kwargs: Any
) -> None:
"""Pass in content as positional arg."""
return super().__init__(content=content, **kwargs)
@classmethod @classmethod
def is_lc_serializable(cls) -> bool: def is_lc_serializable(cls) -> bool:
"""Return whether this class is serializable.""" """Return whether this class is serializable."""

4
libs/core/langchain_core/prompt_values.py
View File

@ -1,7 +1,7 @@
from __future__ import annotations from __future__ import annotations
from abc import ABC, abstractmethod from abc import ABC, abstractmethod
from typing import List, Literal, Sequence from typing import List, Literal, Sequence, cast
from typing_extensions import TypedDict from typing_extensions import TypedDict
@ -105,7 +105,7 @@ class ImagePromptValue(PromptValue):
def to_messages(self) -> List[BaseMessage]: def to_messages(self) -> List[BaseMessage]:
"""Return prompt as messages.""" """Return prompt as messages."""
return [HumanMessage(content=[self.image_url])] return [HumanMessage(content=[cast(dict, self.image_url)])]
class ChatPromptValueConcrete(ChatPromptValue): class ChatPromptValueConcrete(ChatPromptValue):

2
libs/core/langchain_core/prompts/chat.py
View File

@ -446,7 +446,7 @@ class _StringImageMessagePromptTemplate(BaseMessagePromptTemplate):
content=text, additional_kwargs=self.additional_kwargs content=text, additional_kwargs=self.additional_kwargs
) )
else: else:
content = [] content: List = []
for prompt in self.prompt: for prompt in self.prompt:
inputs = {var: kwargs[var] for var in prompt.input_variables} inputs = {var: kwargs[var] for var in prompt.input_variables}
if isinstance(prompt, StringPromptTemplate): if isinstance(prompt, StringPromptTemplate):