github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-25 04:49:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

core: Bump ruff version to 0.11 (#30519)

Browse Source 
Changes are from the new TC006 rule:
https://docs.astral.sh/ruff/rules/runtime-cast-value/
TC006 is auto-fixed.
This commit is contained in:
Christophe Bornet
2025-03-27 18:01:49 +01:00
committed by GitHub
parent 59908f04d4
commit e181d43214
44 changed files with 243 additions and 233 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -530,14 +530,14 @@ class _StringImageMessagePromptTemplate(BaseMessagePromptTemplate):
if isinstance(tmpl, str):
text: str = tmpl
else:
text = cast(_TextTemplateParam, tmpl)["text"] # type: ignore[assignment]
text = cast("_TextTemplateParam", tmpl)["text"] # type: ignore[assignment]
prompt.append(
PromptTemplate.from_template(
text, template_format=template_format
)
)
elif isinstance(tmpl, dict) and "image_url" in tmpl:
img_template = cast(_ImageTemplateParam, tmpl)["image_url"]
img_template = cast("_ImageTemplateParam", tmpl)["image_url"]
input_variables = []
if isinstance(img_template, str):
vars = get_template_variables(img_template, template_format)
@@ -1024,7 +1024,7 @@ class ChatPromptTemplate(BaseChatPromptTemplate):
"partial_variables": partial_vars,
**kwargs,
}
cast(type[ChatPromptTemplate], super()).__init__(messages=_messages, **kwargs)
cast("type[ChatPromptTemplate]", super()).__init__(messages=_messages, **kwargs)
@classmethod
def get_lc_namespace(cls) -> list[str]:
@@ -1382,11 +1382,11 @@ def _create_template_from_message_type(
)
elif message_type in ("ai", "assistant"):
message = AIMessagePromptTemplate.from_template(
cast(str, template), template_format=template_format
cast("str", template), template_format=template_format
)
elif message_type == "system":
message = SystemMessagePromptTemplate.from_template(
cast(str, template), template_format=template_format
cast("str", template), template_format=template_format
)
elif message_type == "placeholder":
if isinstance(template, str):
@@ -1484,7 +1484,7 @@ def _convert_to_message(
else:
_message = message_type_str(
prompt=PromptTemplate.from_template(
cast(str, template), template_format=template_format
cast("str", template), template_format=template_format
)
)
else: