github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-17 10:13:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(core): correct return type hints in BaseChatPromptTemplate (#32009)

Browse Source 
This PR changes the return type hints of the `format_prompt` and
`aformat_prompt` methods in `BaseChatPromptTemplate` from `PromptValue`
to `ChatPromptValue`. Since both methods always return a
`ChatPromptValue`.
This commit is contained in:
Andreas V. Jonsterhaug 2025-07-14 17:00:01 +02:00 committed by GitHub
parent d57216c295
commit 6dcca35a34
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -35,7 +35,7 @@ from langchain_core.messages import (
convert_to_messages, convert_to_messages,
) )
from langchain_core.messages.base import get_msg_title_repr from langchain_core.messages.base import get_msg_title_repr
from langchain_core.prompt_values import ChatPromptValue, ImageURL, PromptValue from langchain_core.prompt_values import ChatPromptValue, ImageURL
from langchain_core.prompts.base import BasePromptTemplate from langchain_core.prompts.base import BasePromptTemplate
from langchain_core.prompts.dict import DictPromptTemplate from langchain_core.prompts.dict import DictPromptTemplate
from langchain_core.prompts.image import ImagePromptTemplate from langchain_core.prompts.image import ImagePromptTemplate
@ -715,20 +715,20 @@ class BaseChatPromptTemplate(BasePromptTemplate, ABC):
""" """
return (await self.aformat_prompt(**kwargs)).to_string() return (await self.aformat_prompt(**kwargs)).to_string()
def format_prompt(self, **kwargs: Any) -> PromptValue: def format_prompt(self, **kwargs: Any) -> ChatPromptValue:
"""Format prompt. Should return a PromptValue. """Format prompt. Should return a ChatPromptValue.
Args: Args:
**kwargs: Keyword arguments to use for formatting. **kwargs: Keyword arguments to use for formatting.
Returns: Returns:
PromptValue. ChatPromptValue.
""" """
messages = self.format_messages(**kwargs) messages = self.format_messages(**kwargs)
return ChatPromptValue(messages=messages) return ChatPromptValue(messages=messages)
async def aformat_prompt(self, **kwargs: Any) -> PromptValue: async def aformat_prompt(self, **kwargs: Any) -> ChatPromptValue:
"""Async format prompt. Should return a PromptValue. """Async format prompt. Should return a ChatPromptValue.
Args: Args:
**kwargs: Keyword arguments to use for formatting. **kwargs: Keyword arguments to use for formatting.