update pydantic model validator

knowsuchagency · Jan 3, 2025 · 3039a3b · 3039a3b
1 parent 768dec0
commit 3039a3b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/promptic.py b/promptic.py
@@ -10,7 +10,7 @@
 from jsonschema import validate as validate_json_schema
 from pydantic import BaseModel
 
-__version__ = "4.0.0"
+__version__ = "4.0.1"
 
 SystemPrompt = Optional[Union[str, List[str], List[Dict[str, str]]]]
 
@@ -193,7 +193,7 @@ def _parse_and_validate_response(self, generated_text: str, return_type: Any):
                     self.state.add_message(
                         {"content": json_result, "role": "assistant"}
                     )
-                return return_type.model_validate_json(json_result)
+                return return_type.model_validate(json.loads(json_result))
             raise ValueError("Failed to extract JSON result from the generated text.")
 
         # Handle json_schema if provided

diff --git a/uv.lock b/uv.lock