meshagent-agents 0.8.0__tar.gz → 0.12.0__tar.gz

meshagent_agents-0.12.0/CHANGELOG.md

@@ -0,0 +1,44 @@
+## [0.12.0]
+- Reduce worker-queue logging verbosity to avoid logging full message payloads
+
+## [0.11.0]
+- Stability
+
+## [0.10.1]
+- Stability
+
+## [0.10.0]
+- Stability
+
+## [0.9.3]
+- Stability
+
+## [0.9.2]
+- Stability
+
+## [0.9.1]
+- Stability
+
+## [0.9.0]
+- Stability
+
+## [0.8.4]
+- Stability
+
+## [0.8.3]
+- Stability
+
+## [0.8.2]
+- Stability
+
+## [0.8.1]
+- Stability
+
+## [0.8.0]
+- Stability
+
+## [0.7.2]
+- Stability
+
+## [0.7.1]
+- Stability

{meshagent_agents-0.8.0/meshagent_agents.egg-info → meshagent_agents-0.12.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: meshagent-agents
-Version: 0.8.0
+Version: 0.12.0
 Summary: Agent Building Blocks for Meshagent
 License-Expression: Apache-2.0
 Project-URL: Documentation, https://docs.meshagent.com
@@ -12,19 +12,20 @@ License-File: LICENSE
 Requires-Dist: pyjwt~=2.10
 Requires-Dist: pytest~=8.4
 Requires-Dist: pytest-asyncio~=0.26
-Requires-Dist: meshagent-api~=0.8.0
-Requires-Dist: meshagent-tools~=0.8.0
-Requires-Dist: meshagent-openai~=0.8.0
+Requires-Dist: meshagent-api~=0.12.0
+Requires-Dist: meshagent-tools~=0.12.0
+Requires-Dist: meshagent-openai~=0.12.0
 Requires-Dist: pydantic~=2.11
 Requires-Dist: opentelemetry-distro~=0.54b1
+Requires-Dist: mistune~=3.1.4
 Provides-Extra: all
-Requires-Dist: meshagent-api[all]~=0.8.0; extra == "all"
+Requires-Dist: meshagent-api[all]~=0.12.0; extra == "all"
 Requires-Dist: chonkie~=0.5.1; extra == "all"
 Requires-Dist: chonkie[semantic]~=0.5.1; extra == "all"
 Requires-Dist: chonkie[openai]~=0.5.1; extra == "all"
 Requires-Dist: aiosmtplib~=4.0.1; extra == "all"
 Provides-Extra: sync
-Requires-Dist: meshagent-api[sync]~=0.8.0; extra == "sync"
+Requires-Dist: meshagent-api[sync]~=0.12.0; extra == "sync"
 Provides-Extra: mail
 Requires-Dist: aiosmtplib~=4.0.1; extra == "mail"
 Provides-Extra: rag

meshagent_agents-0.12.0/meshagent/agents/adapter.py

@@ -0,0 +1,147 @@
+from abc import ABC, abstractmethod
+from .agent import AgentChatContext
+from jsonschema import validate
+from meshagent.tools import Response, Toolkit, ToolkitBuilder, ToolkitConfig
+from meshagent.api import RoomClient, RoomException, RemoteParticipant
+from typing import Any, Optional, Callable, TypeVar, Generic, Literal
+
+TEvent = TypeVar("T")
+
+
+class ToolResponseAdapter(ABC):
+    def __init__(self):
+        pass
+
+    @abstractmethod
+    async def to_plain_text(self, *, room: RoomClient, response: Response):
+        pass
+
+    @abstractmethod
+    async def create_messages(
+        self,
+        *,
+        context: AgentChatContext,
+        tool_call: Any,
+        room: RoomClient,
+        response: Response,
+    ) -> list:
+        pass
+
+
+class LLMAdapter(Generic[TEvent]):
+    @abstractmethod
+    def default_model(self) -> str: ...
+
+    def create_chat_context(self) -> AgentChatContext:
+        return AgentChatContext()
+
+    async def check_for_termination(
+        self, *, context: AgentChatContext, room: RoomClient
+    ):
+        return True
+
+    def tool_providers(self, *, model: str) -> list[ToolkitBuilder]:
+        return []
+
+    def make_toolkit(self, *, model: str, config: ToolkitConfig) -> Toolkit:
+        for tool in self.tool_providers(model=model):
+            if tool.name == config.name:
+                return Toolkit(
+                    name=config.name, tools=[tool.make(model=model, config=config)]
+                )
+
+        raise RoomException(f"Unexpected tool: {config.name} for model {model}")
+
+    @abstractmethod
+    async def next(
+        self,
+        *,
+        context: AgentChatContext,
+        room: RoomClient,
+        toolkits: list[Toolkit],
+        tool_adapter: Optional[ToolResponseAdapter] = None,
+        output_schema: Optional[dict] = None,
+        event_handler: Optional[Callable[[TEvent], None]] = None,
+        model: Optional[str] = None,
+        on_behalf_of: Optional[RemoteParticipant] = None,
+    ) -> Any:
+        pass
+
+    def validate(response: dict, output_schema: dict):
+        validate(response, output_schema)
+
+
+class MessageStreamLLMAdapter(LLMAdapter):
+    def __init__(
+        self, *, participant_name: str, context_mode: Literal["diff", "full"] = "diff"
+    ):
+        self.participant_name = participant_name
+        self.context_mode = context_mode
+
+    def default_model(self) -> str:
+        return "toolkit"
+
+    def create_chat_context(self) -> AgentChatContext:
+        return AgentChatContext()
+
+    async def check_for_termination(
+        self, *, context: AgentChatContext, room: RoomClient
+    ):
+        return True
+
+    async def next(
+        self,
+        *,
+        context: AgentChatContext,
+        room: RoomClient,
+        toolkits: list[Toolkit],
+        tool_adapter: Optional[ToolResponseAdapter] = None,
+        output_schema: Optional[dict] = None,
+        event_handler: Optional[Callable[[TEvent], None]] = None,
+        model: Optional[str] = None,
+        on_behalf_of: Optional[RemoteParticipant] = None,
+    ) -> Any:
+        participant = room.messaging.get_participant_by_name(self.participant_name)
+        if participant is None:
+            raise RoomException("participant is not currently connected")
+
+        stream = await room.messaging.create_stream(
+            to=participant,
+            header={
+                "context": context.to_json(),
+                "model": model,
+                "output_schema": output_schema,
+                "on_behalf_of_id": on_behalf_of.id if on_behalf_of else None,
+            },
+        )
+
+        error = None
+        output = None
+        try:
+            async for chunk in stream.read_chunks():
+                event = chunk.header.get("event")
+                if event is not None and event_handler is not None:
+                    event_handler(event)
+
+                output = chunk.header.get("output")
+                if output is not None:
+                    output.append(output)
+
+                if chunk.header.get("done"):
+                    break
+
+        except Exception as ex:
+            error = ex
+
+        await stream.close()
+
+        if self.context_mode == "diff":
+            context.messages.clear()
+
+        if error:
+            raise error
+
+        return output
+
+    def validate(response: dict, output_schema: dict):
+        validate(response, output_schema)

{meshagent_agents-0.8.0 → meshagent_agents-0.12.0}/meshagent/agents/agent.py

@@ -343,6 +343,7 @@ class TaskRunner(SingleRoomAgent):
         output_schema: Optional[dict] = None,
         labels: Optional[list[str]] = None,
         toolkits: Optional[list[Toolkit]] = None,
+        annotations: Optional[list[str]] = None,
     ):
         super().__init__(
             name=name,
@@ -370,6 +371,7 @@ class TaskRunner(SingleRoomAgent):
         self._supports_tools = supports_tools
         self._input_schema = input_schema
         self._output_schema = output_schema
+        self._annotations = annotations
 
     async def validate_arguments(self, arguments: dict):
         validate(arguments, self.input_schema)
@@ -385,6 +387,10 @@ class TaskRunner(SingleRoomAgent):
     def supports_tools(self):
         return self._supports_tools
 
+    @property
+    def annotations(self):
+        return self._annotations
+
     @property
     def input_schema(self):
         return self._input_schema
@@ -403,6 +409,7 @@ class TaskRunner(SingleRoomAgent):
             "requires": list(map(lambda x: x.to_json(), self.requires)),
             "supports_tools": self.supports_tools,
             "labels": self.labels,
+            "annotations": self.annotations,
         }
 
     async def _register(self):
@@ -418,6 +425,7 @@ class TaskRunner(SingleRoomAgent):
                     "requires": list(map(lambda x: x.to_json(), self.requires)),
                     "supports_tools": self.supports_tools,
                     "labels": self.labels,
+                    "annotations": self.annotations,
                 },
             )
         )["id"]