From b22d6cf416090561930f25c4bf43379c16f8b4fe Mon Sep 17 00:00:00 2001 From: Sternakt Date: Wed, 4 Dec 2024 14:58:25 +0100 Subject: [PATCH] Fix MathReasoning typing in test_structured_output for 3.8 compatibility Signed-off-by: Sternakt --- test/agentchat/test_structured_output.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/test/agentchat/test_structured_output.py b/test/agentchat/test_structured_output.py index 24d90d3c6a..614cb8f604 100644 --- 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: