traia-iatp 0.1.2__py3-none-any.whl → 0.1.67__py3-none-any.whl

traia_iatp/registry/mongodb_registry.py

@@ -26,13 +26,11 @@ 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 ..utils import get_now_in_utc
 else:
     # When imported as a module
     from ..core.models import UtilityAgentRegistryEntry, UtilityAgent
-    from .embeddings import get_embedding_service
     from ..utils import get_now_in_utc
 
 logger = logging.getLogger(__name__)
@@ -127,34 +125,51 @@ class UtilityAgentRegistry:
         if connection_string:
             self.connection_string = connection_string
         else:
-            # Try X.509 certificate authentication first
-            cert_file = os.getenv("MONGODB_X509_CERT_FILE")
-            if cert_file and os.path.exists(cert_file):
-                # For X.509 authentication, we need to extract the subject from the certificate
-                # to use as the username. MongoDB Atlas typically uses the full DN as username.
-                # The connection string format for X.509 is:
-                # mongodb+srv://cluster.mongodb.net/?authSource=$external&authMechanism=MONGODB-X509
-                # Extract just the cluster hostname without query parameters
-                cluster_host = CLUSTER_URI.split('?')[0]
-                self.connection_string = f"mongodb+srv://{cluster_host}?authSource=$external&authMechanism=MONGODB-X509&tls=true&tlsCertificateKeyFile={cert_file}"
-                logger.info(f"Using X.509 certificate authentication from {cert_file}")
+            # Check if running locally or in cloud environment
+            is_local = os.getenv("LOCAL_EXECUTION", "false").lower() == "true"
+            
+            if is_local:
+                # Local execution - require explicit credentials
+                # Try X.509 certificate authentication first
+                cert_file = os.getenv("MONGODB_X509_CERT_FILE")
+                if cert_file and os.path.exists(cert_file):
+                    # For X.509 authentication, we need to extract the subject from the certificate
+                    # to use as the username. MongoDB Atlas typically uses the full DN as username.
+                    # The connection string format for X.509 is:
+                    # mongodb+srv://cluster.mongodb.net/?authSource=$external&authMechanism=MONGODB-X509
+                    # Extract just the cluster hostname without query parameters
+                    cluster_host = CLUSTER_URI.split('?')[0]
+                    self.connection_string = f"mongodb+srv://{cluster_host}?authSource=$external&authMechanism=MONGODB-X509&tls=true&tlsCertificateKeyFile={cert_file}"
+                    logger.info(f"Using X.509 certificate authentication from {cert_file}")
+                else:
+                    # Fallback to username/password authentication
+                    user = os.getenv("MONGODB_USER")
+                    password = os.getenv("MONGODB_PASSWORD")
+                    if user and password:
+                        self.connection_string = f"mongodb+srv://{user}:{password}@{CLUSTER_URI}"
+                        logger.info("Using username/password authentication")
+                    else:
+                        # Try connection string as last resort
+                        self.connection_string = os.getenv("MONGODB_CONNECTION_STRING")
+                        if not self.connection_string:
+                            raise ValueError(
+                                "MongoDB authentication required for local execution. Please provide either:\n"
+                                "1. MONGODB_X509_CERT_FILE - Path to X.509 certificate file\n"
+                                "2. MONGODB_USER and MONGODB_PASSWORD - Username and password\n"
+                                "3. MONGODB_CONNECTION_STRING - Full connection string"
+                            )
             else:
-                # Fallback to username/password authentication
-                user = os.getenv("MONGODB_USER")
-                password = os.getenv("MONGODB_PASSWORD")
-                if user and password:
-                    self.connection_string = f"mongodb+srv://{user}:{password}@{CLUSTER_URI}"
-                    logger.info("Using username/password authentication")
+                # Cloud execution (Lambda) - use IAM role authentication
+                cluster_host = CLUSTER_URI.split('?')[0].rstrip('/')
+                # Preserve useful query parameters from original CLUSTER_URI
+                original_params = CLUSTER_URI.split('?')[1] if '?' in CLUSTER_URI else ""
+                # Combine AWS auth params with original params
+                auth_params = "authSource=$external&authMechanism=MONGODB-AWS"
+                if original_params:
+                    self.connection_string = f"mongodb+srv://{cluster_host}?{auth_params}&{original_params}"
                 else:
