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

{fast_agent_mcp-0.2.50.dist-info → fast_agent_mcp-0.2.51.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fast-agent-mcp
-Version: 0.2.50
+Version: 0.2.51
 Summary: Define, Prompt and Test MCP enabled Agents and Workflows
 Author-email: Shaun Smith <fastagent@llmindset.co.uk>
 License:                                  Apache License

{fast_agent_mcp-0.2.50.dist-info → fast_agent_mcp-0.2.51.dist-info}/RECORD

@@ -27,7 +27,7 @@ mcp_agent/cli/main.py,sha256=Oo13X7LB0Cf7JrkilQXz8Eqi_48cE0Rr2qqDUOQifEQ,3175
 mcp_agent/cli/terminal.py,sha256=GRwD-RGW7saIz2IOWZn5vD6JjiArscELBThm1GTFkuI,1065
 mcp_agent/cli/commands/check_config.py,sha256=15YK0mtDQbVopnMm3HBjOeY2-00FUHj6tt8RvaemKmI,21081
 mcp_agent/cli/commands/go.py,sha256=SHFJlO3SeZzAWvx_Gz72zXKA8n7XfaZ49W6UNS5mbV4,14668
-mcp_agent/cli/commands/quickstart.py,sha256=lcozUGP9RRO8xZaayJg4pQNeY5zDQs-eg-ABm0A15cI,16471
+mcp_agent/cli/commands/quickstart.py,sha256=hwIr1F9zGRSQGf7kwvmirMIK7Qke2s6W95inZ-a2SMQ,21171
 mcp_agent/cli/commands/server_helpers.py,sha256=x5tD_qhf1W4D2li09sfOyfRWCOCa6lmpumYAPsEfIQs,3649
 mcp_agent/cli/commands/setup.py,sha256=eOEd4TL-b0DaDeSJMGOfNOsTEItoZ67W88eTP4aP-bo,6482
 mcp_agent/cli/commands/url_parser.py,sha256=5VdtcHRHzi67YignStVbz7u-rcvNNErw9oJLAUFOtEY,5855
@@ -67,8 +67,8 @@ mcp_agent/llm/augmented_llm_playback.py,sha256=rLzgai496e2RlxqQp_Bp0U-Y1FF1SGsWl
 mcp_agent/llm/augmented_llm_silent.py,sha256=IUnK_1Byy4D9TG0Pj46LFeNezgSTQ8d6MQIHWAImBwE,1846
 mcp_agent/llm/augmented_llm_slow.py,sha256=DDSD8bL2flmQrVHZm-UDs7sR8aHRWkDOcOW-mX_GPok,2067
 mcp_agent/llm/memory.py,sha256=pTOaTDV3EA3X68yKwEtUAu7s0xGIQQ_cKBhfYUnfR0w,8614
-mcp_agent/llm/model_database.py,sha256=7G_rAXRE2RFrly_2PRr5uji7fT0XqaWj_RwhpZVx1LU,10690
-mcp_agent/llm/model_factory.py,sha256=WyuQm0hETOxECA9pjCm-Oo5Oh28186nsSWmjSntw2SE,11784
+mcp_agent/llm/model_database.py,sha256=ehNiw_d5y4f1t_ySJSigxN1NxikXEXFXbkO5E0gLX9M,10943
+mcp_agent/llm/model_factory.py,sha256=dIVs0RBhVy31-8nAjXdnWSIUJmW5UKnQym_oeQNuNiU,11906
 mcp_agent/llm/prompt_utils.py,sha256=yWQHykoK13QRF7evHUKxVF0SpVLN-Bsft0Yixzvn0g0,4825
 mcp_agent/llm/provider_key_manager.py,sha256=LSWIgcXlrUS4sfBvQBCya82qC6NcXQPYLtDHwHNOXR4,3394
 mcp_agent/llm/provider_types.py,sha256=LfuVuFjcM_lMED0lkrNfKK8s8Fs1vbxugQsrcBE2CIY,1119
@@ -86,14 +86,13 @@ mcp_agent/llm/providers/augmented_llm_generic.py,sha256=5Uq8ZBhcFuQTt7koP_5ykolR
 mcp_agent/llm/providers/augmented_llm_google_native.py,sha256=c6zczfs-Iw70j3OYELHJ4S7CRwAddkeXinex_yLMhmU,22194
 mcp_agent/llm/providers/augmented_llm_google_oai.py,sha256=g_g46h-YuxqbRZiO_dVo5zO2OkX1yx7nb6xDaQbOvWs,1137
 mcp_agent/llm/providers/augmented_llm_groq.py,sha256=5pqWgOoEJpvL230rJekBNlmBzUegbgwYitArlXgAmY0,4424
-mcp_agent/llm/providers/augmented_llm_openai.py,sha256=veVi4NcIP5S7xiV5fZLCSRFSiQbU48xQye0z-xx7X8I,25183
+mcp_agent/llm/providers/augmented_llm_openai.py,sha256=ncFlItw4DKKWFQ1czePye2Hpo_qzL22Yanc3aaOUuHA,25343
 mcp_agent/llm/providers/augmented_llm_openrouter.py,sha256=m3wS83fabBOmaZJH9gQ9sFw_2TB4xTb44WCOPB-2NJ4,2001
-mcp_agent/llm/providers/augmented_llm_tensorzero.py,sha256=wxNffqdM1Psvfyya6Pmo__BZjyrek55_8iTRoH4TboI,20577
+mcp_agent/llm/providers/augmented_llm_tensorzero_openai.py,sha256=D53Fry2AfBLOB5z9hM19U6_HMJeVNTpiBCAJb11ulNg,5503
 mcp_agent/llm/providers/augmented_llm_xai.py,sha256=MhlX91IUNynQ_NDknx4EQJLwg-NbR8lcHS1P4JuLOnA,1433
-mcp_agent/llm/providers/google_converter.py,sha256=xCUVw4W7lqBvJMgSKKdzW4Y00zmHBgCR6pOoWLYQQfw,16699
+mcp_agent/llm/providers/google_converter.py,sha256=YmSfkkZR2YzR5DZduGaOW0v7BtjjindCIiuS8zOJ1VA,16941
 mcp_agent/llm/providers/multipart_converter_anthropic.py,sha256=QsgIwWJ14SNZsun3fKDQaUx5AYdk8X6UYeBgYlZgZN0,16607
 mcp_agent/llm/providers/multipart_converter_openai.py,sha256=8vBwX7PB62mV6pe4VvAj9YxzX1Dp1RJMZK1NWOMnioY,17998
-mcp_agent/llm/providers/multipart_converter_tensorzero.py,sha256=Tr3DgogthxqTw04ajblwE8aB3Ft42X7aSXsMlm5FP4k,8619
 mcp_agent/llm/providers/openai_multipart.py,sha256=qKBn7d3jSabnJmVgWweVzqh8q9mBqr09fsPmP92niAQ,6899
 mcp_agent/llm/providers/openai_utils.py,sha256=T4bTCL9f7DsoS_zoKgQKv_FUv_4n98vgbvaUpdWZJr8,1875
 mcp_agent/llm/providers/sampling_converter_anthropic.py,sha256=35WzBWkPklnuMlu5S6XsQIq0YL58NOy8Ja6A_l4m6eM,1612
@@ -155,6 +154,24 @@ mcp_agent/resources/examples/researcher/fastagent.config.yaml,sha256=TbVMHQCKcyt
 mcp_agent/resources/examples/researcher/researcher-eval.py,sha256=CR9m4lyoXijS1whvsBDuk6IA-RmNc6iOYbtloETkITY,1833
 mcp_agent/resources/examples/researcher/researcher-imp.py,sha256=oJxSVnLbZfIn71QbQR1E6j_m_UBrOOGP4SVljXErHLQ,7879
 mcp_agent/resources/examples/researcher/researcher.py,sha256=SZfExi-FfwYETzGt2O3caS3L5E6EemV3IUrJHyzZqHI,1333
+mcp_agent/resources/examples/tensorzero/.env.sample,sha256=khV_apbP4XprpNuIaeVicnHaVHEkwIdWGyZCvW1OLDc,35
+mcp_agent/resources/examples/tensorzero/Makefile,sha256=BOvcJvPBAJN2MDh8brLsy2leHGwuT_bBjPzakOsUSCU,427
+mcp_agent/resources/examples/tensorzero/README.md,sha256=xsA1qWjg2mI240jlbeFYWjYm_pQLsGeQiPiSlEtayeQ,2126
+mcp_agent/resources/examples/tensorzero/agent.py,sha256=8x_W-_LHbwC_onXWZfDIbVQe2PX6qEZ5JRjwIWGOEr4,1272
+mcp_agent/resources/examples/tensorzero/docker-compose.yml,sha256=YgmWgoHHDDTGeqRlU8Z21mLdEqo48IoKZsnlXaDZkrc,2851
+mcp_agent/resources/examples/tensorzero/fastagent.config.yaml,sha256=3uJJ9p-0evDG48zr2QneektO4fe_h4uRPh1dyWfe7-k,355
+mcp_agent/resources/examples/tensorzero/image_demo.py,sha256=BVdCMg1o8r9A9cgKEPmEO84CBeBOxz9zNF9yBxyYCp4,2122
+mcp_agent/resources/examples/tensorzero/simple_agent.py,sha256=BjbD6k9C5zNJl_vAzKKIVSyghYuOsF-LOzqxMTnEOJ4,858
+mcp_agent/resources/examples/tensorzero/demo_images/clam.jpg,sha256=K1NWrhz5QMfgjZfDMMIVctWwbwTyJSsPHLDaUMbcn18,22231
+mcp_agent/resources/examples/tensorzero/demo_images/crab.png,sha256=W7R3bSKKDmZCjxJEmFk0pBXVhXXhfPGM1gIiVuv_eu4,58413
+mcp_agent/resources/examples/tensorzero/demo_images/shrimp.png,sha256=2r3c6yHE25MpVRDTTwxjAGs1ShJ2UI-qxJqbxa-9ew4,7806
+mcp_agent/resources/examples/tensorzero/mcp_server/Dockerfile,sha256=pujKW5xDY5q0PvIYsn8Wsh9Tz2F3sVuKTJc3o8lgt4M,681
+mcp_agent/resources/examples/tensorzero/mcp_server/entrypoint.sh,sha256=yLUXZKd4mF2k_ZFAd6125ZstLmrtksePpfgCNdewSyU,1123
+mcp_agent/resources/examples/tensorzero/mcp_server/mcp_server.py,sha256=8Wr6VzNFihri3mv9wLo37i5Ly1VWL4s4jSnfiRG7Ez0,933
+mcp_agent/resources/examples/tensorzero/mcp_server/pyproject.toml,sha256=3AppZ_HbIEmRGg_phGuEOq-Q63CY_IIsrIBr658ka9U,228
+mcp_agent/resources/examples/tensorzero/tensorzero_config/system_schema.json,sha256=q81vtb8eyX1gU0qOhT_BFNo7nI2Lg2o-D_Bvp8watEI,626
+mcp_agent/resources/examples/tensorzero/tensorzero_config/system_template.minijinja,sha256=_Ekz9YuE76pkmwtcMNJeDlih2U5vPRgFB5wFojAVde8,501
+mcp_agent/resources/examples/tensorzero/tensorzero_config/tensorzero.toml,sha256=wYDKzHyX0A2x42d1vF5a72ot304iTP8_i5Y1rzAcCEA,890
 mcp_agent/resources/examples/workflows/chaining.py,sha256=tY0kA0U8s2rceAO4ogZFtpQEkiUWcrYnYDgHu_-4G50,889
 mcp_agent/resources/examples/workflows/evaluator.py,sha256=XJXrk5r1hrJzfZAMtQ7WIggy6qPttMJG1yqxYELO7C4,3101
 mcp_agent/resources/examples/workflows/fastagent.config.yaml,sha256=qaxk-p7Pl7JepdL3a7BTl0CIp4LHCXies7pFdVWS9xk,783
@@ -168,8 +185,8 @@ mcp_agent/resources/examples/workflows/short_story.txt,sha256=X3y_1AyhLFN2AKzCKv
 mcp_agent/tools/tool_definition.py,sha256=L3Pxl-uLEXqlVoo-bYuFTFALeI-2pIU44YgFhsTKEtM,398
 mcp_agent/ui/console_display.py,sha256=XXrHr950wSBSedEKUaaGkXjOzuFpQYzUKKiyaZ58Mps,28280
 mcp_agent/ui/console_display_legacy.py,sha256=sm2v61-IPVafbF7uUaOyhO2tW_zgFWOjNS83IEWqGgI,14931
-fast_agent_mcp-0.2.50.dist-info/METADATA,sha256=jNGAWFccNjB_-7zrb8G6ambu93q8jtpvKxiA4Jp3rkQ,31048
-fast_agent_mcp-0.2.50.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-fast_agent_mcp-0.2.50.dist-info/entry_points.txt,sha256=QaX5kLdI0VdMPRdPUF1nkG_WdLUTNjp_icW6e3EhNYU,232
-fast_agent_mcp-0.2.50.dist-info/licenses/LICENSE,sha256=Gx1L3axA4PnuK4FxsbX87jQ1opoOkSFfHHSytW6wLUU,10935
-fast_agent_mcp-0.2.50.dist-info/RECORD,,
+fast_agent_mcp-0.2.51.dist-info/METADATA,sha256=6gg7Cc2qW7R5rVOUtSyhlrbBtZyTEOisFsdU00tohUk,31048
+fast_agent_mcp-0.2.51.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+fast_agent_mcp-0.2.51.dist-info/entry_points.txt,sha256=QaX5kLdI0VdMPRdPUF1nkG_WdLUTNjp_icW6e3EhNYU,232
+fast_agent_mcp-0.2.51.dist-info/licenses/LICENSE,sha256=Gx1L3axA4PnuK4FxsbX87jQ1opoOkSFfHHSytW6wLUU,10935
+fast_agent_mcp-0.2.51.dist-info/RECORD,,

mcp_agent/cli/commands/quickstart.py

@@ -76,6 +76,25 @@ EXAMPLE_TYPES = {
         ],
         "create_subdir": True,
     },
+    "tensorzero": {
+        "description": "A complete example showcasing the TensorZero integration.\n"
+        "Includes the T0 Gateway, an MCP server, an interactive agent, and \n"
+        "multi-modal functionality.",
+        "files": [
+            ".env.sample",
+            "Makefile",
+            "README.md",
+            "agent.py",
+            "docker-compose.yml",
+            "fastagent.config.yaml",
+            "image_demo.py",
+            "simple_agent.py",
+            "mcp_server/",
+            "demo_images/",
+            "tensorzero_config/"
+        ],
+        "create_subdir": True,
+    },
 }
 
 
@@ -225,6 +244,27 @@ def copy_example_files(example_type: str, target_dir: Path, force: bool = False)
     return created
 
 
+def copy_project_template(source_dir: Path, dest_dir: Path, console: Console, force: bool = False):
+    """
+    Recursively copies a project template directory.
+    This is a helper to handle project-based quickstarts like TensorZero.
+    """
+    if dest_dir.exists():
+        if force:
+            console.print(f"[yellow]--force specified. Removing existing directory: {dest_dir}[/yellow]")
+            shutil.rmtree(dest_dir)
+        else:
+            console.print(f"[bold yellow]Directory '{dest_dir.name}' already exists.[/bold yellow] Use --force to overwrite.")
+            return False
+
+    try:
+        shutil.copytree(source_dir, dest_dir)
+        return True
+    except Exception as e:
+        console.print(f"[red]Error copying project template: {e}[/red]")
+        return False
+
+
 def show_overview() -> None:
     """Display an overview of available examples in a nicely formatted table."""
     console.print("\n[bold cyan]fast-agent quickstarts[/bold cyan]")
@@ -397,6 +437,73 @@ def _show_completion_message(example_type: str, created: list[str]) -> None:
         console.print("\n[yellow]No files were created.[/yellow]")
 
 
+@app.command(name="tensorzero", help="Create the TensorZero integration example project.")
+def tensorzero(
+    directory: Path = typer.Argument(
+        Path("."),
+        help="Directory where the 'tensorzero' project folder will be created.",
+    ),
+    force: bool = typer.Option(False, "--force", "-f", help="Force overwrite if project directory exists"),
+):
+    """Create the TensorZero project example."""
+    console.print("[bold green]Setting up the TensorZero quickstart example...[/bold green]")
+
+    dest_project_dir = directory.resolve() / "tensorzero"
+
+    # --- Find Source Directory ---
+    from importlib.resources import files
+    try:
+        # This path MUST match the "to" path from hatch_build.py
+        source_dir = files("mcp_agent").joinpath("resources").joinpath("examples").joinpath("tensorzero")
+        if not source_dir.is_dir():
+            raise FileNotFoundError  # Fallback to dev mode if resource isn't a dir
+    except (ImportError, ModuleNotFoundError, FileNotFoundError):
+        console.print("[yellow]Package resources not found. Falling back to development mode.[/yellow]")
+        # This path is relative to the project root in a development environment
+        source_dir = Path(__file__).parent.parent.parent.parent / "examples" / "tensorzero"
+
+    if not source_dir.exists() or not source_dir.is_dir():
+        console.print(f"[red]Error: Source project directory not found at '{source_dir}'[/red]")
+        raise typer.Exit(1)
+
+    console.print(f"Source directory: [dim]{source_dir}[/dim]")
+    console.print(f"Destination: [dim]{dest_project_dir}[/dim]")
+
+    # --- Copy Project and Show Message ---
+    if copy_project_template(source_dir, dest_project_dir, console, force):
+        console.print(
+            f"\n[bold green]✅ Success![/bold green] Your TensorZero project has been created in: [cyan]{dest_project_dir}[/cyan]"
+        )
+        console.print("\n[bold yellow]Next Steps:[/bold yellow]")
+        console.print("\n1. [bold]Navigate to your new project directory:[/bold]")
+        console.print(f"   [cyan]cd {dest_project_dir.relative_to(Path.cwd())}[/cyan]")
+
+        console.print("\n2. [bold]Set up your API keys:[/bold]")
+        console.print("   [cyan]cp .env.sample .env[/cyan]")
+        console.print(
+            "   [dim]Then, open the new '.env' file and add your OpenAI or Anthropic API key.[/dim]"
+        )
+
+        console.print("\n3. [bold]Start the required services (TensorZero Gateway & MCP Server):[/bold]")
+        console.print("   [cyan]docker compose up --build -d[/cyan]")
+        console.print(
+            "   [dim](This builds and starts the necessary containers in the background)[/dim]"
+        )
+
+        console.print("\n4. [bold]Run the interactive agent:[/bold]")
+        console.print("   [cyan]make agent[/cyan]  (or `uv run agent.py`)")
+        console.print("\nEnjoy exploring the TensorZero integration with fast-agent! ✨")
+
+
+@app.command(name="t0", help="Alias for the TensorZero quickstart.", hidden=True)
+def t0_alias(
+    directory: Path = typer.Argument(Path("."), help="Directory for the 'tensorzero' project folder."),
+    force: bool = typer.Option(False, "--force", "-f", help="Force overwrite"),
+):
+    """Alias for the `tensorzero` command."""
+    tensorzero(directory, force)
+
+
 @app.callback(invoke_without_command=True)
 def main(ctx: typer.Context) -> None:
     """Quickstart applications for fast-agent."""

mcp_agent/llm/model_database.py

@@ -120,7 +120,9 @@ class ModelDatabase:
     OPENAI_O3_MINI_SERIES = ModelParameters(
         context_window=200000, max_output_tokens=100000, tokenizes=TEXT_ONLY
     )
-
+    OPENAI_GPT_OSS_SERIES = ModelParameters(
+        context_window=131072, max_output_tokens=32766, tokenizes=TEXT_ONLY, json_mode="object"
+    )
     # TODO update to 32000
     ANTHROPIC_OPUS_4_VERSIONED = ModelParameters(
         context_window=200000, max_output_tokens=32000, tokenizes=ANTHROPIC_MULTIMODAL
@@ -236,6 +238,8 @@ class ModelDatabase:
         "moonshotai/kimi-k2-instruct": KIMI_MOONSHOT,
         "qwen/qwen3-32b": QWEN3_REASONER,
         "deepseek-r1-distill-llama-70b": DEEPSEEK_DISTILL,
+        "openai/gpt-oss-120b": OPENAI_GPT_OSS_SERIES,
+        "openai/gpt-oss-20b": OPENAI_GPT_OSS_SERIES,
     }
 
     @classmethod

mcp_agent/llm/model_factory.py

@@ -22,7 +22,7 @@ from mcp_agent.llm.providers.augmented_llm_google_oai import GoogleOaiAugmentedL
 from mcp_agent.llm.providers.augmented_llm_groq import GroqAugmentedLLM
 from mcp_agent.llm.providers.augmented_llm_openai import OpenAIAugmentedLLM
 from mcp_agent.llm.providers.augmented_llm_openrouter import OpenRouterAugmentedLLM
