euriai 1.0.12__py3-none-any.whl → 1.0.15__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.12"
+__version__ = "1.0.15"
 
 # Core imports that should always work
 try:

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/langgraph.py

@@ -156,15 +156,15 @@ class EuriaiAINode:
             # Format prompt with state variables
             formatted_prompt = self.prompt_template.format(**state)
             
-            # Prepare messages
-            messages = []
+            # Prepare user prompt (combine system message if provided)
+            user_prompt = formatted_prompt
             if self.system_message:
-                messages.append({"role": "system", "content": self.system_message})
-            messages.append({"role": "user", "content": formatted_prompt})
+                # If there's a system message, combine it with the user prompt
+                user_prompt = f"System: {self.system_message}\n\nUser: {formatted_prompt}"
             
             # Make API call
             response = self.client.generate_completion(
-                messages=messages,
+                prompt=user_prompt,
                 temperature=self.temperature,
                 max_tokens=self.max_tokens
             )
@@ -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.dist-info → euriai-1.0.15.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: euriai
-Version: 1.0.12
+Version: 1.0.15
 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.dist-info → euriai-1.0.15.dist-info}/RECORD

@@ -1,5 +1,5 @@
-euriai/__init__.py,sha256=fOu_GfzC0JWBu232SdRHJQoBNk-pP4pb77qlGoEBKm0,6630
-euriai/autogen.py,sha256=SACnTH7d_gnb6nIaaMN4bvMkNUxpeRq_VZ3o0thEB08,22678
+euriai/__init__.py,sha256=BW-vN8t7z6wkpm_Vu4xM-roD2NiMhF-IKb6MJV1KzKM,6630
+euriai/autogen.py,sha256=z1WHftUgu3_Sn8zDXmf31onikS0p8TwH5JE4llL7ogk,21144
 euriai/cli.py,sha256=hF1wiiL2QQSfWf8WlLQyNVDBd4YkbiwmMSoPxVbyPTM,3290
 euriai/client.py,sha256=L-o6hv9N3md-l-hz-kz5nYVaaZqnrREZlo_0jguhF7E,4066
 euriai/crewai.py,sha256=nfMMOOhKiCIc2v42nj2Zf9rP0U5m4GrfK_6zkXL77gw,7594
@@ -8,12 +8,12 @@ euriai/embedding.py,sha256=uP66Ph1k9Ou6J5RAkztJxlfyj0S0MESOvZ4ulhnVo-o,1270
 euriai/euri_chat.py,sha256=DEAiet1ReRwB4ljkPYaTl1Nb5uc20-JF-3PQjGQZXk4,3567
 euriai/euri_embed.py,sha256=g7zs1G-ZBDJjOGJtkkfIcV4LPtRcm9wpVWmrfMGn5EM,2919
 euriai/langchain.py,sha256=gVF9eh21RC1WtDn7SQoEREUDqOObm5IRx6BFZtB5xcc,34968
-euriai/langgraph.py,sha256=sw9e-PnfwAwmp_tUCnAGIUB78GyJsMkAzxOGvFUafiM,34128
+euriai/langgraph.py,sha256=YTApEQ5gte_HI06x_zMBiNt1O87JlVspaWref954_7I,34636
 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.12.dist-info/METADATA,sha256=cWWOJemR6qbJjHI3CRQ8kpIa5hpIuUeCtGchD8A8XHU,6807
-euriai-1.0.12.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-euriai-1.0.12.dist-info/entry_points.txt,sha256=9OkET8KIGcsjQn8UlnpPKRT75s2KW34jq1__1SXtpMA,43
-euriai-1.0.12.dist-info/top_level.txt,sha256=TG1htJ8cuD62MXn-NJ7DVF21QHY16w6M_QgfF_Er_EQ,7
-euriai-1.0.12.dist-info/RECORD,,
+euriai-1.0.15.dist-info/METADATA,sha256=6ky-wJh84rTRc521m-nHTFXYbvtmRPaxQ83JmbfXTg8,6807
+euriai-1.0.15.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+euriai-1.0.15.dist-info/entry_points.txt,sha256=9OkET8KIGcsjQn8UlnpPKRT75s2KW34jq1__1SXtpMA,43
+euriai-1.0.15.dist-info/top_level.txt,sha256=TG1htJ8cuD62MXn-NJ7DVF21QHY16w6M_QgfF_Er_EQ,7
+euriai-1.0.15.dist-info/RECORD,,