uagents-core 0.3.4__py3-none-any.whl → 0.3.5__py3-none-any.whl

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/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/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

{uagents_core-0.3.4.dist-info → uagents_core-0.3.5.dist-info}/METADATA

@@ -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.dist-info → uagents_core-0.3.5.dist-info}/RECORD

@@ -12,12 +12,12 @@ uagents_core/models.py,sha256=fxsFjRochkJUdgPltT1HWgBl-9K-HVQWl8sSMVgeJFY,1066
 uagents_core/protocol.py,sha256=T9jasOkltne33E16Y7VrqcB2moWVsv-Qh4XLZotyz8g,5428
 uagents_core/registration.py,sha256=lmDXnsAs2CvpgE1Ik0qja_0RAY7zJruK11_jUrlqlWs,3226
 uagents_core/storage.py,sha256=VLqMSFXOspzrlbUNvaqS95ht9oJze1c-p4CD6Ul80h4,5150
-uagents_core/types.py,sha256=_W3EN1wEIRFxuhhBxyZxQH_dA_3AtoPPReIzrgcTUHc,1167
+uagents_core/types.py,sha256=nPK2ebRL6S-vLZx2AR3OO2RTEb_N2Yno6PqL8KHkeo4,1993
 uagents_core/utils/__init__.py,sha256=v0MaxDYCTtQlwbblEHCfLtbeTnA2hCmKKJk7mlcE20U,135
-uagents_core/utils/messages.py,sha256=5o3ahVaeGwzWMAxDs0fl2pC1duDq2VS_-czoVtZpkOc,5066
+uagents_core/utils/messages.py,sha256=x_99pLzMjKKuWrpSAAS9mtVUAjFEZco64mMmHJdtxEI,5149
 uagents_core/utils/registration.py,sha256=Lp17tn4knw4qpKx_mZO5gDHD6pGssUhux6SGDhhb7rc,14607
-uagents_core/utils/resolver.py,sha256=X18oe-WPU_6pfTc6x_Oa69kZQYCMidkaSeCs68Poaik,2169
+uagents_core/utils/resolver.py,sha256=zcHKkTWxuIHq1kR89RDocruGs3i1S2baQpg4TK4j8QY,2868
 uagents_core/utils/subscriptions.py,sha256=l29elYqKycgU3s2SN_oh-svfhDkqZce0-EIiv4jjxJw,2450
-uagents_core-0.3.4.dist-info/METADATA,sha256=ETB4mlX4APkU5KarF2ivzpT7C9I0xmsfXV_37DYYZrY,1009
-uagents_core-0.3.4.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-uagents_core-0.3.4.dist-info/RECORD,,
+uagents_core-0.3.5.dist-info/METADATA,sha256=UbvgotVaLQXTyBJUWwW3dmjs6inQO2J7iPz2ybU2Amw,1009
+uagents_core-0.3.5.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+uagents_core-0.3.5.dist-info/RECORD,,