-from mcp_agent.llm.providers.augmented_llm_tensorzero import TensorZeroAugmentedLLM
+from mcp_agent.llm.providers.augmented_llm_tensorzero_openai import TensorZeroOpenAIAugmentedLLM
 from mcp_agent.llm.providers.augmented_llm_xai import XAIAugmentedLLM
 from mcp_agent.mcp.interfaces import AugmentedLLMProtocol
 
@@ -39,7 +39,7 @@ LLMClass = Union[
     Type[SlowLLM],
     Type[DeepSeekAugmentedLLM],
     Type[OpenRouterAugmentedLLM],
-    Type[TensorZeroAugmentedLLM],
+    Type[TensorZeroOpenAIAugmentedLLM],
     Type[GoogleNativeAugmentedLLM],
     Type[GenericAugmentedLLM],
     Type[AzureOpenAIAugmentedLLM],
@@ -144,6 +144,8 @@ class ModelFactory:
         "gemini25": "gemini-2.5-flash-preview-05-20",
         "gemini25pro": "gemini-2.5-pro-preview-05-06",
         "kimi": "groq.moonshotai/kimi-k2-instruct",
+        "gpt-oss": "groq.openai/gpt-oss-120b",
+        "gpt-oss-20b": "groq.openai/gpt-oss-20b",
     }
 
     # Mapping of providers to their LLM classes
