traia-iatp 0.1.1__tar.gz → 0.1.2__tar.gz

{traia_iatp-0.1.1/src/traia_iatp.egg-info → traia_iatp-0.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: traia-iatp
-Version: 0.1.1
+Version: 0.1.2
 Summary: Inter-Agent Transfer Protocol (IATP) - Enable AI Agents to utilize other AI Agents as tools
 Project-URL: Documentation, https://pypi.org/project/traia-iatp
 Project-URL: Source, https://github.com/Traia-IO/IATP

{traia_iatp-0.1.1 → traia_iatp-0.1.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "traia-iatp"
-version = "0.1.1"
+version = "0.1.2"
 description = "Inter-Agent Transfer Protocol (IATP) - Enable AI Agents to utilize other AI Agents as tools"
 readme = "README.md"
 requires-python = ">=3.12.8"

{traia_iatp-0.1.1 → traia_iatp-0.1.2}/src/traia_iatp/server/templates/__main__.py.j2

@@ -32,6 +32,14 @@ from starlette.routing import Route
 from starlette.middleware.base import BaseHTTPMiddleware
 from starlette.responses import Response
 
+# Import AgentOps for monitoring and observability
+try:
+    import agentops
+    AGENTOPS_AVAILABLE = True
+except ImportError:
+    AGENTOPS_AVAILABLE = False
+    agentops = None
+
 
 import sys
 from pathlib import Path
@@ -385,6 +393,32 @@ def generate_self_signed_cert(cert_path: str = "cert.pem", key_path: str = "key.
 
 async def main():
     """Main function to start the A2A server with HTTP/2 support."""
+    
+    # Initialize AgentOps for monitoring and observability
+    agentops_session_id = None
+    if AGENTOPS_AVAILABLE:
+        agentops_api_key = os.environ.get("AGENTOPS_API_KEY")
+        if agentops_api_key:
+            try:
+                # Initialize AgentOps with CrewAI-friendly settings
+                agentops_session_id = agentops.init(
+                    api_key=agentops_api_key,
+                    skip_auto_end_session=True,  # Let CrewAI handle session lifecycle
+                    tags=["{{ agent_name }}", "{{ mcp_server_name }}", "IATP", "A2A"],
+                    auto_start_session=True
+                )
+                logger.info(f"✅ AgentOps initialized successfully - Session ID: {agentops_session_id}")
+                logger.info("📊 View session at: https://app.agentops.ai")
+            except Exception as e:
+                logger.warning(f"⚠️  Failed to initialize AgentOps: {e}")
+                logger.warning("   Continuing without AgentOps monitoring...")
+        else:
+            logger.info("ℹ️  AGENTOPS_API_KEY not set - AgentOps monitoring disabled")
+            logger.info("   Set AGENTOPS_API_KEY environment variable to enable monitoring")
+    else:
+        logger.info("ℹ️  AgentOps not installed - monitoring disabled")
+        logger.info("   Install with: pip install agentops")
+    
     # Get configuration from environment
     host = os.environ.get("HOST", "0.0.0.0")
     port = int(os.environ.get("PORT", 8000))
@@ -441,8 +475,30 @@ async def main():
     logger.info(f"SSE endpoints available at: /a2a/tasks/subscribe and /a2a/tasks/resubscribe")
     logger.info(f"HTTP/2 multiplexing enabled with max {config.h2_max_concurrent_streams} concurrent streams")
     
-    # Run the server with Hypercorn
-    await serve(app, config)
+    if agentops_session_id:
+        logger.info(f"📊 AgentOps Session ID: {agentops_session_id}")
+        logger.info("📊 Monitor agent performance at: https://app.agentops.ai")
+    
+    try:
+        # Run the server with Hypercorn
+        await serve(app, config)
+    except KeyboardInterrupt:
+        logger.info("🛑 Server shutdown requested")
+        if AGENTOPS_AVAILABLE and agentops_session_id:
+            try:
+                agentops.end_session('Success')
+                logger.info("✅ AgentOps session ended successfully")
+            except Exception as e:
+                logger.warning(f"⚠️  Failed to end AgentOps session: {e}")
+    except Exception as e:
+        logger.error(f"❌ Server error: {e}")
+        if AGENTOPS_AVAILABLE and agentops_session_id:
+            try:
+                agentops.end_session('Fail', end_state_reason=str(e))
+                logger.info("📊 AgentOps session ended with failure status")
+            except Exception as ae:
+                logger.warning(f"⚠️  Failed to end AgentOps session: {ae}")
+        raise
 
 
 if __name__ == "__main__":

{traia_iatp-0.1.1 → traia_iatp-0.1.2}/src/traia_iatp/server/templates/agent.py.j2

@@ -13,6 +13,18 @@ import os
 # Import MCP integration from traia_iatp.mcp
 from traia_iatp.mcp import MCPServerConfig, MCPAgentBuilder, run_with_mcp_tools, MCPServerInfo
 
+# Import AgentOps for operation tracking
+try:
+    import agentops
+    from agentops.sdk.decorators import operation
+    AGENTOPS_AVAILABLE = True
+except ImportError:
+    AGENTOPS_AVAILABLE = False
+    agentops = None
+    # Create a no-op decorator if AgentOps is not available
+    def operation(func):
+        return func
+
 logger = logging.getLogger(__name__)
 
 
@@ -52,6 +64,7 @@ class {{ class_name }}Agent:
             tools_subset=tools_subset
         )
     
+    @operation
     def process_request(self, request: str, context: Dict[str, Any] = None) -> str:
         """Process a request using the MCP server capabilities."""
         try:

{traia_iatp-0.1.1 → traia_iatp-0.1.2}/src/traia_iatp/server/templates/agent_executor.py.j2

@@ -20,6 +20,18 @@ from a2a.utils import new_agent_text_message
 from crewai import Task
 from traia_iatp.mcp import MCPServerConfig, MCPAgentBuilder, run_with_mcp_tools, MCPServerInfo
 
+# Import AgentOps for operation tracking
+try:
+    import agentops
+    from agentops.sdk.decorators import operation
+    AGENTOPS_AVAILABLE = True
+except ImportError:
+    AGENTOPS_AVAILABLE = False
+    agentops = None
+    # Create a no-op decorator if AgentOps is not available
+    def operation(func):
+        return func
+
 logger = logging.getLogger(__name__)
 
 # Create a thread pool for CPU-bound CrewAI operations
@@ -72,6 +84,7 @@ class {{ class_name }}AgentExecutor(AgentExecutor):
             tags=mcp_config.metadata.get("tags", [])
         )
     
+    @operation
     async def execute(self, context: RequestContext, event_queue: EventQueue) -> None:
         """Process a request using the {{ agent_name }} capabilities."""
         try:
@@ -104,6 +117,7 @@ class {{ class_name }}AgentExecutor(AgentExecutor):
                 msg.taskId = context.task_id
             await event_queue.enqueue_event(msg)
     
+    @operation
     async def _execute_standard(self, context: RequestContext, event_queue: EventQueue, request_text: str) -> None:
         """Execute standard (non-streaming) request."""
         # Get additional context if provided
@@ -152,6 +166,7 @@ class {{ class_name }}AgentExecutor(AgentExecutor):
             msg.taskId = context.task_id
         await event_queue.enqueue_event(msg)
     
+    @operation
     async def _execute_streaming(self, context: RequestContext, event_queue: EventQueue, request_text: str) -> None:
         """Execute streaming request using SSE."""
         try:
@@ -202,6 +217,7 @@ class {{ class_name }}AgentExecutor(AgentExecutor):
                 })
             )
     
