euriai 1.0.12__tar.gz → 1.0.14__tar.gz

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

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

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

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

@@ -285,23 +285,17 @@ class EuriaiAutoGen:
         Returns:
             Configured AssistantAgent
         """
-        # Import AutoGen AssistantAgent class and detect version
-        autogen_version = "0.2"  # Default to old version
-        AssistantAgent = None
-        
+        # Import AutoGen AssistantAgent class
         try:
             # Try newer AutoGen structure first (v0.6+)
             try:
                 from autogen_agentchat.agents import AssistantAgent
-                autogen_version = "0.6+"
             except ImportError:
                 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")
@@ -309,53 +303,25 @@ class EuriaiAutoGen:
             # because check_optional_dependency raises ImportError
             raise  # Re-raise the original ImportError
         
-        model_name = model or self.default_model
+        # 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
+        }]
         
-        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
-            )
+        # Create agent
+        agent = AssistantAgent(
+            name=name,
+            system_message=system_message,
+            llm_config={"config_list": config_list},
+            **kwargs
+        )
+        
+        # Register the custom model client
+        agent.register_model_client(model_client_cls=EuriaiModelClient)
         
         self.agents.append(agent)
         return agent

{euriai-1.0.12 → euriai-1.0.14}/euriai/langgraph.py

@@ -518,6 +518,17 @@ class EuriaiLangGraph:
             if self.compiled_graph is None:
                 self.compile_graph()
             
+            # Prepare config for checkpointer if needed
+            if config is None:
+                config = {}
+            
+            # If checkpointer is enabled and no thread_id provided, create one
+            if self.checkpointer is not None and "configurable" not in config:
+                import uuid
+                config["configurable"] = {
+                    "thread_id": str(uuid.uuid4())
+                }
+            
             # Execute workflow
             result = self.compiled_graph.invoke(input_state, config=config)
             

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

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

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

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