Fix MathReasoning typing in test_structured_output for 3.8 compatibility

Signed-off-by: Sternakt <[email protected]>
ag2ai · Dec 4, 2024 · b22d6cf · b22d6cf
1 parent 89ac536
commit b22d6cf
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/test/agentchat/test_structured_output.py b/test/agentchat/test_structured_output.py
@@ -7,6 +7,7 @@
 
 import os
 import sys
+from typing import List
 from unittest.mock import MagicMock
 
 import pytest
@@ -70,7 +71,7 @@ class Step(BaseModel):
 
 
 class MathReasoning(BaseModel):
-    steps: list[Step]
+    steps: List[Step]
     final_answer: str
 
     def format(self) -> str: