Skip to content
Closed
Show file tree
Hide file tree
Changes from 2 commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 7 additions & 1 deletion flaml/autogen/agentchat/contrib/math_user_proxy_agent.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,13 @@
from time import sleep
from typing import Any, Callable, Dict, List, Optional, Union

from pydantic import BaseModel, Extra, root_validator
import pydantic
from packaging import version

if version(pydantic.__version__) < version.parse("2.0.0"):
from pydantic import BaseModel, Extra, root_validator
else:
from pydantic.v1 import BaseModel, Extra, root_validator

from flaml.autogen.agentchat import Agent, UserProxyAgent
from flaml.autogen.code_utils import UNKNOWN, execute_code, extract_code, infer_lang
Expand Down
2 changes: 1 addition & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -90,7 +90,7 @@
"tensorboardX==2.6", # test_forecast_panel
"requests<2.29.0", # https://github.com/docker/docker-py/issues/3113
"packaging",
"pydantic==1.10.9",
"pydantic",
"sympy",
"wolframalpha",
"dill", # a drop in replacement of pickle
Expand Down
Loading