euriai 1.0.10__tar.gz → 1.0.12__tar.gz

{euriai-1.0.10 → euriai-1.0.12}/PKG-INFO

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

{euriai-1.0.10 → euriai-1.0.12}/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.10"
+__version__ = "1.0.12"
 
 # Core imports that should always work
 try:

{euriai-1.0.10 → euriai-1.0.12}/euriai/autogen.py

@@ -5,12 +5,19 @@ from euriai.client import EuriaiClient
 try:
     # Try newer AutoGen structure first (v0.6+)
     try:
-        from autogen_agentchat import AssistantAgent, UserProxyAgent, GroupChat, GroupChatManager
+        from autogen_agentchat.agents import AssistantAgent, UserProxyAgent
+        from autogen_agentchat.teams import GroupChat, GroupChatManager
         import autogen_agentchat as autogen
     except ImportError:
-        # Fall back to older AutoGen structure
-        import autogen
-        from autogen import AssistantAgent, UserProxyAgent, GroupChat, GroupChatManager
+        # Try alternative newer structure
+        try:
+            from autogen_agentchat.base import AssistantAgent, UserProxyAgent
+            from autogen_agentchat import GroupChat, GroupChatManager
+            import autogen_agentchat as autogen
+        except ImportError:
+            # Fall back to older AutoGen structure
+            import autogen
+            from autogen import AssistantAgent, UserProxyAgent, GroupChat, GroupChatManager
 except ImportError:
     autogen = None
     AssistantAgent = UserProxyAgent = GroupChat = GroupChatManager = None
@@ -278,14 +285,23 @@ class EuriaiAutoGen:
         Returns:
             Configured AssistantAgent
         """
-        # Import AutoGen AssistantAgent class
+        # Import AutoGen AssistantAgent class and detect version
+        autogen_version = "0.2"  # Default to old version
+        AssistantAgent = None
+        
         try:
             # Try newer AutoGen structure first (v0.6+)
             try:
-                from autogen_agentchat import AssistantAgent
+                from autogen_agentchat.agents import AssistantAgent
+                autogen_version = "0.6+"
             except ImportError:
-                # Fall back to older AutoGen structure
-                from autogen import AssistantAgent
+                try:
+                    from autogen_agentchat.base import AssistantAgent
+                    autogen_version = "0.6+"
+                except ImportError:
+                    # Fall back to older AutoGen structure
+                    from autogen import AssistantAgent
+                    autogen_version = "0.2"
         except ImportError:
             from . import check_optional_dependency
             check_optional_dependency("pyautogen", "AutoGen", "autogen")
@@ -293,25 +309,53 @@ class EuriaiAutoGen:
             # because check_optional_dependency raises ImportError
             raise  # Re-raise the original ImportError
         
-        # Create config for Euri API
-        config_list = [{
-            "model": model or self.default_model,
-            "model_client_cls": "EuriaiModelClient",
-            "api_key": self.api_key,
-            "temperature": temperature,
-            "max_tokens": max_tokens
-        }]
-        
-        # Create agent
-        agent = AssistantAgent(
-            name=name,
-            system_message=system_message,
-            llm_config={"config_list": config_list},
-            **kwargs
-        )
+        model_name = model or self.default_model
         
-        # Register the custom model client
-        agent.register_model_client(model_client_cls=EuriaiModelClient)
+        if autogen_version == "0.6+":
+            # Use new AutoGen 0.6+ API with model_client
+            try:
+                # Try to import the model client for new API
+                from autogen_ext.models.openai import OpenAIChatCompletionClient
+                
+                # Create model client that connects to Euri API
+                model_client = OpenAIChatCompletionClient(
+                    model=model_name,
+                    api_key=self.api_key,
+                    base_url="https://api.euriai.com/v1/",
+                    temperature=temperature,
+                    max_tokens=max_tokens
+                )
+                
+                # Create and return assistant agent with new API
+                agent = AssistantAgent(
+                    name=name,
+                    model_client=model_client,
+                    system_message=system_message or "You are a helpful AI assistant.",
+                    **kwargs
+                )
+                
+            except ImportError:
+                raise ImportError(
+                    "AutoGen 0.6+ detected but autogen_ext.models.openai is not available. "
+                    "Please install the full AutoGen package with: pip install pyautogen[openai]"
+                )
+        else:
+            # Use old AutoGen 0.2 API with llm_config
+            config_list = [{
+                "model": model_name,
+                "api_key": self.api_key,
+                "base_url": "https://api.euriai.com/v1/",
+                "temperature": temperature,
+                "max_tokens": max_tokens
+            }]
+            
+            # Create agent with old API
+            agent = AssistantAgent(
+                name=name,
+                system_message=system_message,
+                llm_config={"config_list": config_list},
+                **kwargs
+            )
         
         self.agents.append(agent)
         return agent
@@ -339,10 +383,13 @@ class EuriaiAutoGen:
         try:
             # Try newer AutoGen structure first (v0.6+)
             try:
-                from autogen_agentchat import UserProxyAgent
+                from autogen_agentchat.agents import UserProxyAgent
             except ImportError:
-                # Fall back to older AutoGen structure
-                from autogen import UserProxyAgent
+                try:
+                    from autogen_agentchat.base import UserProxyAgent
+                except ImportError:
+                    # Fall back to older AutoGen structure
+                    from autogen import UserProxyAgent
         except ImportError:
             from . import check_optional_dependency
             check_optional_dependency("pyautogen", "AutoGen", "autogen")
@@ -387,10 +434,13 @@ class EuriaiAutoGen:
         try:
             # Try newer AutoGen structure first (v0.6+)
             try:
-                from autogen_agentchat import GroupChat
+                from autogen_agentchat.teams import GroupChat
             except ImportError:
-                # Fall back to older AutoGen structure
-                from autogen import GroupChat
+                try:
+                    from autogen_agentchat import GroupChat
+                except ImportError:
+                    # Fall back to older AutoGen structure
+                    from autogen import GroupChat
         except ImportError:
             from . import check_optional_dependency
             check_optional_dependency("pyautogen", "AutoGen", "autogen")
@@ -432,10 +482,13 @@ class EuriaiAutoGen:
         try:
             # Try newer AutoGen structure first (v0.6+)
             try:
-                from autogen_agentchat import GroupChatManager
+                from autogen_agentchat.teams import GroupChatManager
             except ImportError:
-                # Fall back to older AutoGen structure
-                from autogen import GroupChatManager
+                try:
+                    from autogen_agentchat import GroupChatManager
+                except ImportError:
+                    # Fall back to older AutoGen structure
+                    from autogen import GroupChatManager
         except ImportError:
             from . import check_optional_dependency
             check_optional_dependency("pyautogen", "AutoGen", "autogen")

{euriai-1.0.10 → euriai-1.0.12}/euriai.egg-info/PKG-INFO

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

{euriai-1.0.10 → euriai-1.0.12}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="euriai",
-    version="1.0.10",
+    version="1.0.12",
     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",