restore: llamaindex_conversable_agent.py

ag2ai · Dec 2, 2024 · 4b5dcbf · 4b5dcbf
1 parent 14d0042
commit 4b5dcbf
Showing 1 changed file with 3 additions and 11 deletions.
diff --git a/autogen/agentchat/contrib/llamaindex_conversable_agent.py b/autogen/agentchat/contrib/llamaindex_conversable_agent.py
@@ -17,20 +17,12 @@
     from llama_index.core.base.llms.types import ChatMessage
     from llama_index.core.chat_engine.types import AgentChatResponse
     from pydantic import BaseModel
-    from pydantic import __version__ as pydantic_version
-
-    # let's Avoid: AttributeError: type object 'Config' has no attribute 'copy'
-    if pydantic_version >= "2.0":
-        from pydantic import ConfigDict
-
-        Config = ConfigDict(arbitrary_types_allowed=True)
-    else:
-
-        class Config:
-            arbitrary_types_allowed = True
 
     # Add Pydantic configuration to allow arbitrary types
     # Added to mitigate PydanticSchemaGenerationError
+    class Config:
+        arbitrary_types_allowed = True
+
     BaseModel.model_config = Config
 
 except ImportError as e: