dict() -> model_dump()

langchain-ai · Sep 4, 2024 · 5572e09 · 5572e09
1 parent 3d5afcb
commit 5572e09
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/libs/partners/prompty/tests/unit_tests/fake_chat_model.py b/libs/partners/prompty/tests/unit_tests/fake_chat_model.py
@@ -21,7 +21,7 @@ def _call(
         run_manager: Optional[CallbackManagerForLLMRun] = None,
         **kwargs: Any,
     ) -> str:
-        return json.dumps([message.dict() for message in messages])
+        return json.dumps([message.model_dump() for message in messages])
 
     async def _agenerate(
         self,