From 0ea05878a30c0df21ee083f68c72457e18f65c2e Mon Sep 17 00:00:00 2001
From: "Edward Z. Yang" <ezyang@mit.edu>
Date: Sun, 13 Apr 2025 22:13:51 +0800
Subject: [PATCH] Update

[ghstack-poisoned]
---
 codemcp/agno.py | 81 +++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 81 insertions(+)
 create mode 100644 codemcp/agno.py

diff --git a/codemcp/agno.py b/codemcp/agno.py
new file mode 100644
index 0000000..89a090a
--- /dev/null
+++ b/codemcp/agno.py
@@ -0,0 +1,81 @@
+import asyncio
+from typing import Union
+from urllib.parse import quote
+
+from agno.agent import Agent
+from agno.api.playground import PlaygroundEndpointCreate, create_playground_endpoint
+from agno.cli.console import console
+from agno.cli.settings import agno_cli_settings
+from agno.models.anthropic import Claude
+from agno.playground import Playground
+from agno.tools.mcp import MCPTools
+from agno.utils.log import logger
+from fastapi import FastAPI
+from rich import box
+from rich.panel import Panel
+
+
+async def serve_playground_app_async(
+    app: Union[str, FastAPI],
+    *,
+    scheme: str = "http",
+    host: str = "localhost",
+    port: int = 7777,
+    reload: bool = False,
+    prefix="/v1",
+    **kwargs,
+):
+    import uvicorn
+
+    try:
+        create_playground_endpoint(
+            playground=PlaygroundEndpointCreate(
+                endpoint=f"{scheme}://{host}:{port}", playground_data={"prefix": prefix}
+            ),
+        )
+    except Exception as e:
+        logger.error(f"Could not create playground endpoint: {e}")
+        logger.error("Please try again.")
+        return
+
+    logger.info(f"Starting playground on {scheme}://{host}:{port}")
+    # Encode the full endpoint (host:port)
+    encoded_endpoint = quote(f"{host}:{port}")
+
+    # Create a panel with the playground URL
+    url = f"{agno_cli_settings.playground_url}?endpoint={encoded_endpoint}"
+    panel = Panel(
+        f"[bold green]Playground URL:[/bold green] [link={url}]{url}[/link]",
+        title="Agent Playground",
+        expand=False,
+        border_style="cyan",
+        box=box.HEAVY,
+        padding=(2, 2),
+    )
+
+    # Print the panel
+    console.print(panel)
+
+    config = uvicorn.Config(app=app, host=host, port=port, reload=reload, **kwargs)
+    server = uvicorn.Server(config)
+    await server.serve()
+
+
+async def main():
+    async with MCPTools(
+        f"{sys.executable()} -m codemcp.hot_reload_entry"
+    ) as codemcp:
+        # TODO: cli-ify the model
+        agent = Agent(
+            model=Claude(id="claude-3-7-sonnet-20250219"),
+            tools=[codemcp],
+            instructions="",
+            markdown=True,
+            show_tool_calls=True,
+        )
+        playground = Playground(agents=[agent]).get_app()
+        await serve_playground_app_async(playground)
+
+
+if __name__ == "__main__":
+    asyncio.run(main())