-                    # Try connection string as last resort
-                    self.connection_string = os.getenv("MONGODB_CONNECTION_STRING")
-                    if not self.connection_string:
-                        raise ValueError(
-                            "MongoDB authentication required. Please provide either:\n"
-                            "1. MONGODB_X509_CERT_FILE - Path to X.509 certificate file\n"
-                            "2. MONGODB_USER and MONGODB_PASSWORD - Username and password\n"
-                            "3. MONGODB_CONNECTION_STRING - Full connection string"
-                        )
+                    self.connection_string = f"mongodb+srv://{cluster_host}?{auth_params}"
+                logger.info("Using IAM role authentication for cloud execution")
         
         self.database_name = database_name
         self.client = _create_mongodb_client_with_retry(self.connection_string)
@@ -196,7 +211,7 @@ class UtilityAgentRegistry:
                 except Exception as e:
                     logger.warning(f"Could not create index {index_name}: {e}")
     
-    async def add_utility_agent(self, agent: UtilityAgent, tags: List[str] = None) -> UtilityAgentRegistryEntry:
+    async def add_utility_agent(self, agent: UtilityAgent, tags: List[str] = None, endpoints: List[Dict[str, Any]] = None) -> UtilityAgentRegistryEntry:
         """Add a utility agent to the cloud registry.
         
         Args:
@@ -225,6 +240,8 @@ class UtilityAgentRegistry:
         
         if os.getenv("ENABLE_EMBEDDINGS", "true").lower() == "true":
             try:
+                # Lazy import to avoid loading OpenAI/Cohere unless embeddings are enabled
+                from .embeddings import get_embedding_service
                 embedding_service = get_embedding_service()
                 
                 # Generate embedding for description
@@ -266,14 +283,19 @@ class UtilityAgentRegistry:
             except Exception as e:
                 logger.warning(f"Failed to generate embeddings: {e}. Proceeding without embeddings.")
         
-        # Create entry with enhanced fields
+        # Create entry with enhanced fields including d402 payment info
         entry = UtilityAgentRegistryEntry(
             agent_id=agent_id,
             name=agent.name,
             description=agent.description,
+            mcp_server_id=agent.mcp_server_id,  # MCP server reference
             capabilities=agent.capabilities,
             tags=tags or agent.tags,
-            metadata=agent.metadata
+            metadata=agent.metadata,
+            # D402 payment fields (must be at root level)
+            contract_address=agent.contract_address,
+            d402_enabled=agent.d402_enabled,
+            d402_payment_info=agent.d402_config  # d402_config becomes d402_payment_info in registry
         )
         
         # Add agent card if available
@@ -281,9 +303,8 @@ class UtilityAgentRegistry:
             entry.agent_card = agent.agent_card
             entry.skills = agent.agent_card.skills
         
-        # Add endpoints if available
+        # Add base_url if available
         if agent.endpoints:
-            entry.endpoints = agent.endpoints
             # Store base_url for indexing since all endpoints are derived from it
             entry.base_url = agent.endpoints.base_url
         
@@ -308,6 +329,8 @@ class UtilityAgentRegistry:
         # Generate embedding for the search text we just created
         if os.getenv("ENABLE_EMBEDDINGS", "true").lower() == "true" and entry.search_text:
             try:
+                # Lazy import to avoid loading OpenAI/Cohere unless embeddings are enabled
+                from .embeddings import get_embedding_service
                 embedding_service = get_embedding_service()
                 search_text_embedding = await embedding_service.generate_embedding(entry.search_text)
             except Exception as e:
@@ -317,6 +340,43 @@ class UtilityAgentRegistry:
         # Convert to dict and ensure all values are JSON-serializable
         entry_dict = entry.model_dump(mode='json')
         
+        # Add iatp_endpoints (protocol structure) and endpoints array (payment info) separately
+        if agent.endpoints:
+            # Store IATPEndpoints as iatp_endpoints (protocol info)
+            entry_dict["iatp_endpoints"] = agent.endpoints.model_dump(mode='json')
+            
+            # Store endpoints as array matching MCP server format (payment info)
+            # Use endpoints parameter if provided (includes actual payment configuration)
+            if endpoints and len(endpoints) > 0:
+                # Use endpoints array (includes actual payment configuration)
+                entry_dict["endpoints"] = endpoints
+                logger.info(f"Using {len(endpoints)} endpoints with payment configuration")
+            else:
+                # Fallback: Create endpoint from d402_config (for backward compatibility)
+                logger.info("Creating endpoint from d402_config (PostgreSQL endpoints not provided)")
+                endpoints_array = [
+                    {
+                        "endpoint_path": "/a2a",
+                        "endpoint_name": "a2a",
+                        "endpoint_method": "POST",
+                        "endpoint_description": "A2A protocol JSON-RPC endpoint",
+                        "icon_url": None,
+                        "endpoint_input_schema": None,  # A2A protocol defines this
+                        "endpoint_output_schema": None,  # A2A protocol defines this
+                        # Payment info from d402_config
+                        "payment_price_float": float(agent.d402_config.get("price_usd", 0.01)) if agent.d402_config else 0.01,
+                        "payment_price_wei": agent.d402_config.get("price_wei") if agent.d402_config else None,
+                        "settlement_token_address": agent.d402_config.get("token_address") if agent.d402_config else None,
+                        "settlement_token_symbol": agent.d402_config.get("token_symbol") if agent.d402_config else None,
+                        "settlement_token_network": agent.d402_config.get("network", "sepolia") if agent.d402_config else "sepolia",
+                        "settlement_token_chain_id": agent.d402_config.get("chain_id") if agent.d402_config else None
+                    }
+                ]
+                entry_dict["endpoints"] = endpoints_array
+        else:
+            entry_dict["iatp_endpoints"] = None
+            entry_dict["endpoints"] = []
+        
         # Add embeddings if available
         embeddings = {}
         if description_embedding:
@@ -333,9 +393,13 @@ class UtilityAgentRegistry:
         if embeddings:
             entry_dict["embeddings"] = embeddings
         
-        # Store the full UtilityAgent data as well
-        agent_dict = agent.model_dump(mode='json')
-        entry_dict["utility_agent_data"] = agent_dict
+        # Do NOT store utility_agent_data as a nested structure - all fields are already at root level
+        # The entry_dict from UtilityAgentRegistryEntry already contains all necessary fields:
+        # - agent_id, name, description, capabilities, tags, metadata (at root)
+        # - agent_card, skills, endpoints (at root)  
+        # - contract_address, d402_enabled, d402_payment_info (at root)
+        # - base_url, search_text, is_active, registered_at (at root)
+        # No utility_agent_data nesting is created
         
         result = self.registry.replace_one(
             {"agent_id": agent_id},  # Use agent_id as the key for upsert
@@ -401,7 +465,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)
         
@@ -456,6 +520,58 @@ class UtilityAgentRegistry:
             logger.warning(f"No utility agent found with ID {agent_id} or no changes made")
             return False
     
+    async def update_utility_agent_status(
+        self,
+        agent_id: str,
+        is_active: Optional[bool] = None,
+        core_tests_passed: Optional[bool] = None,
+        crewai_tests_passed: Optional[bool] = None
+    ) -> Dict[str, Any]:
+        """
+        Update utility agent status and test results in MongoDB.
+        
+        Similar to MCP server status updates, allows updating:
+        - is_active: Whether the agent is active and available
+        - core_tests_passed: Whether core (direct A2A) tests passed
+        - crewai_tests_passed: Whether CrewAI agency tests passed
+        
+        Args:
+            agent_id: ID of the utility agent
+            is_active: Optional new active status
+            core_tests_passed: Optional core test status
+            crewai_tests_passed: Optional CrewAI test status
+            
+        Returns:
+            Dict with success status and updated fields
+        """
+        update_data = {}
+        
+        if is_active is not None:
+            update_data["metadata.is_active"] = is_active
+        
+        if core_tests_passed is not None:
+            update_data["metadata.core_tests_passed"] = core_tests_passed
+        
+        if crewai_tests_passed is not None:
+            update_data["metadata.crewai_tests_passed"] = crewai_tests_passed
+        
+        if not update_data:
+            return {"success": False, "error": "No fields to update"}
+        
+        update_data["updated_at"] = get_now_in_utc()
+        
+        result = self.registry.update_one(
+            {"agent_id": agent_id},
+            {"$set": update_data}
+        )
+        
+        if result.modified_count > 0:
+            logger.info(f"Updated utility agent {agent_id} - {list(update_data.keys())}")
+            return {"success": True, "modified_count": result.modified_count, "fields": list(update_data.keys())}
+        else:
+            logger.warning(f"No documents modified for agent_id: {agent_id}")
+            return {"success": False, "error": "No documents modified"}
+    
     async def get_statistics(self) -> Dict[str, Any]:
         """Get registry statistics."""
         total_agents = self.registry.count_documents({})
@@ -499,34 +615,51 @@ class MCPServerRegistry:
         if connection_string:
             self.connection_string = connection_string
         else:
-            # Try X.509 certificate authentication first
-            cert_file = os.getenv("MONGODB_X509_CERT_FILE")
-            if cert_file and os.path.exists(cert_file):
-                # For X.509 authentication, we need to extract the subject from the certificate
-                # to use as the username. MongoDB Atlas typically uses the full DN as username.
-                # The connection string format for X.509 is:
-                # mongodb+srv://cluster.mongodb.net/?authSource=$external&authMechanism=MONGODB-X509
-                # Extract just the cluster hostname without query parameters
-                cluster_host = CLUSTER_URI.split('?')[0]
-                self.connection_string = f"mongodb+srv://{cluster_host}?authSource=$external&authMechanism=MONGODB-X509&tls=true&tlsCertificateKeyFile={cert_file}"
-                logger.info(f"Using X.509 certificate authentication from {cert_file}")
+            # Check if running locally or in cloud environment
+            is_local = os.getenv("LOCAL_EXECUTION", "false").lower() == "true"
+            
+            if is_local:
+                # Local execution - require explicit credentials
+                # Try X.509 certificate authentication first
+                cert_file = os.getenv("MONGODB_X509_CERT_FILE")
+                if cert_file and os.path.exists(cert_file):
+                    # For X.509 authentication, we need to extract the subject from the certificate
+                    # to use as the username. MongoDB Atlas typically uses the full DN as username.
+                    # The connection string format for X.509 is:
+                    # mongodb+srv://cluster.mongodb.net/?authSource=$external&authMechanism=MONGODB-X509
+                    # Extract just the cluster hostname without query parameters
+                    cluster_host = CLUSTER_URI.split('?')[0]
+                    self.connection_string = f"mongodb+srv://{cluster_host}?authSource=$external&authMechanism=MONGODB-X509&tls=true&tlsCertificateKeyFile={cert_file}"
+                    logger.info(f"Using X.509 certificate authentication from {cert_file}")
+                else:
+                    # Fallback to username/password authentication
+                    user = os.getenv("MONGODB_USER")
+                    password = os.getenv("MONGODB_PASSWORD")
+                    if user and password:
+                        self.connection_string = f"mongodb+srv://{user}:{password}@{CLUSTER_URI}"
+                        logger.info("Using username/password authentication")
+                    else:
+                        # Try connection string as last resort
+                        self.connection_string = os.getenv("MONGODB_CONNECTION_STRING")
+                        if not self.connection_string:
+                            raise ValueError(
+                                "MongoDB authentication required for local execution. Please provide either:\n"
+                                "1. MONGODB_X509_CERT_FILE - Path to X.509 certificate file\n"
+                                "2. MONGODB_USER and MONGODB_PASSWORD - Username and password\n"
+                                "3. MONGODB_CONNECTION_STRING - Full connection string"
+                            )
             else:
-                # Fallback to username/password authentication
-                user = os.getenv("MONGODB_USER")
-                password = os.getenv("MONGODB_PASSWORD")
-                if user and password:
-                    self.connection_string = f"mongodb+srv://{user}:{password}@{CLUSTER_URI}"
-                    logger.info("Using username/password authentication")
+                # Cloud execution (Lambda) - use IAM role authentication
+                cluster_host = CLUSTER_URI.split('?')[0].rstrip('/')
+                # Preserve useful query parameters from original CLUSTER_URI
+                original_params = CLUSTER_URI.split('?')[1] if '?' in CLUSTER_URI else ""
+                # Combine AWS auth params with original params
+                auth_params = "authSource=$external&authMechanism=MONGODB-AWS"
+                if original_params:
+                    self.connection_string = f"mongodb+srv://{cluster_host}?{auth_params}&{original_params}"
                 else:
-                    # Try connection string as last resort
-                    self.connection_string = os.getenv("MONGODB_CONNECTION_STRING")
-                    if not self.connection_string:
-                        raise ValueError(
-                            "MongoDB authentication required. Please provide either:\n"
-                            "1. MONGODB_X509_CERT_FILE - Path to X.509 certificate file\n"
-                            "2. MONGODB_USER and MONGODB_PASSWORD - Username and password\n"
-                            "3. MONGODB_CONNECTION_STRING - Full connection string"
-                        )
+                    self.connection_string = f"mongodb+srv://{cluster_host}?{auth_params}"
+                logger.info("Using IAM role authentication for cloud execution")
         
         self.client = _create_mongodb_client_with_retry(self.connection_string)
         self.db: Database = self.client[database_name]
@@ -557,7 +690,16 @@ class MCPServerRegistry:
         description: str,
         server_type: str = "streamable-http",
         capabilities: List[str] = None,
-        metadata: Dict[str, Any] = None
+        metadata: Dict[str, Any] = None,
+        core_tests_passed: bool = False,
+        crewai_tests_passed: bool = False,
+        endpoints: Optional[List[Dict[str, Any]]] = None,
+        user_uuid: Optional[str] = None,
+        operator_address: Optional[str] = None,
+        server_address: Optional[str] = None,
+        server_network: Optional[str] = None,
+        server_chain_id: Optional[int] = None,
+        accepts: Optional[Dict[str, Any]] = None
     ) -> str:
         """Add an MCP server to the registry with retry logic."""
         # Generate embeddings if enabled
@@ -566,6 +708,8 @@ class MCPServerRegistry:
         
         if os.getenv("ENABLE_EMBEDDINGS", "true").lower() == "true":
             try:
+                # Lazy import to avoid loading OpenAI/Cohere unless embeddings are enabled
+                from .embeddings import get_embedding_service
                 embedding_service = get_embedding_service()
                 
                 # Generate embedding for description
@@ -587,11 +731,29 @@ class MCPServerRegistry:
             "description": description,
             "server_type": server_type,
             "capabilities": capabilities or [],
+            "endpoints": endpoints or [],
             "metadata": metadata or {},
             "registered_at": get_now_in_utc(),
-            "is_active": True
+            "core_tests_passed": core_tests_passed,
+            "crewai_tests_passed": crewai_tests_passed,
+            "is_active": False,
+            "user_uuid": user_uuid
         }
         
+        # Add blockchain fields if provided (from PostgreSQL database)
+        if operator_address:
+            doc["operator_address"] = operator_address
+        if server_address:
+            doc["server_address"] = server_address
+        if server_network:
+            doc["server_network"] = server_network
+        if server_chain_id:
+            doc["server_chain_id"] = server_chain_id
+        
+        # Add token acceptance information if provided
+        if accepts:
+            doc["accepts"] = accepts
+        
         # Add embeddings if available
         if description_embedding:
             doc["description_embedding"] = description_embedding
@@ -638,8 +800,18 @@ class MCPServerRegistry:
 
     
     async def get_mcp_server(self, name: str) -> Optional[Dict[str, Any]]:
-        """Get an MCP server by name."""
-        doc = self.collection.find_one({"name": name})
+        """Get an MCP server by name.
+        
+        Note: Excludes embedding fields to reduce response size and improve performance.
+        Embeddings are only used internally for vector search.
+        """
+        # Exclude embedding fields from response (large arrays not needed by clients)
+        projection = {
+            "description_embedding": 0,
+            "capabilities_embedding": 0
+        }
+        
+        doc = self.collection.find_one({"name": name}, projection)
         if doc:
             doc["_id"] = str(doc["_id"])
             return doc
@@ -700,7 +872,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/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/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/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/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/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/scripts/__init__.py

@@ -0,0 +1,2 @@
+"""IATP utility scripts."""
+