euriai 1.0.16__tar.gz → 1.0.18__tar.gz

{euriai-1.0.16 → euriai-1.0.18}/PKG-INFO

@@ -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 → euriai-1.0.18}/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-1.0.16 → euriai-1.0.18}/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 → euriai-1.0.18}/euriai.egg-info/PKG-INFO

@@ -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 → euriai-1.0.18}/setup.py

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