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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: traia-iatp
-Version: 0.1.1
+Version: 0.1.3
 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.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "traia-iatp"
-version = "0.1.1"
+version = "0.1.3"
 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.3}/src/traia_iatp/cli/main.py

@@ -438,7 +438,7 @@ def example_crew():
 # Example: Using utility agencies in a CrewAI crew
 
 from crewai import Agent, Crew, Task
-from iatp import create_utility_agency_tools
+from traia_iatp import create_utility_agency_tools
 
 # Find and create tools from utility agencies
 tools = create_utility_agency_tools(

{traia_iatp-0.1.1 → traia_iatp-0.1.3}/src/traia_iatp/mcp/traia_mcp_adapter.py

@@ -236,7 +236,7 @@ all HTTP requests when using streamable-http transport.
 Basic Usage
 -----------
 ```python
-from iatp.mcp import TraiaMCPAdapter
+from traia_iatp.mcp import TraiaMCPAdapter
 
 # Standard connection (no authentication)
 server_params = {
@@ -264,7 +264,7 @@ with TraiaMCPAdapter(server_params) as tools:
 Using Helper Functions
 ----------------------
 ```python
-from iatp.mcp import create_mcp_adapter_with_auth
+from traia_iatp.mcp import create_mcp_adapter_with_auth
 
 # Create authenticated adapter
 adapter = create_mcp_adapter_with_auth(

{traia_iatp-0.1.1 → traia_iatp-0.1.3}/src/traia_iatp/registry/mongodb_registry.py

@@ -26,8 +26,8 @@ if __name__ == "__main__":
     # When running as a script, import directly
     import sys
     sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))))
-    from iatp.core.models import UtilityAgentRegistryEntry, UtilityAgent
-    from iatp.registry.embeddings import get_embedding_service
+    from src.traia_iatp.core.models import UtilityAgentRegistryEntry, UtilityAgent
+    from src.traia_iatp.registry.embeddings import get_embedding_service
     from ..utils import get_now_in_utc
 else:
     # When imported as a module
@@ -401,7 +401,7 @@ class UtilityAgentRegistry:
             return False
             
         # Create new endpoints based on the new base URL
-        from iatp.utils.iatp_utils import create_iatp_endpoints
+        from ..utils.iatp_utils import create_iatp_endpoints
         supports_streaming = doc.get("endpoints", {}).get("streaming_endpoint") is not None
         new_endpoints = create_iatp_endpoints(new_base_url, supports_streaming)
         
@@ -700,7 +700,7 @@ if __name__ == "__main__":
             )
             
             # Create test agent with endpoints
-            from iatp.utils.iatp_utils import create_iatp_endpoints
+            from ..utils.iatp_utils import create_iatp_endpoints
             test_agent.endpoints = create_iatp_endpoints("http://weather-agent:8100", supports_streaming=True)
             
             # Add to registry

{traia_iatp-0.1.1 → traia_iatp-0.1.3}/src/traia_iatp/registry/readmes/EMBEDDINGS_SETUP.md

@@ -116,7 +116,7 @@ import os
 os.environ["ENABLE_EMBEDDINGS"] = "true"
 os.environ["OPENAI_API_KEY"] = "your-key"
 
-from iatp.registry.mongodb_registry import UtilityAgencyRegistry
+from traia_iatp.registry.mongodb_registry import UtilityAgencyRegistry
 
 # Create registry (embeddings will be auto-generated)
 registry = UtilityAgencyRegistry()

{traia_iatp-0.1.1 → traia_iatp-0.1.3}/src/traia_iatp/registry/readmes/IATP_SEARCH_API_GUIDE.md

@@ -22,7 +22,7 @@ The API automatically detects and uses the appropriate authentication method:
 Find a single utility agent by specific criteria.
 
 ```python
-from iatp.registry.iatp_search_api import find_utility_agent
+from traia_iatp.registry.iatp_search_api import find_utility_agent
 
 # Find by exact name
 agent = find_utility_agent(name="hyperliquid-mcp-traia-utility-agency")
@@ -41,7 +41,7 @@ agent = find_utility_agent(query="trading bot")
 List utility agents with optional filters.
 
 ```python
