examples: mcp related

Author: pchalasani

examples/mcp/biomcp.py

@@ -22,8 +22,7 @@
 
 async def main(model: str = ""):
     transport = StdioTransport(
-        command="uv",
-        args=["run", "--with", "biomcp-python", "biomcp", "run"]
+        command="uv", args=["run", "--with", "biomcp-python", "biomcp", "run"]
     )
     all_tools = await get_langroid_tools_async(transport)
 

examples/mcp/gitmcp.py

@@ -0,0 +1,102 @@
+"""
+Simple example of using the GitMCP server to "chat" about a GitHub repository.
+
+https://github.com/idosal/git-mcp
+
+The server offers several tools, and we can enable ALL of them to be used
+by a Langroid agent.
+
+Run like this (-m model optional; defaults to gpt-4.1-mini):
+
+    uv run examples/mcp/gitmcp.py -m ollama/qwen2.5-coder:32b
+
+"""
+
+from textwrap import dedent
+from typing import List
+import langroid as lr
+import langroid.language_models as lm
+from langroid.mytypes import NonToolAction
+from langroid.agent.tools.mcp.fastmcp_client import get_langroid_tools_async
+from fastmcp.client.transports import SSETransport
+from fire import Fire
+
+
+def get_gitmcp_url() -> str:
+    from rich.prompt import Prompt
+    from rich.console import Console
+
+    console = Console()
+    import re
+
+    short_pattern = re.compile(r"^([^/]+)/([^/]+)$")
+    url_pattern = re.compile(
+        r"^(?:https?://)?(?:www\.)?github\.com/([^/]+)/([^/]+)(?:\.git)?/?$"
+    )
+
+    while True:
+        user_input = Prompt.ask(
+            "[bold blue]Enter the GitHub repository (owner/repo or full URL)"
+        ).strip()
+        m = short_pattern.match(user_input)
+        if m:
+            owner, repo = m.groups()
+        else:
+            m = url_pattern.match(user_input)
+            if m:
+                owner, repo = m.groups()
+            else:
+                console.print(
+                    "[red]Invalid format. Please enter 'owner/repo' or a full GitHub URL."
+                )
+                continue
+        break
+
+    github_url = f"https://github.com/{owner}/{repo}"
+    console.print(f"Full GitHub URL set to [green]{github_url}[/]")
+
+    gitmcp_url = f"https://gitmcp.io/{owner}/{repo}"
+    console.print(f"GitMCP URL set to [green]{gitmcp_url}[/]")
+    return gitmcp_url
+
+
+async def main(model: str = ""):
+
+    gitmcp_url = get_gitmcp_url()
+
+    transport = SSETransport(
+        url=gitmcp_url,
+    )
+    all_tools: List[lr.ToolMessage] = await get_langroid_tools_async(transport)
+
+    agent = lr.ChatAgent(
+        lr.ChatAgentConfig(
+            # forward to user when LLM doesn't use a tool
+            handle_llm_no_tool=NonToolAction.FORWARD_USER,
+            llm=lm.OpenAIGPTConfig(
+                chat_model=model or "gpt-4.1-mini",
+                max_output_tokens=100_000,
+                async_stream_quiet=False,
+            ),
+            system_message=dedent(
+                """
+                Make best use of any of the TOOLs available to you,
+                to answer the user's questions.
+                """
+            ),
+        )
+    )
+
+    # enable the agent to use all tools
+    agent.enable_message(all_tools)
+    # make task with interactive=False =>
+    # waits for user only when LLM doesn't use a tool
+    task = lr.Task(agent, interactive=False)
+    await task.run_async(
+        "Based on the TOOLs available to you, greet the user and"
+        "tell them what kinds of help you can provide."
+    )
+
+
+if __name__ == "__main__":
+    Fire(main)