uagents-core 0.3.4__tar.gz → 0.3.5__tar.gz

{uagents_core-0.3.4 → uagents_core-0.3.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: uagents-core
-Version: 0.3.4
+Version: 0.3.5
 Summary: Core components for agent based systems
 License: Apache 2.0
 Author: Ed FitzGerald

{uagents_core-0.3.4 → uagents_core-0.3.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "uagents-core"
-version = "0.3.4"
+version = "0.3.5"
 description = "Core components for agent based systems"
 authors = [
     { name = "Ed FitzGerald", email = "edward.fitzgerald@fetch.ai" },

{uagents_core-0.3.4 → uagents_core-0.3.5}/uagents_core/types.py

@@ -1,4 +1,5 @@
 import uuid
+from abc import ABC, abstractmethod
 from enum import Enum
 from typing import Any, Literal
 
@@ -48,3 +49,31 @@ class MsgStatus(BaseModel):
     destination: str
     endpoint: str
     session: uuid.UUID | None = None
+
+
+class Resolver(ABC):
+    @abstractmethod
+    async def resolve(self, destination: str) -> tuple[str | None, list[str]]:
+        """
+        Resolve the destination to an address and endpoint.
+
+        Args:
+            destination (str): The destination name or address to resolve.
+
+        Returns:
+            tuple[str | None, list[str]]: The address (if available) and resolved endpoints.
+        """
+        raise NotImplementedError
+
+    @abstractmethod
+    def sync_resolve(self, destination: str) -> list[str]:
+        """
+        Resolve the destination to a list of endpoints.
+
+        Args:
+            destination (str): The destination name or address to resolve.
+
+        Returns:
+            list[str]: The resolved endpoints.
+        """
+        raise NotImplementedError

{uagents_core-0.3.4 → uagents_core-0.3.5}/uagents_core/utils/messages.py

@@ -14,8 +14,8 @@ from uagents_core.helpers import weighted_random_sample
 from uagents_core.identity import Identity
 from uagents_core.logger import get_logger
 from uagents_core.models import Model
-from uagents_core.types import DeliveryStatus, MsgStatus
-from uagents_core.utils.resolver import lookup_endpoint_for_agent
+from uagents_core.types import DeliveryStatus, MsgStatus, Resolver
+from uagents_core.utils.resolver import AlmanacResolver
 
 logger = get_logger("uagents_core.utils.messages")
 
@@ -89,6 +89,7 @@ def send_message_to_agent(
     session_id: UUID | None = None,
     strategy: Literal["first", "random", "all"] = "first",
     agentverse_config: AgentverseConfig | None = None,
+    resolver: Resolver | None = None,
 ) -> list[MsgStatus]:
     """
     Send a message to an agent with default settings.
@@ -103,9 +104,12 @@ def send_message_to_agent(
         agentverse_config (AgentverseConfig, optional): The configuration for the agentverse.
     """
     agentverse_config = agentverse_config or AgentverseConfig()
-    endpoints = lookup_endpoint_for_agent(
-        agent_identifier=destination, agentverse_config=agentverse_config
-    )
+
+    if not resolver:
+        resolver = AlmanacResolver(
+            agentverse_config=agentverse_config,
+        )
+    endpoints = resolver.sync_resolve(destination)
     if not endpoints:
         logger.error("No endpoints found for agent", extra={"destination": destination})
         return []

{uagents_core-0.3.4 → uagents_core-0.3.5}/uagents_core/utils/resolver.py

@@ -14,6 +14,7 @@ from uagents_core.config import (
 from uagents_core.helpers import weighted_random_sample
 from uagents_core.identity import parse_identifier
 from uagents_core.logger import get_logger
+from uagents_core.types import Resolver
 
 logger = get_logger("uagents_core.utils.resolver")
 
@@ -71,3 +72,20 @@ def lookup_endpoint_for_agent(
         )
 
     return []
+
+
+class AlmanacResolver(Resolver):
+    def __init__(self, agentverse_config: AgentverseConfig | None = None):
+        self.agentverse_config = agentverse_config or AgentverseConfig()
+
+    async def resolve(self, destination: str) -> tuple[str | None, list[str]]:
+        endpoints = lookup_endpoint_for_agent(
+            agent_identifier=destination, agentverse_config=self.agentverse_config
+        )
+        return None, endpoints
+
+    def sync_resolve(self, destination: str) -> list[str]:
+        endpoints = lookup_endpoint_for_agent(
+            agent_identifier=destination, agentverse_config=self.agentverse_config
+        )
+        return endpoints