-from iatp.registry.iatp_search_api import list_utility_agents
+from traia_iatp.registry.iatp_search_api import list_utility_agents
 
 # List all active agents
 agents = list_utility_agents(limit=20)
@@ -57,7 +57,7 @@ agents = list_utility_agents(capabilities=["market_info", "trading_orders"])
 Search utility agents using vector search.
 
 ```python
-from iatp.registry.iatp_search_api import search_utility_agents
+from traia_iatp.registry.iatp_search_api import search_utility_agents
 
 # Search using default search_text embedding (recommended for best performance)
 agents = await search_utility_agents("trading hyperliquid", limit=5)
@@ -83,7 +83,7 @@ agents = await search_utility_agents(
 Find a single MCP server by specific criteria.
 
 ```python
-from iatp.registry.iatp_search_api import find_mcp_server
+from traia_iatp.registry.iatp_search_api import find_mcp_server
 
 # Find by exact name
 server = find_mcp_server(name="hyperliquid-mcp")
@@ -96,7 +96,7 @@ server = find_mcp_server(capability="trading_orders")
 List MCP servers with optional filters.
 
 ```python
-from iatp.registry.iatp_search_api import list_mcp_servers
+from traia_iatp.registry.iatp_search_api import list_mcp_servers
 
 # List all servers
 servers = list_mcp_servers(limit=10)
@@ -109,7 +109,7 @@ servers = list_mcp_servers(capabilities=["market_info"])
 Search MCP servers using vector search.
 
 ```python
-from iatp.registry.iatp_search_api import search_mcp_servers
+from traia_iatp.registry.iatp_search_api import search_mcp_servers
 
 # Search using default embedding fields (description and capabilities)
 servers = await search_mcp_servers("trading", limit=5)
