github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-23 19:39:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

core: upgrade mypy to recent mypy (#18753)

Browse Source 
Testing this works per package on CI
This commit is contained in:
Eugene Yurtsev
2024-03-07 15:25:19 -05:00
committed by GitHub
parent e188d4ecb0
commit 8c71f92cb2
23 changed files with 172 additions and 155 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -91,7 +91,7 @@ class BaseMessagePromptTemplate(Serializable, ABC):
Returns:
Combined prompt template.
"""
prompt = ChatPromptTemplate(messages=[self])
prompt = ChatPromptTemplate(messages=[self]) # type: ignore[call-arg]
return prompt + other
@@ -603,17 +603,17 @@ class ChatPromptTemplate(BaseChatPromptTemplate):
"""
# Allow for easy combining
if isinstance(other, ChatPromptTemplate):
return ChatPromptTemplate(messages=self.messages + other.messages)
return ChatPromptTemplate(messages=self.messages + other.messages) # type: ignore[call-arg]
elif isinstance(
other, (BaseMessagePromptTemplate, BaseMessage, BaseChatPromptTemplate)
):
return ChatPromptTemplate(messages=self.messages + [other])
return ChatPromptTemplate(messages=self.messages + [other]) # type: ignore[call-arg]
elif isinstance(other, (list, tuple)):
_other = ChatPromptTemplate.from_messages(other)
return ChatPromptTemplate(messages=self.messages + _other.messages)
return ChatPromptTemplate(messages=self.messages + _other.messages) # type: ignore[call-arg]
elif isinstance(other, str):
prompt = HumanMessagePromptTemplate.from_template(other)
return ChatPromptTemplate(messages=self.messages + [prompt])
return ChatPromptTemplate(messages=self.messages + [prompt]) # type: ignore[call-arg]
else:
raise NotImplementedError(f"Unsupported operand type for +: {type(other)}")
@@ -684,7 +684,7 @@ class ChatPromptTemplate(BaseChatPromptTemplate):
Returns:
a chat prompt template
"""
return cls(
return cls( # type: ignore[call-arg]
messages=[
ChatMessagePromptTemplate.from_template(template, role=role)
for role, template in string_messages

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

@@ -255,7 +255,7 @@ class PromptTemplate(StringPromptTemplate):
return cls(
input_variables=input_variables,
template=template,
template_format=template_format,
template_format=template_format, # type: ignore[arg-type]
partial_variables=_partial_variables,
**kwargs,
)