@@ -157,7 +159,7 @@ class ModelFactory:
         Provider.GOOGLE: GoogleNativeAugmentedLLM,
         Provider.XAI: XAIAugmentedLLM,
         Provider.OPENROUTER: OpenRouterAugmentedLLM,
-        Provider.TENSORZERO: TensorZeroAugmentedLLM,
+        Provider.TENSORZERO: TensorZeroOpenAIAugmentedLLM,
         Provider.AZURE: AzureOpenAIAugmentedLLM,
         Provider.ALIYUN: AliyunAugmentedLLM,
         Provider.BEDROCK: BedrockAugmentedLLM,

mcp_agent/llm/providers/augmented_llm_openai.py

@@ -391,8 +391,13 @@ class OpenAIAugmentedLLM(AugmentedLLM[ChatCompletionMessageParam, ChatCompletion
             # Convert to dict and remove None values
             message_dict = message.model_dump()
             message_dict = {k: v for k, v in message_dict.items() if v is not None}
-            if model_name == "deepseek-r1-distill-llama-70b":
+            if model_name in (
+                "deepseek-r1-distill-llama-70b",
+                "openai/gpt-oss-120b",
+                "openai/gpt-oss-20b",
+            ):
                 message_dict.pop("reasoning", None)
+                message_dict.pop("channel", None)
 
             messages.append(message_dict)
 

mcp_agent/llm/providers/augmented_llm_tensorzero_openai.py

@@ -0,0 +1,127 @@
+from typing import Any, Dict, List, Optional
+
+from openai.types.chat import ChatCompletionMessageParam, ChatCompletionSystemMessageParam
+
+from mcp_agent.core.request_params import RequestParams
+from mcp_agent.llm.provider_types import Provider
+from mcp_agent.llm.providers.augmented_llm_openai import OpenAIAugmentedLLM
+
+
+class TensorZeroOpenAIAugmentedLLM(OpenAIAugmentedLLM):
+    """
+    An LLM augmentation that interacts with TensorZero's OpenAI-compatible inference endpoint.
+    This class extends the base OpenAIAugmentedLLM to handle TensorZero-specific
+    features, such as system template variables and custom parameters.
+    """
+
+    def __init__(self, *args, **kwargs) -> None:
+        """
+        Initializes the TensorZeroOpenAIAugmentedLLM.
+
+        Args:
+            *args: Variable length argument list.
+            **kwargs: Arbitrary keyword arguments.
+        """
+        self._t0_episode_id = kwargs.pop("episode_id", None)
+        self._t0_function_name = kwargs.get("model", "")
+
+        super().__init__(*args, provider=Provider.TENSORZERO, **kwargs)
+        self.logger.info("TensorZeroOpenAIAugmentedLLM initialized.")
+
+    def _initialize_default_params(self, kwargs: dict) -> RequestParams:
+        """
+        Initializes TensorZero-specific default parameters. Ensures the model name
+        is correctly prefixed for the TensorZero API.
+        """
+        model = kwargs.get("model", "")
+        if not model.startswith("tensorzero::"):
+            model = f"tensorzero::function_name::{model}"
+
+        self.logger.debug(f"Initializing with TensorZero model: {model}")
+
+        return RequestParams(
+            model=model,
+            systemPrompt=self.instruction,
+            parallel_tool_calls=True,
+            max_iterations=10,
+            use_history=True,
+        )
+
+    def _base_url(self) -> str:
+        """
+        Constructs the TensorZero OpenAI-compatible endpoint URL.
+        """
+        default_url = "http://localhost:3000/openai/v1"
+        if self.context and self.context.config and hasattr(self.context.config, "tensorzero"):
+            base_url = getattr(self.context.config.tensorzero, "base_url", default_url)
+            # Ensure the path is correctly appended
+            if not base_url.endswith('/openai/v1'):
+                base_url = f"{base_url.rstrip('/')}/openai/v1"
+            self.logger.debug(f"Using TensorZero base URL from config: {base_url}")
+            return base_url
+        self.logger.debug(f"Using default TensorZero base URL: {default_url}")
+        return default_url
+
+    def _prepare_api_request(
+            self,
+            messages: List[ChatCompletionMessageParam],
+            tools: Optional[List[Any]],
+            request_params: RequestParams
+    ) -> Dict[str, Any]:
+        """
+        Prepares the API request for the TensorZero OpenAI-compatible endpoint.
+        This method injects system template variables and other TensorZero-specific
+        parameters into the request. It also handles multimodal inputs.
+        """
+        self.logger.debug("Preparing API request for TensorZero OpenAI endpoint.")
+
+        # Start with the base arguments from the parent class
+        arguments = super()._prepare_api_request(messages, tools, request_params)
+
+        # Handle system template variables
+        if request_params.template_vars:
+            self.logger.debug(f"Injecting template variables: {request_params.template_vars}")
+            system_message_found = False
+            for i, msg in enumerate(messages):
+                if msg.get("role") == "system":
+                    # If content is a string, convert it to the TensorZero format
+                    if isinstance(msg.get("content"), str):
+                        messages[i] = ChatCompletionSystemMessageParam(
+                            role="system",
+                            content=[request_params.template_vars]
+                        )
+                    elif isinstance(msg.get("content"), list):
+                        # If content is already a list, merge the template vars
+                        msg["content"][0].update(request_params.template_vars)
+                    system_message_found = True
+                    break
+
+            if not system_message_found:
+                # If no system message exists, create one
+                messages.insert(0, ChatCompletionSystemMessageParam(
+                    role="system",
+                    content=[request_params.template_vars]
+                ))
+
+        # Add TensorZero-specific extra body parameters
+        extra_body = arguments.get("extra_body", {})
+
+        if self._t0_episode_id:
+            extra_body["tensorzero::episode_id"] = str(self._t0_episode_id)
+            self.logger.debug(f"Added tensorzero::episode_id: {self._t0_episode_id}")
+
+        # Merge metadata arguments
+        if request_params.metadata and isinstance(request_params.metadata, dict):
+            t0_args = request_params.metadata.get("tensorzero_arguments")
+            if t0_args:
+                self.logger.debug(f"Merging tensorzero_arguments from metadata: {t0_args}")
+                for msg in messages:
+                    if msg.get("role") == "system" and isinstance(msg.get("content"), list):
+                        msg["content"][0].update(t0_args)
+                        break
+
+        if extra_body:
+            arguments["extra_body"] = extra_body
+
+        self.logger.debug(f"Final API request arguments: {arguments}")
+        return arguments

mcp_agent/llm/providers/google_converter.py

@@ -336,6 +336,10 @@ class GoogleConverter:
         """
         Converts a single google.genai types.Content to a fast-agent PromptMessageMultipart.
         """
+        # Official fix for GitHub issue #207: Handle None content or content.parts
+        if content is None or not hasattr(content, "parts") or content.parts is None:
+            return PromptMessageMultipart(role="assistant", content=[])
+
         if content.role == "model" and any(part.function_call for part in content.parts):
             return PromptMessageMultipart(role="assistant", content=[])
 

mcp_agent/resources/examples/tensorzero/.env.sample

@@ -0,0 +1,2 @@
+OPENAI_API_KEY=
+ANTHROPIC_API_KEY=

mcp_agent/resources/examples/tensorzero/Makefile

@@ -0,0 +1,31 @@
+.PHONY: all
+
+build:
+	docker compose build
+
+up:
+	docker compose up -d
+
+logs:
+	docker compose logs -f
+
+tensorzero-logs:
+	docker compose logs -f gateway
+
+mcp-logs:
+	docker compose logs -f mcp-server
+
+minio-logs:
+	docker compose logs -f minio
+
+stop:
+	docker compose stop
+
+agent:
+	uv run agent.py --model=tensorzero.test_chat
+
+simple-agent:
+	uv run simple_agent.py --model=tensorzero.simple_chat
+
+image-test:
+	uv run image_demo.py 

mcp_agent/resources/examples/tensorzero/README.md

@@ -0,0 +1,55 @@
+# About the tensorzero / fast-agent integration
+
+[TensorZero](https://www.tensorzero.com/) is an open source project designed to help LLM application developers rapidly improve their inference calls. Its core features include:
+
+- A uniform inference interface to all leading LLM platforms.
+- The ability to dynamic route to different platforms and program failovers.
+- Automated parameter tuning and training
+- Advance templating features for your system prompts
+- Organization of LLM inference data into a Clickhouse DB allowing for sophisticated downstream analytics
+- A bunch of other good stuff is always in development
+
+`tensorzero` is powerful heavy, so we provide here a quickstart example that combines the basic components of `fast-agent`, an MCP server, `tensorzero`, and other supporting services into a cohesive whole.
+
+## Quickstart guide
+
+- Build and activate the `uv` `fast-agent` environment
+- Ensure that ports `3000`, `4000`, `8000`, `9000`, and `9001` are unallocated before running this demo.
+- Run `cp .env.sample .env` and then drop in at least one of `OPENAI_API_KEY` or `ANTHROPIC_API_KEY`. Make sure the accounts are funded.
+- `make up`
+- `make agent`
+
+The demo test's our implementation's ability to:
+
+- Implement the T0 model gateway as an inference backend
+- Implement T0's dynamic templating feature
+- Have in-conversation memory
+- Describe and execute tool calls
+- Remember previous tool calls
+
+A version of a conversation to test all of this could be:
+
+```
+Hi.
+
+Tell me a poem.
+
+Do you have any tools that you can use?
+
+Please demonstrate the use of that tool on your last response.
+
+Please summarize the conversation so far.
+
+What tool calls have you executed in this session, and what were their results?
+```
+
+## Multimodal support
+
+Run `make image-test` to test the gateway's ability to handle base64-encoded image data
+
+## Development notes:
+
+- `make stop` will stop the MCP server and the tensorzero server
+- `make tenzorzero-logs` will tail the tensorzero server logs
+- `make mcp-logs` will tail the MCP server logs
+- Generic `make logs` dumps all log output from all services to terminal

mcp_agent/resources/examples/tensorzero/agent.py

@@ -0,0 +1,35 @@
+import asyncio
+
+from mcp_agent.core.fastagent import FastAgent
+from mcp_agent.core.request_params import RequestParams
+
+# Explicitly provide the path to the config file in the current directory
+CONFIG_FILE = "fastagent.config.yaml"
+fast = FastAgent("fast-agent example", config_path=CONFIG_FILE, ignore_unknown_args=True)
+
+# Define T0 system variables here
+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.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"],
+    request_params=RequestParams(template_vars=my_t0_system_vars),
+)
+async def main():
+    async with fast.run() as agent_app:  # Get the AgentApp wrapper
+        print("\nStarting interactive session with template_vars set via decorator...")
+        await agent_app.interactive()
+
+
+if __name__ == "__main__":
+    asyncio.run(main())  # type: ignore