+    @operation
     async def _stream_mcp_response(self, request_text: str, context: RequestContext) -> AsyncGenerator[str, None]:
         """
         Stream responses from MCP server.

{traia_iatp-0.1.1 → traia_iatp-0.1.2}/src/traia_iatp/server/templates/env.example.j2

@@ -5,6 +5,9 @@
 PORT=8000
 HOST=0.0.0.0
 
+# Required ENV variables
+AGENTOPS_API_KEY=your-agentops-api-key-here
+
 # Language Model Configuration
 LLM_MODEL=openai/gpt-4.1
 # You can also use other models:

{traia_iatp-0.1.1 → traia_iatp-0.1.2}/src/traia_iatp/server/templates/pyproject.toml.j2

@@ -7,6 +7,7 @@ requires-python = ">=3.12"
 dependencies = [
     "crewai>=0.130.0",
     "crewai-tools[mcp]>=0.47.1",
+    "agentops>=0.3.0",
     "a2a-sdk>=0.2.6",
     "fastmcp>=2.8.1",
     "hypercorn>=0.17.3",  # HTTP/2 support via h2 dependency

{traia_iatp-0.1.1 → traia_iatp-0.1.2/src/traia_iatp.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: traia-iatp
-Version: 0.1.1
+Version: 0.1.2
 Summary: Inter-Agent Transfer Protocol (IATP) - Enable AI Agents to utilize other AI Agents as tools
 Project-URL: Documentation, https://pypi.org/project/traia-iatp
 Project-URL: Source, https://github.com/Traia-IO/IATP