euriai 1.0.7__tar.gz → 1.0.8__tar.gz

{euriai-1.0.7 → euriai-1.0.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: euriai
-Version: 1.0.7
+Version: 1.0.8
 Summary: Python client for Euri API (euron.one) with CLI, LangChain, and LlamaIndex integration
 Author: Euri
 Author-email: tech@euron.one

{euriai-1.0.7 → euriai-1.0.8}/euriai/__init__.py

@@ -4,7 +4,7 @@ Euri AI Python SDK
 A comprehensive Python SDK for the Euri AI API with integrations for popular frameworks.
 """
 
-__version__ = "1.0.7"
+__version__ = "1.0.8"
 
 # Core imports that should always work
 try:

{euriai-1.0.7 → euriai-1.0.8}/euriai/autogen.py

@@ -9,6 +9,49 @@ except ImportError:
     autogen = None
     AssistantAgent = UserProxyAgent = GroupChat = GroupChatManager = None
 
+def _ensure_autogen_imports():
+    """Ensure AutoGen classes are properly imported."""
+    global autogen, AssistantAgent, UserProxyAgent, GroupChat, GroupChatManager
+    
+    try:
+        # Force re-import of autogen module
+        import importlib
+        import sys
+        
+        # Remove any cached modules to force fresh import
+        if 'autogen' in sys.modules:
+            importlib.reload(sys.modules['autogen'])
+        
+        # Import the main module
+        import autogen as _autogen
+        
+        # Import individual classes with explicit error handling
+        from autogen import AssistantAgent as _AssistantAgent
+        from autogen import UserProxyAgent as _UserProxyAgent  
+        from autogen import GroupChat as _GroupChat
+        from autogen import GroupChatManager as _GroupChatManager
+        
+        # Verify that all classes are not None
+        if _AssistantAgent is None or _UserProxyAgent is None:
+            raise ImportError("AutoGen classes imported as None")
+        
+        # Update global variables
+        autogen = _autogen
+        AssistantAgent = _AssistantAgent
+        UserProxyAgent = _UserProxyAgent
+        GroupChat = _GroupChat
+        GroupChatManager = _GroupChatManager
+        
+        # Verify the globals were updated
+        if AssistantAgent is None:
+            raise ImportError("Failed to update global AssistantAgent variable")
+            
+        return True
+        
+    except ImportError as e:
+        from . import check_optional_dependency
+        check_optional_dependency("pyautogen", "AutoGen", "autogen")
+
 class EuriaiModelClient:
     """
     Custom model client that uses Euri API for AutoGen integration.
@@ -197,9 +240,7 @@ class EuriaiAutoGen:
             api_key: Your Euri API key
             default_model: Default model to use
         """
-        if autogen is None or AssistantAgent is None or UserProxyAgent is None:
-            from . import check_optional_dependency
-            check_optional_dependency("pyautogen", "AutoGen", "autogen")
+        _ensure_autogen_imports()
         
         self.api_key = api_key
         self.default_model = default_model
@@ -216,7 +257,7 @@ class EuriaiAutoGen:
         temperature: float = 0.7,
         max_tokens: int = 1000,
         **kwargs
-    ) -> AssistantAgent:
+    ) -> "AssistantAgent":
         """
         Create an assistant agent with Euri API integration.
         
@@ -231,8 +272,13 @@ class EuriaiAutoGen:
         Returns:
             Configured AssistantAgent
         """
-        # Check if AutoGen is properly imported
-        if AssistantAgent is None:
+        # Import AutoGen classes directly to ensure they're available
+        try:
+            from autogen import AssistantAgent
+            # Verify the class is not None
+            if AssistantAgent is None:
+                raise ImportError("AutoGen AssistantAgent class is None")
+        except ImportError:
             from . import check_optional_dependency
             check_optional_dependency("pyautogen", "AutoGen", "autogen")
         
@@ -265,7 +311,7 @@ class EuriaiAutoGen:
         is_termination_msg: Optional[callable] = None,
         code_execution_config: Optional[Dict[str, Any]] = None,
         **kwargs
-    ) -> UserProxyAgent:
+    ) -> "UserProxyAgent":
         """
         Create a user proxy agent.
         
@@ -278,8 +324,13 @@ class EuriaiAutoGen:
         Returns:
             Configured UserProxyAgent
         """
-        # Check if AutoGen is properly imported
-        if UserProxyAgent is None:
+        # Import AutoGen classes directly to ensure they're available
+        try:
+            from autogen import UserProxyAgent
+            # Verify the class is not None
+            if UserProxyAgent is None:
+                raise ImportError("AutoGen UserProxyAgent class is None")
+        except ImportError:
             from . import check_optional_dependency
             check_optional_dependency("pyautogen", "AutoGen", "autogen")
         
@@ -316,10 +367,8 @@ class EuriaiAutoGen:
         Returns:
             Configured GroupChat
         """
-        # Check if AutoGen is properly imported
-        if GroupChat is None:
-            from . import check_optional_dependency
-            check_optional_dependency("pyautogen", "AutoGen", "autogen")
+        # Ensure AutoGen classes are properly imported
+        _ensure_autogen_imports()
         
         self.group_chat = GroupChat(
             agents=agents,
@@ -353,10 +402,8 @@ class EuriaiAutoGen:
         Returns:
             Configured GroupChatManager
         """
-        # Check if AutoGen is properly imported
-        if GroupChatManager is None:
-            from . import check_optional_dependency
-            check_optional_dependency("pyautogen", "AutoGen", "autogen")
+        # Ensure AutoGen classes are properly imported
+        _ensure_autogen_imports()
         
         # Create config for Euri API
         config_list = [{

{euriai-1.0.7 → euriai-1.0.8}/euriai.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: euriai
-Version: 1.0.7
+Version: 1.0.8
 Summary: Python client for Euri API (euron.one) with CLI, LangChain, and LlamaIndex integration
 Author: Euri
 Author-email: tech@euron.one

{euriai-1.0.7 → euriai-1.0.8}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="euriai",
-    version="1.0.7",
+    version="1.0.8",
     description="Python client for Euri API (euron.one) with CLI, LangChain, and LlamaIndex integration",
     long_description=open("README.md", encoding="utf-8").read(),
     long_description_content_type="text/markdown",