fast-agent-mcp 0.2.50__py3-none-any.whl → 0.2.51__py3-none-any.whl

mcp_agent/resources/examples/tensorzero/docker-compose.yml

@@ -0,0 +1,105 @@
+# This is a simplified example for learning purposes. Do not use this in production.
+# For production-ready deployments, see: https://www.tensorzero.com/docs/gateway/deployment
+
+# Top-level volumes definition
+volumes:
+  minio_data: {}
+
+services:
+  clickhouse:
+    image: clickhouse/clickhouse-server:24.12-alpine
+    environment:
+      CLICKHOUSE_USER: chuser
+      CLICKHOUSE_DEFAULT_ACCESS_MANAGEMENT: 1
+      CLICKHOUSE_PASSWORD: chpassword
+    ports:
+      - "8123:8123"
+    healthcheck:
+      test: wget --spider --tries 1 http://chuser:chpassword@clickhouse:8123/ping
+      start_period: 30s
+      start_interval: 1s
+      timeout: 1s
+
+  gateway:
+    image: tensorzero/gateway
+    volumes:
+      - ./tensorzero_config:/app/config:ro
+    env_file:
+      - ./.env
+    command: --config-file /app/config/tensorzero.toml
+    environment: # WARNING: Insecure default credentials for local testing ONLY. Don't send this to production.
+      TENSORZERO_CLICKHOUSE_URL: http://chuser:chpassword@clickhouse:8123/tensorzero
+      S3_ACCESS_KEY_ID: user
+      S3_SECRET_ACCESS_KEY: password
+    ports:
+      - "3000:3000"
+    extra_hosts:
+      - "host.docker.internal:host-gateway"
+    depends_on:
+      clickhouse:
+        condition: service_healthy
+      minio:
+        condition: service_healthy
+      mcp-server:
+        condition: service_healthy
+
+  gateway-ui:
+    image: tensorzero/ui
+    volumes:
+      - ./tensorzero_config:/app/config:ro
+    env_file:
+      - ./.env
+    command: --config-file /app/config/tensorzero.toml
+    environment:
+      TENSORZERO_CLICKHOUSE_URL: http://chuser:chpassword@clickhouse:8123/tensorzero
+      TENSORZERO_GATEWAY_URL: http://gateway:3000
+      S3_ACCESS_KEY_ID: user
+      S3_SECRET_ACCESS_KEY: password
+    ports:
+      - "4000:4000"
+    depends_on:
+      clickhouse:
+        condition: service_healthy
+
+  mcp-server:
+    build:
+      context: ./mcp_server
+      dockerfile: Dockerfile
+    working_dir: /app
+    volumes:
+      - ./mcp_server:/app
+    ports:
+      - "8000:8000"
+    depends_on:
+      minio:
+        condition: service_healthy
+    healthcheck:
+      test:
+        [
+          "CMD",
+          "wget",
+          "--spider",
+          "--tries=1",
+          "http://localhost:8000/t0-example-server/sse",
+        ]
+      interval: 10s
+      timeout: 5s
+      retries: 12
+      start_period: 20s
+
+  minio:
+    image: minio/minio:latest
+    ports:
+      - "9000:9000" # API port
+      - "9001:9001" # Console port
+    volumes:
+      - minio_data:/data
+    environment: # WARNING: Insecure default credentials for local testing ONLY.
+      MINIO_ROOT_USER: user
+      MINIO_ROOT_PASSWORD: password
+    command: server /data --console-address :9001
+    healthcheck:
+      test: ["CMD", "curl", "-f", "http://localhost:9000/minio/health/live"]
+      interval: 30s
+      timeout: 20s
+      retries: 3

mcp_agent/resources/examples/tensorzero/fastagent.config.yaml

@@ -0,0 +1,19 @@
+default_model: haiku
+
+tensorzero:
+  base_url: http://localhost:3000
+  api_key: verysecret
+
+logger:
+  level: "info"
+  progress_display: true
+  show_chat: true
+  show_tools: true
+  truncate_tools: true
+
+mcp:
+  servers:
+    tester:
+      transport: "sse"
+      url: "http://localhost:8000/t0-example-server/sse"
+      read_transport_sse_timeout_seconds: 300

mcp_agent/resources/examples/tensorzero/image_demo.py