@@ -126,7 +126,7 @@ servers = await search_mcp_servers(
 Get detailed MCP server information by name (returns raw MongoDB document).
 
 ```python
-from iatp.registry.iatp_search_api import get_mcp_server
+from traia_iatp.registry.iatp_search_api import get_mcp_server
 
 # Get full server details
 server_doc = get_mcp_server("hyperliquid-mcp")
@@ -172,7 +172,7 @@ class MCPServerInfo:
 #!/usr/bin/env python
 import asyncio
 import os
-from iatp.registry.iatp_search_api import (
+from traia_iatp.registry.iatp_search_api import (
     find_utility_agent,
     list_utility_agents,
     search_utility_agents,

{traia_iatp-0.1.1 → traia_iatp-0.1.3}/src/traia_iatp/registry/readmes/MONGODB_X509_AUTH.md

@@ -132,7 +132,7 @@ openssl s_client -connect your-cluster.mongodb.net:27017 -CAfile cert.pem
 The registry automatically detects and uses X.509 authentication when configured:
 
 ```python
-from iatp.registry.mongodb_registry import UtilityAgentRegistry
+from traia_iatp.registry.mongodb_registry import UtilityAgentRegistry
 
 # No need to pass credentials - uses MONGODB_X509_CERT_FILE automatically
 registry = UtilityAgentRegistry()

{traia_iatp-0.1.1 → traia_iatp-0.1.3}/src/traia_iatp/registry/readmes/README.md

@@ -65,7 +65,7 @@ export MONGODB_CONNECTION_STRING=mongodb+srv://user:pass@cluster.mongodb.net/...
 ### Using the High-Level API (No MongoDB Required)
 
 ```python
-from iatp.registry.iatp_search_api import find_utility_agent, list_mcp_servers
+from traia_iatp.registry.iatp_search_api import find_utility_agent, list_mcp_servers
 
 # Find a specific utility agent
 agent = await find_utility_agent(name="trading-agent")
@@ -77,7 +77,7 @@ servers = list_mcp_servers(limit=10)
 ### Direct MongoDB Access
 
 ```python
-from iatp.registry.mongodb_registry import UtilityAgentRegistry
+from traia_iatp.registry.mongodb_registry import UtilityAgentRegistry
 
 # Registry will automatically use configured authentication
 registry = UtilityAgentRegistry()
@@ -168,7 +168,7 @@ See [atlas_search_indexes.json](./atlas_search_indexes.json) for index definitio
 ### Register a New Utility Agent
 
 ```python
-from iatp.core.models import UtilityAgent, AgentEndpoints
+from traia_iatp.core.models import UtilityAgent, AgentEndpoints
 
 agent = UtilityAgent(
     name="My Trading Bot",

{traia_iatp-0.1.1 → traia_iatp-0.1.3}/src/traia_iatp/registry/readmes/REFACTORING_SUMMARY.md

@@ -95,7 +95,7 @@ This refactoring separates the concerns between registry management (write opera
 
 ### Registry Operations (Write)
 ```python
-from iatp.registry.mongodb_registry import UtilityAgentRegistry
+from traia_iatp.registry.mongodb_registry import UtilityAgentRegistry
 
 # Add new agent
 registry = UtilityAgentRegistry()
@@ -108,7 +108,7 @@ await registry.update_agent_base_url(agent_id, new_url)
 
 ### Search Operations (Read)
 ```python
-from iatp.registry.iatp_search_api import search_utility_agents, find_utility_agent
+from traia_iatp.registry.iatp_search_api import search_utility_agents, find_utility_agent
 
 # Search for agents
 agents = await search_utility_agents("hyperliquid trading", limit=5)
@@ -120,7 +120,7 @@ agent = find_utility_agent(capability="market_data")
 ### Discovery Test
 ```python
 # The discovery test now uses the search API
-from iatp.registry.iatp_search_api import search_utility_agents
+from traia_iatp.registry.iatp_search_api import search_utility_agents
 
 # Search for Hyperliquid agent
 search_results = await search_utility_agents(
@@ -137,19 +137,19 @@ search_results = await search_utility_agents(
 1. **If using registry for search/query:**
    ```python
    # OLD
-   from iatp.registry.mongodb_registry import UtilityAgentRegistry
+   from traia_iatp.registry.mongodb_registry import UtilityAgentRegistry
    registry = UtilityAgentRegistry()
    agents = await registry.query_agents(query="trading")
    
    # NEW
-   from iatp.registry.iatp_search_api import search_utility_agents
+   from traia_iatp.registry.iatp_search_api import search_utility_agents
    agents = await search_utility_agents("trading")
    ```
 
 2. **If using registry for write operations:**
    ```python
    # No changes needed - these methods are still available
-   from iatp.registry.mongodb_registry import UtilityAgentRegistry
+   from traia_iatp.registry.mongodb_registry import UtilityAgentRegistry
    registry = UtilityAgentRegistry()
    await registry.add_utility_agent(agent)
    ```

{traia_iatp-0.1.1 → traia_iatp-0.1.3}/src/traia_iatp/server/templates/README.md

@@ -63,7 +63,7 @@ The templates expect the following variables:
 Use the `UtilityAgencyTemplateGenerator` class to generate agencies:
 
 ```python
-from iatp.server.template_generator import UtilityAgencyTemplateGenerator
+from traia_iatp.server.template_generator import UtilityAgencyTemplateGenerator
 
 generator = UtilityAgencyTemplateGenerator()
 output_path = generator.generate_agency(

{traia_iatp-0.1.1 → traia_iatp-0.1.3}/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.3}/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.3}/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.3}/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.3}/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.3}/src/traia_iatp/special_agencies/registry_search_agency.py

@@ -357,7 +357,7 @@ python -m uvicorn server:app --host 0.0.0.0 --port 8000
 ## Usage Example
 
 ```python
-from iatp import create_utility_agency_tools
+from traia_iatp import create_utility_agency_tools
 
 # This agency will be discoverable like any other
 tools = create_utility_agency_tools(

{traia_iatp-0.1.1 → traia_iatp-0.1.3}/src/traia_iatp/utils/iatp_utils.py

@@ -4,7 +4,7 @@ import logging
 from typing import Dict, Any, Optional, List
 import httpx
 
-from iatp.core.models import AgentCard, AgentSkill, AgentCapabilities, IATPEndpoints
+from ..core.models import AgentCard, AgentSkill, AgentCapabilities, IATPEndpoints
 
 logger = logging.getLogger(__name__)
 

{traia_iatp-0.1.1 → traia_iatp-0.1.3/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.3
 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