Fix test_structured_output test

ag2ai · Dec 5, 2024 · f2c2fcb · f2c2fcb
1 parent a67ad8e
commit f2c2fcb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/agentchat/test_structured_output.py b/test/agentchat/test_structured_output.py
@@ -38,7 +38,7 @@ class ResponseModel(BaseModel):
         difficulty: float
 
     for config in config_list:
-        config["response_format"] = MathReasoning
+        config["response_format"] = ResponseModel
 
     llm_config = {"config_list": config_list, "cache_seed": 43}