@@ -0,0 +1,67 @@
+import asyncio
+import base64
+import mimetypes
+from pathlib import Path
+from typing import List, Union
+
+from mcp.types import ImageContent, TextContent
+
+from mcp_agent.core.fastagent import FastAgent
+from mcp_agent.core.prompt import Prompt
+from mcp_agent.core.request_params import RequestParams
+
+AGENT_NAME = "tensorzero_image_tester"
+TENSORZERO_MODEL = "tensorzero.test_chat"
+TEXT_PROMPT = (
+    "Provide a description of the similarities and differences between these three images."
+)
+LOCAL_IMAGE_FILES = [
+    Path("./demo_images/clam.jpg"),
+    Path("./demo_images/shrimp.png"),
+    Path("./demo_images/crab.png"),
+]
+
+MY_T0_SYSTEM_VARS = {
+    "TEST_VARIABLE_1": "Roses are red",
+    "TEST_VARIABLE_2": "Violets are blue",
+    "TEST_VARIABLE_3": "Sugar is sweet",
+    "TEST_VARIABLE_4": "Vibe code responsibly 👍",
+}
+
+fast = FastAgent("TensorZero Image Demo - Base64 Only")
+
+
+@fast.agent(
+    name=AGENT_NAME,
+    model=TENSORZERO_MODEL,
+    request_params=RequestParams(template_vars=MY_T0_SYSTEM_VARS),
+)
+async def main():
+    content_parts: List[Union[TextContent, ImageContent]] = []
+    content_parts.append(TextContent(type="text", text=TEXT_PROMPT))
+
+    for file_path in LOCAL_IMAGE_FILES:
+        mime_type, _ = mimetypes.guess_type(file_path)
+        if not mime_type or not mime_type.startswith("image/"):
+            ext = file_path.suffix.lower()
+            if ext == ".jpg" or ext == ".jpeg":
+                mime_type = "image/jpeg"
+            elif ext == ".png":
+                mime_type = "image/png"
+        if mime_type is None:
+            mime_type = "image/png"  # Default fallback if still None
+
+        with open(file_path, "rb") as image_file:
+            image_bytes = image_file.read()
+
+        encoded_data = base64.b64encode(image_bytes).decode("utf-8")
+        content_parts.append(ImageContent(type="image", mimeType=mime_type, data=encoded_data))
+
+    message = Prompt.user(*content_parts)
+    async with fast.run() as agent_app:
+        agent = getattr(agent_app, AGENT_NAME)
+        await agent.send(message)
+
+
+if __name__ == "__main__":
+    asyncio.run(main())  # type: ignore

mcp_agent/resources/examples/tensorzero/mcp_server/Dockerfile

