euriai 1.0.7__py3-none-any.whl → 1.0.8__py3-none-any.whl

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/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.dist-info → euriai-1.0.8.dist-info}/METADATA

@@ -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.dist-info → euriai-1.0.8.dist-info}/RECORD

@@ -1,5 +1,5 @@
-euriai/__init__.py,sha256=Dvycjk0FyVASMXaSDBMHzpCsfJlJlmO6nLNtZ9rvrRs,6629
-euriai/autogen.py,sha256=7lT_T1RT9RdytrSwbj3UhQ6_mVdoMy4Mr7y4HdWDPkc,16961
+euriai/__init__.py,sha256=gfkPrWaBzOwtzPHz4IaRIkr2euZTec0me8nMtxOlOeQ,6629
+euriai/autogen.py,sha256=1AkzH0ctH4vR6kbPzYNzR2J6n9kdDpZIx9qQGgxl-Bg,18611
 euriai/cli.py,sha256=hF1wiiL2QQSfWf8WlLQyNVDBd4YkbiwmMSoPxVbyPTM,3290
 euriai/client.py,sha256=L-o6hv9N3md-l-hz-kz5nYVaaZqnrREZlo_0jguhF7E,4066
 euriai/crewai.py,sha256=nfMMOOhKiCIc2v42nj2Zf9rP0U5m4GrfK_6zkXL77gw,7594
@@ -12,8 +12,8 @@ euriai/langgraph.py,sha256=sw9e-PnfwAwmp_tUCnAGIUB78GyJsMkAzxOGvFUafiM,34128
 euriai/llamaindex.py,sha256=c-ujod2bjL6QIvfAyuIxm1SvSCS00URFElYybKQ5Ew0,26551
 euriai/n8n.py,sha256=hjkckqyW_hZNL78UkBCof1WvKCKCIjwdvZdAgx6NrB8,3764
 euriai/smolagents.py,sha256=xlixGx2IWzAPTpSJGsYIK2L-SHGY9Mw1-8GbwVsEYtU,28507
-euriai-1.0.7.dist-info/METADATA,sha256=gxZb_093o7FZ1-G599QQ6HCEuUpEPCMS-OXz72siVHE,6806
-euriai-1.0.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-euriai-1.0.7.dist-info/entry_points.txt,sha256=9OkET8KIGcsjQn8UlnpPKRT75s2KW34jq1__1SXtpMA,43
-euriai-1.0.7.dist-info/top_level.txt,sha256=TG1htJ8cuD62MXn-NJ7DVF21QHY16w6M_QgfF_Er_EQ,7
-euriai-1.0.7.dist-info/RECORD,,
+euriai-1.0.8.dist-info/METADATA,sha256=SrcXi2Fcm7qHODmX1Du_XptolZTWvWR3WIXHfRYVGXs,6806
+euriai-1.0.8.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+euriai-1.0.8.dist-info/entry_points.txt,sha256=9OkET8KIGcsjQn8UlnpPKRT75s2KW34jq1__1SXtpMA,43
+euriai-1.0.8.dist-info/top_level.txt,sha256=TG1htJ8cuD62MXn-NJ7DVF21QHY16w6M_QgfF_Er_EQ,7
+euriai-1.0.8.dist-info/RECORD,,