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

euriai/langgraph.py

@@ -248,6 +248,7 @@ class EuriaiLangGraph:
         default_temperature: float = 0.7,
         default_max_tokens: int = 1000,
         enable_checkpointing: bool = True,
+        enable_usage_tracking: bool = True,
         verbose: bool = True
     ):
         """
@@ -260,6 +261,7 @@ class EuriaiLangGraph:
             default_temperature: Default temperature
             default_max_tokens: Default max tokens
             enable_checkpointing: Enable workflow checkpointing
+            enable_usage_tracking: Enable usage statistics tracking
             verbose: Enable verbose logging
         """
         if not LANGGRAPH_AVAILABLE:
@@ -273,6 +275,7 @@ class EuriaiLangGraph:
         self.default_model = default_model
         self.default_temperature = default_temperature
         self.default_max_tokens = default_max_tokens
+        self.enable_usage_tracking = enable_usage_tracking
         self.verbose = verbose
         
         # Initialize graph
@@ -297,8 +300,10 @@ class EuriaiLangGraph:
             "total_runs": 0,
             "total_nodes_executed": 0,
             "avg_execution_time": 0.0,
+            "total_api_calls": 0,
             "errors": 0,
-            "successful_runs": 0
+            "successful_runs": 0,
+            "ai_nodes": {}  # Node-specific statistics
         }
         
         # Thread pool for async operations
@@ -344,6 +349,15 @@ class EuriaiLangGraph:
         self.nodes[name] = ai_node
         self.graph.add_node(name, ai_node)
         
+        # Initialize node-specific statistics
+        if self.enable_usage_tracking:
+            self.usage_stats["ai_nodes"][name] = {
+                "total_calls": 0,
+                "total_tokens": 0,
+                "errors": 0,
+                "avg_response_time": 0.0
+            }
+        
         if self.verbose:
             print(f"Added AI node: {name} (model: {ai_node.model})")
     
@@ -399,6 +413,24 @@ class EuriaiLangGraph:
         if self.verbose:
             print(f"Added condition node: {name} with routes: {routes}")
     
+    def add_conditional_node(
+        self,
+        name: str,
+        condition_func: Callable[[Dict[str, Any]], str],
+        routes: Dict[str, str]
+    ) -> None:
+        """
+        Alias for add_condition_node for more intuitive naming.
+        
+        Add a conditional node that routes based on state.
+        
+        Args:
+            name: Node name
+            condition_func: Function that returns route key based on state
+            routes: Mapping of route keys to target nodes
+        """
+        return self.add_condition_node(name, condition_func, routes)
+    
     def add_embedding_node(
         self,
         name: str,
@@ -704,6 +736,71 @@ class EuriaiLangGraph:
         
         return results
     
+    def get_usage_stats(self) -> Dict[str, Any]:
+        """
+        Get comprehensive usage statistics for the workflow.
+        
+        Returns:
+            Dictionary containing usage statistics including:
+            - total_runs: Total number of workflow runs
+            - successful_runs: Number of successful runs
+            - errors: Number of errors
+            - avg_execution_time: Average execution time per run
+            - total_api_calls: Total API calls made
+            - ai_nodes: Node-specific statistics
+        """
+        if not self.enable_usage_tracking:
+            return {"message": "Usage tracking is disabled"}
+        
+        # Update AI node statistics from individual nodes
+        for node_name, ai_node in self.ai_nodes.items():
+            if node_name in self.usage_stats["ai_nodes"]:
+                node_stats = ai_node.usage_stats
+                self.usage_stats["ai_nodes"][node_name].update({
+                    "total_calls": node_stats["total_calls"],
+                    "total_tokens": node_stats["total_tokens"],
+                    "errors": node_stats["errors"],
+                    "avg_response_time": node_stats["avg_response_time"]
+                })
+                
+                # Add to total API calls
+                self.usage_stats["total_api_calls"] += node_stats["total_calls"]
+        
+        return self.usage_stats.copy()
+    
+    def reset_usage_stats(self) -> None:
+        """Reset all usage statistics."""
+        if not self.enable_usage_tracking:
+            return
+        
+        self.usage_stats = {
+            "total_runs": 0,
+            "total_nodes_executed": 0,
+            "avg_execution_time": 0.0,
+            "total_api_calls": 0,
+            "errors": 0,
+            "successful_runs": 0,
+            "ai_nodes": {}
+        }
+        
+        # Reset individual AI node stats
+        for ai_node in self.ai_nodes.values():
+            ai_node.usage_stats = {
+                "total_calls": 0,
+                "total_tokens": 0,
+                "errors": 0,
+                "avg_response_time": 0.0
+            }
+        
+        # Reinitialize AI node stats
+        for name in self.ai_nodes.keys():
+            self.usage_stats["ai_nodes"][name] = {
+                "total_calls": 0,
+                "total_tokens": 0,
+                "errors": 0,
+                "avg_response_time": 0.0
+            }
+    
     def create_workflow_pattern(self, pattern_type: WorkflowType, **kwargs) -> None:
         """
         Create a pre-defined workflow pattern.

{euriai-1.0.16.dist-info → euriai-1.0.18.dist-info}/METADATA

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

{euriai-1.0.16.dist-info → euriai-1.0.18.dist-info}/RECORD

@@ -1,4 +1,4 @@
-euriai/__init__.py,sha256=bMcJY6pFQlDwgqVx_Y0hsXrixtbjQUadIQ3uBXYnBSw,6427
+euriai/__init__.py,sha256=7HAeI4LEAaq9UPyxXxt5iV3HBOIn4lGBnJ3VFCGfbBU,6427
 euriai/autogen.py,sha256=z1WHftUgu3_Sn8zDXmf31onikS0p8TwH5JE4llL7ogk,21144
 euriai/cli.py,sha256=hF1wiiL2QQSfWf8WlLQyNVDBd4YkbiwmMSoPxVbyPTM,3290
 euriai/client.py,sha256=L-o6hv9N3md-l-hz-kz5nYVaaZqnrREZlo_0jguhF7E,4066
@@ -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=7VXN3NDH-NwbBraCTVT_LKtWGXeglUMi7jmIGqKf9rk,36959
+euriai/langgraph.py,sha256=XfOLj5J5KXIPG_BnXV2MzjbloLNOl0wIf0MbnovoznY,40503
 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.16.dist-info/METADATA,sha256=C__QQAA_DaQCVfs4pxD5E_hAzW-coIxg3e31KptMjRk,6807
-euriai-1.0.16.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-euriai-1.0.16.dist-info/entry_points.txt,sha256=9OkET8KIGcsjQn8UlnpPKRT75s2KW34jq1__1SXtpMA,43
-euriai-1.0.16.dist-info/top_level.txt,sha256=TG1htJ8cuD62MXn-NJ7DVF21QHY16w6M_QgfF_Er_EQ,7
-euriai-1.0.16.dist-info/RECORD,,
+euriai-1.0.18.dist-info/METADATA,sha256=iDxnmLXiPmv3BaMVQlGCThABooMqKAQd13e_xUv4SJo,6807
+euriai-1.0.18.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+euriai-1.0.18.dist-info/entry_points.txt,sha256=9OkET8KIGcsjQn8UlnpPKRT75s2KW34jq1__1SXtpMA,43
+euriai-1.0.18.dist-info/top_level.txt,sha256=TG1htJ8cuD62MXn-NJ7DVF21QHY16w6M_QgfF_Er_EQ,7
+euriai-1.0.18.dist-info/RECORD,,