From 6a1a0d977a994a743d5a297a2345cc9f760d0297 Mon Sep 17 00:00:00 2001 From: Julian Weng Date: Thu, 20 Jun 2024 11:03:21 -0400 Subject: [PATCH] partners[minor]: Fix value error message for with_structured_output (#22877) Currently, calling `with_structured_output()` with an invalid method argument raises `Unrecognized method argument. Expected one of 'function_calling' or 'json_format'`, but the JSON mode option [is now referred to](https://python.langchain.com/v0.2/docs/how_to/structured_output/#the-with_structured_output-method) by `'json_mode'`. This fixes that. Co-authored-by: Harrison Chase --- libs/partners/fireworks/langchain_fireworks/chat_models.py | 2 +- libs/partners/groq/langchain_groq/chat_models.py | 2 +- libs/partners/openai/langchain_openai/chat_models/base.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/libs/partners/fireworks/langchain_fireworks/chat_models.py b/libs/partners/fireworks/langchain_fireworks/chat_models.py index fb439a6811..c94d16a2d9 100644 --- a/libs/partners/fireworks/langchain_fireworks/chat_models.py +++ b/libs/partners/fireworks/langchain_fireworks/chat_models.py @@ -904,7 +904,7 @@ class ChatFireworks(BaseChatModel): else: raise ValueError( f"Unrecognized method argument. Expected one of 'function_calling' or " - f"'json_format'. Received: '{method}'" + f"'json_mode'. Received: '{method}'" ) if include_raw: diff --git a/libs/partners/groq/langchain_groq/chat_models.py b/libs/partners/groq/langchain_groq/chat_models.py index df49602b73..d791a4c207 100644 --- a/libs/partners/groq/langchain_groq/chat_models.py +++ b/libs/partners/groq/langchain_groq/chat_models.py @@ -1034,7 +1034,7 @@ class ChatGroq(BaseChatModel): else: raise ValueError( f"Unrecognized method argument. Expected one of 'function_calling' or " - f"'json_format'. Received: '{method}'" + f"'json_mode'. Received: '{method}'" ) if include_raw: diff --git a/libs/partners/openai/langchain_openai/chat_models/base.py b/libs/partners/openai/langchain_openai/chat_models/base.py index 06be457610..2927427951 100644 --- a/libs/partners/openai/langchain_openai/chat_models/base.py +++ b/libs/partners/openai/langchain_openai/chat_models/base.py @@ -1139,7 +1139,7 @@ class BaseChatOpenAI(BaseChatModel): else: raise ValueError( f"Unrecognized method argument. Expected one of 'function_calling' or " - f"'json_format'. Received: '{method}'" + f"'json_mode'. Received: '{method}'" ) if include_raw: