python-sdk/examples/echo_agent.py at package-scripts · openSUSE-Python/python-sdk · GitHub

History

86 lines (74 loc) · 2.18 KB

Raw

# /// script

# requires-python = ">=3.10,<3.15"

# dependencies = [

# "agent-client-protocol",

# ]

# ///

import asyncio

from typing import Any

from uuid import uuid4

from acp import (

Agent,

InitializeResponse,

NewSessionResponse,

PromptResponse,

run_agent,

text_block,

update_agent_message,

)

from acp.interfaces import Client

from acp.schema import (

AudioContentBlock,

ClientCapabilities,

EmbeddedResourceContentBlock,

HttpMcpServer,

ImageContentBlock,

Implementation,

McpServerStdio,

ResourceContentBlock,

SseMcpServer,

TextContentBlock,

)

class EchoAgent(Agent):

_conn: Client

def on_connect(self, conn: Client) -> None:

self._conn = conn

async def initialize(

self,

protocol_version: int,

client_capabilities: ClientCapabilities | None = None,

client_info: Implementation | None = None,

**kwargs: Any,

) -> InitializeResponse:

return InitializeResponse(protocol_version=protocol_version)

async def new_session(

self,

cwd: str,

additional_directories: list[str] | None = None,

mcp_servers: list[HttpMcpServer | SseMcpServer | McpServerStdio] | None = None,

**kwargs: Any,

) -> NewSessionResponse:

return NewSessionResponse(session_id=uuid4().hex)

async def prompt(

self,

session_id: str,

prompt: list[

TextContentBlock

| ImageContentBlock

| AudioContentBlock

| ResourceContentBlock

| EmbeddedResourceContentBlock

**kwargs: Any,

) -> PromptResponse:

for block in prompt:

text = block.get("text", "") if isinstance(block, dict) else getattr(block, "text", "")

chunk = update_agent_message(text_block(text))

chunk.field_meta = {"echo": True}

chunk.content.field_meta = {"echo": True}

await self._conn.session_update(session_id=session_id, update=chunk, source="echo_agent")

return PromptResponse(stop_reason="end_turn")

async def main() -> None:

await run_agent(EchoAgent())

if __name__ == "__main__":

asyncio.run(main())

python-sdk/examples/echo_agent.py at package-scripts · openSUSE-Python/python-sdk · GitHub

Uh oh!

Expand file tree

Breadcrumbs

echo_agent.py

Latest commit

History

Breadcrumbs

echo_agent.py

File metadata and controls