@@ -0,0 +1,25 @@
+FROM python:3.12-slim
+WORKDIR /app
+
+RUN apt-get update && apt-get install -y curl wget && \
+    wget https://dl.min.io/client/mc/release/linux-amd64/mc -O /usr/local/bin/mc && \
+    chmod +x /usr/local/bin/mc && \
+    apt-get clean && rm -rf /var/lib/apt/lists/*
+
+RUN pip install uv
+
+# Copy dependency files only
+COPY pyproject.toml /tmp/
+#COPY uv.lock /tmp/
+
+# Install dependencies
+WORKDIR /tmp
+RUN uv pip install --system .
+
+# Switch back to /app (this will be overridden by volume mount)
+WORKDIR /app
+
+# These files will come from the volume mount at runtime
+EXPOSE 8000
+ENTRYPOINT ["/app/entrypoint.sh"]
+CMD ["uvicorn", "mcp_server:app", "--host", "0.0.0.0", "--port", "8000"]

mcp_agent/resources/examples/tensorzero/mcp_server/entrypoint.sh

@@ -0,0 +1,35 @@
+#!/bin/sh
+
+echo "Entrypoint: Waiting for MinIO to be healthy..."
+
+# Simple loop to check MinIO health endpoint (within the Docker network)
+# Adjust timeout as needed
+TIMEOUT=60
+START_TIME=$(date +%s)
+while ! curl -sf http://minio:9000/minio/health/live > /dev/null; do
+    CURRENT_TIME=$(date +%s)
+    ELAPSED=$(($CURRENT_TIME - $START_TIME))
+    if [ $ELAPSED -ge $TIMEOUT ]; then
+        echo "Entrypoint: Timeout waiting for MinIO!"
+        exit 1
+    fi
+    echo "Entrypoint: MinIO not ready, sleeping..."
+    sleep 2
+done
+echo "Entrypoint: MinIO is healthy."
+
+echo "Entrypoint: Configuring mc client and creating bucket 'tensorzero'..."
+
+# Configure mc to talk to the MinIO server using the service name
+# Use --insecure because we are using http
+mc --insecure alias set local http://minio:9000 user password
+
+# Create the bucket if it doesn't exist
+# Use --insecure because we are using http
+mc --insecure ls local/tensorzero > /dev/null 2>&1 || mc --insecure mb local/tensorzero
+
+echo "Entrypoint: Bucket 'tensorzero' check/creation complete."
+
+echo "Entrypoint: Executing the main container command: $@"
+
+exec "$@"

mcp_agent/resources/examples/tensorzero/mcp_server/mcp_server.py

@@ -0,0 +1,31 @@
+import uvicorn
+from mcp.server.fastmcp.server import FastMCP
+from starlette.applications import Starlette
+from starlette.routing import Mount
+
+SERVER_PATH = "t0-example-server"
+
+
+mcp_instance = FastMCP(name="t0-example-server")
+mcp_instance.settings.message_path = f"/{SERVER_PATH}/messages/"
+mcp_instance.settings.sse_path = f"/{SERVER_PATH}/sse"
+
+
+@mcp_instance.tool()
+def example_tool(input_text: str) -> str:
+    """Example tool that reverses the text of a given string."""
+    reversed_text = input_text[::-1]
+    return reversed_text
+
+
+app = Starlette(
+    routes=[
+        Mount("/", app=mcp_instance.sse_app()),
+    ]
+)
+
+if __name__ == "__main__":
+    print(f"Starting minimal MCP server ({mcp_instance.name}) on http://127.0.0.1:8000")
+    print(f" -> SSE endpoint: {mcp_instance.settings.sse_path}")
+    print(f" -> Message endpoint: {mcp_instance.settings.message_path}")
+    uvicorn.run(app, host="127.0.0.1", port=8000)

mcp_agent/resources/examples/tensorzero/mcp_server/pyproject.toml

@@ -0,0 +1,11 @@
+[project]
+name = "mcp-server"
+version = "0.1.0"
+description = "Add your description here"
+readme = "README.md"
+requires-python = ">=3.12"
+dependencies = [
+    "fastmcp>=2.10.6",
+    "starlette>=0.46.2",
+    "uvicorn>=0.34.3",
+]

mcp_agent/resources/examples/tensorzero/simple_agent.py

@@ -0,0 +1,25 @@
+import asyncio
+
+from mcp_agent.core.fastagent import FastAgent
+
+CONFIG_FILE = "fastagent.config.yaml"
+fast = FastAgent("fast-agent example", config_path=CONFIG_FILE, ignore_unknown_args=True)
+
+
+@fast.agent(
+    name="default",
+    instruction="""
+        You are an agent dedicated to helping developers understand the relationship between TensoZero and fast-agent. If the user makes a request 
+        that requires you to invoke the test tools, please do so. When you use the tool, describe your rationale for doing so. 
+    """,
+    servers=["tester"],
+)
+async def main():
+    async with fast.run() as agent_app:
+        agent_name = "default"
+        print("\nStarting interactive session with template_vars set via decorator...")
+        await agent_app.interactive(agent=agent_name)
+
+
+if __name__ == "__main__":
+    asyncio.run(main())  # type: ignore

mcp_agent/resources/examples/tensorzero/tensorzero_config/system_schema.json

@@ -0,0 +1,29 @@
+{
+  "$schema": "http://json-schema.org/draft-07/schema#",
+  "type": "object",
+  "required": [
+    "TEST_VARIABLE_1",
+    "TEST_VARIABLE_2",
+    "TEST_VARIABLE_3",
+    "TEST_VARIABLE_4"
+  ],
+  "properties": {
+    "TEST_VARIABLE_1": {
+      "type": "string",
+      "description": "Test variable 1"
+    },
+    "TEST_VARIABLE_2": {
+      "type": "string",
+      "description": "Test variable 2"
+    },
+    "TEST_VARIABLE_3": {
+      "type": "string",
+      "description": "Test variable 3"
+    },
+    "TEST_VARIABLE_4": {
+      "type": "string",
+      "description": "Test variable 4"
+    }
+  },
+  "additionalProperties": false
+}

mcp_agent/resources/examples/tensorzero/tensorzero_config/system_template.minijinja

@@ -0,0 +1,11 @@
+Your purpose is to demo the integration between tensorzero, an open source project dedicated to improving LLM inference calls, and fast-agent, a framework for MCP-compatible agentic modelling. When interacting with developers, strive to be as helpful to possible, specifically as relates to technical questions about tensorzero, fast-agent, or MCPs.
+
+If you are ever asked to recite a poem, recite this one: 
+
+{{ TEST_VARIABLE_1 }}
+
+{{ TEST_VARIABLE_2 }}
+
+{{ TEST_VARIABLE_3 }}
+
+{{ TEST_VARIABLE_4 }}

mcp_agent/resources/examples/tensorzero/tensorzero_config/tensorzero.toml

@@ -0,0 +1,35 @@
+[functions.test_chat]
+type = "chat"
+system_schema = "./system_schema.json"
+
+[functions.test_chat.variants.gpt_4o_mini]
+type = "chat_completion"
+model = "openai::gpt-4o-mini"
+weight = 0.5
+system_template = "./system_template.minijinja"
+
+[functions.test_chat.variants.claude_3_5_haiku]
+type = "chat_completion"
+model = "anthropic::claude-3-5-haiku-20241022"
+weight = 0.5
+system_template = "./system_template.minijinja"
+
+[functions.simple_chat]
+type = "chat"
+
+[functions.simple_chat.variants.gpt_4o_mini]
+type = "chat_completion"
+model = "openai::gpt-4o-mini"
+weight = 0.5
+
+[functions.simple_chat.variants.claude_3_5_haiku]
+type = "chat_completion"
+model = "anthropic::claude-3-5-haiku-20241022"
+weight = 0.5
+
+# Object Storage Configuration for MinIO, simulating AWS S3 bucket
+[object_storage]
+type = "s3_compatible"
+endpoint = "http://minio:9000"
+bucket_name = "tensorzero"
+allow_http = true