replace parse_raw with model_validate_json

defog-ai · Jan 18, 2025 · 208ca8d · 208ca8d
1 parent 7e48c91
commit 208ca8d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/defog_utils/utils_llm.py b/defog_utils/utils_llm.py
@@ -432,7 +432,7 @@ def chat_gemini(
 
     if response_format:
         # convert the content into Pydantic class
-        content = response_format.parse_raw(content)
+        content = response_format.model_validate_json(content)
 
     return LLMResponse(
         model=model,
@@ -494,7 +494,7 @@ async def chat_gemini_async(
 
     if response_format:
         # convert the content into Pydantic class
-        content = response_format.parse_raw(content)
+        content = response_format.model_validate_json(content)
 
     return LLMResponse(
         model=model,