github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-08-31 10:23:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

core: add error message for non-structured llm to StructuredPrompt (#22684)

Browse Source 
previously was the blank `NotImplementedError` from
`BaseLanguageModel.with_structured_output`
This commit is contained in:
Erick Friis
2024-06-07 12:42:09 -07:00
committed by GitHub
parent 02ff78deb8
commit 9e03864d64
1 changed files with 11 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
libs/core/langchain_core/prompts/structured.py
View File

@@ -118,10 +118,19 @@ class StructuredPrompt(ChatPromptTemplate):
if isinstance(other, BaseLanguageModel) or hasattr(
other, "with_structured_output"
):
return RunnableSequence(self, other.with_structured_output(self.schema_))
try:
return RunnableSequence(
self, other.with_structured_output(self.schema_)
)
except NotImplementedError as e:
raise NotImplementedError(
"Structured prompts must be piped to a language model that "
"implements with_structured_output."
) from e
else:
raise NotImplementedError(
"Structured prompts need to be piped to a language model."
"Structured prompts must be piped to a language model that "
"implements with_structured_output."
)
def pipe(