fast-agent-mcp 0.1.11__py3-none-any.whl → 0.1.13__py3-none-any.whl

mcp_agent/cli/commands/bootstrap.py

@@ -2,10 +2,11 @@
 
 import shutil
 from pathlib import Path
+
 import typer
 from rich.console import Console
-from rich.table import Table
 from rich.panel import Panel
+from rich.table import Table
 
 app = typer.Typer(
     help="Create example applications",
@@ -49,9 +50,7 @@ EXAMPLE_TYPES = {
 }
 
 
-def copy_example_files(
-    example_type: str, target_dir: Path, force: bool = False
-) -> list[str]:
+def copy_example_files(example_type: str, target_dir: Path, force: bool = False) -> list[str]:
     """Copy example files from resources to target directory."""
     created = []
 
@@ -72,12 +71,7 @@ def copy_example_files(
             console.print(f"Created mount-point directory: {mount_point_dir}")
 
     # Use the resources directory from the package
-    source_dir = (
-        Path(__file__).parent.parent.parent
-        / "resources"
-        / "examples"
-        / ("workflows" if example_type == "workflow" else f"{example_type}")
-    )
+    source_dir = Path(__file__).parent.parent.parent / "resources" / "examples" / ("workflows" if example_type == "workflow" else f"{example_type}")
 
     if not source_dir.exists():
         console.print(f"[red]Error: Source directory not found: {source_dir}[/red]")
@@ -116,9 +110,7 @@ def copy_example_files(
                     continue
 
                 if target.exists() and not force:
-                    console.print(
-                        f"[yellow]Skipping[/yellow] mount-point/{filename} (already exists)"
-                    )
+                    console.print(f"[yellow]Skipping[/yellow] mount-point/{filename} (already exists)")
                     continue
 
                 shutil.copy2(source, target)
@@ -126,22 +118,18 @@ def copy_example_files(
                 console.print(f"[green]Created[/green] mount-point/{filename}")
 
             except Exception as e:
-                console.print(
-                    f"[red]Error copying mount-point/{filename}: {str(e)}[/red]"
-                )
+                console.print(f"[red]Error copying mount-point/{filename}: {str(e)}[/red]")
 
     return created
 
 
-def show_overview():
+def show_overview() -> None:
     """Display an overview of available examples in a nicely formatted table."""
     console.print("\n[bold cyan]fast-agent Example Applications[/bold cyan]")
     console.print("Build agents and compose workflows through practical examples\n")
 
     # Create a table for better organization
-    table = Table(
-        show_header=True, header_style="bold magenta", box=None, padding=(0, 2)
-    )
+    table = Table(show_header=True, header_style="bold magenta", box=None, padding=(0, 2))
     table.add_column("Example")
     table.add_column("Description")
     table.add_column("Files")
@@ -149,9 +137,7 @@ def show_overview():
     for name, info in EXAMPLE_TYPES.items():
         files_list = "\n".join(f"• {f}" for f in info["files"])
         if "mount_point_files" in info:
-            files_list += "\n[blue]mount-point:[/blue]\n" + "\n".join(
-                f"• {f}" for f in info["mount_point_files"]
-            )
+            files_list += "\n[blue]mount-point:[/blue]\n" + "\n".join(f"• {f}" for f in info["mount_point_files"])
         table.add_row(f"[green]{name}[/green]", info["description"], files_list)
 
     console.print(table)
@@ -178,10 +164,8 @@ def workflow(
         Path("."),
         help="Directory where workflow examples will be created",
     ),
-    force: bool = typer.Option(
-        False, "--force", "-f", help="Force overwrite existing files"
-    ),
-):
+    force: bool = typer.Option(False, "--force", "-f", help="Force overwrite existing files"),
+) -> None:
     """Create workflow pattern examples."""
     target_dir = directory.resolve()
     if not target_dir.exists():
@@ -198,10 +182,8 @@ def researcher(
         Path("."),
         help="Directory where researcher examples will be created (in 'researcher' subdirectory)",
     ),
-    force: bool = typer.Option(
-        False, "--force", "-f", help="Force overwrite existing files"
-    ),
-):
+    force: bool = typer.Option(False, "--force", "-f", help="Force overwrite existing files"),
+) -> None:
     """Create researcher pattern examples."""
     target_dir = directory.resolve()
     if not target_dir.exists():
@@ -218,10 +200,8 @@ def data_analysis(
         Path("."),
         help="Directory where data analysis examples will be created (creates 'data-analysis' subdirectory with mount-point)",
     ),
-    force: bool = typer.Option(
-        False, "--force", "-f", help="Force overwrite existing files"
-    ),
-):
+    force: bool = typer.Option(False, "--force", "-f", help="Force overwrite existing files"),
+) -> None:
     """Create data analysis examples with sample dataset."""
     target_dir = directory.resolve()
     if not target_dir.exists():
@@ -232,7 +212,7 @@ def data_analysis(
     _show_completion_message("data-analysis", created)
 
 
-def _show_completion_message(example_type: str, created: list[str]):
+def _show_completion_message(example_type: str, created: list[str]) -> None:
     """Show completion message and next steps."""
     if created:
         console.print("\n[green]Setup completed successfully![/green]")
@@ -249,33 +229,23 @@ def _show_completion_message(example_type: str, created: list[str]):
             console.print("   - evaluator.py: Add evaluation capabilities")
             console.print("   - human_input.py: Incorporate human feedback")
             console.print("3. Run an example with: uv run <example>.py")
-            console.print(
-                "4. Try a different model with --model=<model>, or update the agent config"
-            )
+            console.print("4. Try a different model with --model=<model>, or update the agent config")
 
         elif example_type == "researcher":
-            console.print(
-                "1. Set up the Brave MCP Server (get an API key from https://brave.com/search/api/)"
-            )
+            console.print("1. Set up the Brave MCP Server (get an API key from https://brave.com/search/api/)")
             console.print("2. Try `uv run researcher.py` for the basic version")
-            console.print(
-                "3. Try `uv run researcher-eval.py` for the eval/optimize version"
-            )
+            console.print("3. Try `uv run researcher-eval.py` for the eval/optimize version")
         elif example_type == "data-analysis":
-            console.print(
-                "1. Run uv `analysis.py` to perform data analysis and visualization"
-            )
+            console.print("1. Run uv `analysis.py` to perform data analysis and visualization")
             console.print("2. The dataset is available in the mount-point directory:")
             console.print("   - mount-point/WA_Fn-UseC_-HR-Employee-Attrition.csv")
-            console.print(
-                "On Windows platforms, please edit the fastagent.config.yaml and adjust the volume mount point."
-            )
+            console.print("On Windows platforms, please edit the fastagent.config.yaml and adjust the volume mount point.")
     else:
         console.print("\n[yellow]No files were created.[/yellow]")
 
 
 @app.callback(invoke_without_command=True)
-def main(ctx: typer.Context):
+def main(ctx: typer.Context) -> None:
     """Create example applications and learn FastAgent patterns."""
     if ctx.invoked_subcommand is None:
         show_overview()

mcp_agent/cli/commands/config.py

@@ -1,11 +1,11 @@
+from typing import NoReturn
+
 import typer
 
 app = typer.Typer()
 
 
 @app.command()
-def show():
+def show() -> NoReturn:
     """Show the configuration."""
-    raise NotImplementedError(
-        "The show configuration command has not been implemented yet"
-    )
+    raise NotImplementedError("The show configuration command has not been implemented yet")

mcp_agent/cli/commands/setup.py

@@ -1,7 +1,8 @@
 from pathlib import Path
+
 import typer
-from rich.prompt import Confirm
 from rich.console import Console
+from rich.prompt import Confirm
 
 app = typer.Typer()
 console = Console()
@@ -166,17 +167,13 @@ def init(
         "-c",
         help="Directory where configuration files will be created",
     ),
-    force: bool = typer.Option(
-        False, "--force", "-f", help="Force overwrite existing files"
-    ),
-):
+    force: bool = typer.Option(False, "--force", "-f", help="Force overwrite existing files"),
+) -> None:
     """Initialize a new FastAgent project with configuration files and example agent."""
 
     config_path = Path(config_dir).resolve()
     if not config_path.exists():
-        should_create = Confirm.ask(
-            f"Directory {config_path} does not exist. Create it?", default=True
-        )
+        should_create = Confirm.ask(f"Directory {config_path} does not exist. Create it?", default=True)
         if should_create:
             config_path.mkdir(parents=True)
         else:
@@ -198,38 +195,26 @@ def init(
 
     # Create configuration files
     created = []
-    if create_file(
-        config_path / "fastagent.config.yaml", FASTAGENT_CONFIG_TEMPLATE, force
-    ):
+    if create_file(config_path / "fastagent.config.yaml", FASTAGENT_CONFIG_TEMPLATE, force):
         created.append("fastagent.yaml")
 
-    if create_file(
-        config_path / "fastagent.secrets.yaml", FASTAGENT_SECRETS_TEMPLATE, force
-    ):
+    if create_file(config_path / "fastagent.secrets.yaml", FASTAGENT_SECRETS_TEMPLATE, force):
         created.append("fastagent.secrets.yaml")
 
     if create_file(config_path / "agent.py", AGENT_EXAMPLE_TEMPLATE, force):
         created.append("agent.py")
 
     # Only create .gitignore if none exists in parent directories
-    if needs_gitignore and create_file(
-        config_path / ".gitignore", GITIGNORE_TEMPLATE, force
-    ):
+    if needs_gitignore and create_file(config_path / ".gitignore", GITIGNORE_TEMPLATE, force):
         created.append(".gitignore")
 
     if created:
         console.print("\n[green]Setup completed successfully![/green]")
         if "fastagent.secrets.yaml" in created:
             console.print("\n[yellow]Important:[/yellow] Remember to:")
-            console.print(
-                "1. Add your API keys to fastagent-secrets.yaml or set OPENAI_API_KEY and ANTHROPIC_API_KEY environment variables"
-            )
-            console.print(
-                "2. Keep fastagent.secrets.yaml secure and never commit it to version control"
-            )
-            console.print(
-                "3. Update fastagent.config.yaml to set a default model (currently system default is 'haiku')"
-            )
+            console.print("1. Add your API keys to fastagent-secrets.yaml or set OPENAI_API_KEY and ANTHROPIC_API_KEY environment variables")
+            console.print("2. Keep fastagent.secrets.yaml secure and never commit it to version control")
+            console.print("3. Update fastagent.config.yaml to set a default model (currently system default is 'haiku')")
         console.print("\nTo get started, run:")
         console.print("  uv run agent.py")
     else:

mcp_agent/cli/main.py

@@ -3,8 +3,9 @@
 import typer
 from rich.console import Console
 from rich.table import Table
+
+from mcp_agent.cli.commands import bootstrap, setup
 from mcp_agent.cli.terminal import Application
-from mcp_agent.cli.commands import setup, bootstrap
 
 app = typer.Typer(
     help="MCP Agent CLI - Build effective agents using Model Context Protocol",
@@ -20,7 +21,7 @@ application = Application()
 console = Console()
 
 
-def show_welcome():
+def show_welcome() -> None:
     """Show a welcome message with available commands."""
     from importlib.metadata import version
 
@@ -38,9 +39,7 @@ def show_welcome():
     table.add_column("Description")
 
     table.add_row("setup", "Set up a new agent project with configuration files")
-    table.add_row(
-        "bootstrap", "Create example applications (workflow, researcher, etc.)"
-    )
+    table.add_row("bootstrap", "Create example applications (workflow, researcher, etc.)")
     # table.add_row("config", "Manage agent configuration settings")
 
     console.print(table)
@@ -62,10 +61,8 @@ def main(
     ctx: typer.Context,
     verbose: bool = typer.Option(False, "--verbose", "-v", help="Enable verbose mode"),
     quiet: bool = typer.Option(False, "--quiet", "-q", help="Disable output"),
-    color: bool = typer.Option(
-        True, "--color/--no-color", help="Enable/disable color output"
-    ),
-):
+    color: bool = typer.Option(True, "--color/--no-color", help="Enable/disable color output"),
+) -> None:
     """FastAgent CLI - Build effective agents using Model Context Protocol (MCP).
 
     Use --help with any command for detailed usage information.

mcp_agent/cli/terminal.py

@@ -2,7 +2,7 @@ from mcp_agent.console import console, error_console
 
 
 class Application:
-    def __init__(self, verbosity: int = 0, enable_color: bool = True):
+    def __init__(self, verbosity: int = 0, enable_color: bool = True) -> None:
         self.verbosity = verbosity
         # Use the central console instances, respecting color setting
         if not enable_color:
@@ -13,7 +13,7 @@ class Application:
             self.console = console
             self.error_console = error_console
 
-    def log(self, message: str, level: str = "info"):
+    def log(self, message: str, level: str = "info") -> None:
         if level == "info" or (level == "debug" and self.verbosity > 0):
             if level == "error":
                 self.error_console.print(f"[{level.upper()}] {message}")

mcp_agent/config.py

@@ -295,11 +295,7 @@ def get_settings(config_path: str | None = None) -> Settings:
         """Recursively merge two dictionaries, preserving nested structures."""
         merged = base.copy()
         for key, value in update.items():
-            if (
-                key in merged
-                and isinstance(merged[key], dict)
-                and isinstance(value, dict)
-            ):
+            if key in merged and isinstance(merged[key], dict) and isinstance(value, dict):
                 merged[key] = deep_merge(merged[key], value)
             else:
                 merged[key] = value

mcp_agent/context.py

@@ -4,41 +4,33 @@ A central context object to store global state that is shared across the applica
 
 import asyncio
 import concurrent.futures
-from typing import Any, Optional, Union, TYPE_CHECKING
-
-from pydantic import BaseModel, ConfigDict
+from typing import TYPE_CHECKING, Any, Optional, Union
 
 from mcp import ServerSession
-
 from opentelemetry import trace
+from opentelemetry.exporter.otlp.proto.http.trace_exporter import OTLPSpanExporter
 from opentelemetry.propagate import set_global_textmap
 from opentelemetry.sdk.resources import Resource
 from opentelemetry.sdk.trace import TracerProvider
 from opentelemetry.sdk.trace.export import BatchSpanProcessor, ConsoleSpanExporter
 from opentelemetry.trace.propagation.tracecontext import TraceContextTextMapPropagator
-from opentelemetry.exporter.otlp.proto.http.trace_exporter import OTLPSpanExporter
+from pydantic import BaseModel, ConfigDict
 
-from mcp_agent.config import get_settings
-from mcp_agent.config import Settings
-from mcp_agent.executor.executor import Executor
+from mcp_agent.config import Settings, get_settings
 from mcp_agent.executor.decorator_registry import (
     DecoratorRegistry,
     register_asyncio_decorators,
 )
+from mcp_agent.executor.executor import AsyncioExecutor, Executor
 from mcp_agent.executor.task_registry import ActivityRegistry
-from mcp_agent.executor.executor import AsyncioExecutor
-
 from mcp_agent.logging.events import EventFilter
-from mcp_agent.logging.logger import LoggingConfig
+from mcp_agent.logging.logger import LoggingConfig, get_logger
 from mcp_agent.logging.transport import create_transport
 from mcp_agent.mcp_server_registry import ServerRegistry
-from mcp_agent.workflows.llm.llm_selector import ModelSelector
-from mcp_agent.logging.logger import get_logger
-
 
 if TYPE_CHECKING:
-    from mcp_agent.human_input.types import HumanInputCallback
     from mcp_agent.executor.workflow_signal import SignalWaitCallback
+    from mcp_agent.human_input.types import HumanInputCallback
 else:
     # Runtime placeholders for the types
     HumanInputCallback = Any
@@ -58,7 +50,6 @@ class Context(BaseModel):
     human_input_handler: Optional[HumanInputCallback] = None
     signal_notification: Optional[SignalWaitCallback] = None
     upstream_session: Optional[ServerSession] = None  # TODO: saqadri - figure this out
-    model_selector: Optional[ModelSelector] = None
 
     # Registries
     server_registry: Optional[ServerRegistry] = None
@@ -73,7 +64,7 @@ class Context(BaseModel):
     )
 
 
-async def configure_otel(config: "Settings"):
+async def configure_otel(config: "Settings") -> None:
     """
     Configure OpenTelemetry based on the application config.
     """
@@ -114,9 +105,7 @@ async def configure_otel(config: "Settings"):
         tracer_provider.add_span_processor(BatchSpanProcessor(exporter))
 
         if config.otel.console_debug:
-            tracer_provider.add_span_processor(
-                BatchSpanProcessor(ConsoleSpanExporter())
-            )
+            tracer_provider.add_span_processor(BatchSpanProcessor(ConsoleSpanExporter()))
     else:
         # Default to console exporter in development
         tracer_provider.add_span_processor(BatchSpanProcessor(ConsoleSpanExporter()))
@@ -125,7 +114,7 @@ async def configure_otel(config: "Settings"):
     trace.set_tracer_provider(tracer_provider)
 
 
-async def configure_logger(config: "Settings"):
+async def configure_logger(config: "Settings") -> None:
     """
     Configure logging and tracing based on the application config.
     """
@@ -141,7 +130,7 @@ async def configure_logger(config: "Settings"):
     )
 
 
-async def configure_usage_telemetry(_config: "Settings"):
+async def configure_usage_telemetry(_config: "Settings") -> None:
     """
     Configure usage telemetry based on the application config.
     TODO: saqadri - implement usage tracking
@@ -167,9 +156,7 @@ async def configure_executor(config: "Settings"):
         return executor
 
 
-async def initialize_context(
-    config: Optional[Union["Settings", str]] = None, store_globally: bool = False
-):
+async def initialize_context(config: Optional[Union["Settings", str]] = None, store_globally: bool = False):
     """
     Initialize the global application context.
     """
@@ -204,7 +191,7 @@ async def initialize_context(
     return context
 
 
-async def cleanup_context():
+async def cleanup_context() -> None:
     """
     Cleanup the global application context.
     """

mcp_agent/context_dependent.py

@@ -1,6 +1,5 @@
 from contextlib import contextmanager
-from typing import Optional, TYPE_CHECKING
-
+from typing import TYPE_CHECKING, Any, Optional
 
 if TYPE_CHECKING:
     from mcp_agent.context import Context
@@ -12,7 +11,7 @@ class ContextDependent:
     Provides both global fallback and instance-specific context support.
     """
 
-    def __init__(self, context: Optional["Context"] = None, **kwargs):
+    def __init__(self, context: Optional["Context"] = None, **kwargs: dict[str, Any]) -> None:
         self._context = context
         super().__init__(**kwargs)
 
@@ -32,10 +31,7 @@ class ContextDependent:
 
             return get_current_context()
         except Exception as e:
-            raise RuntimeError(
-                f"No context available for {self.__class__.__name__}. "
-                "Either initialize MCPApp first or pass context explicitly."
-            ) from e
+            raise RuntimeError(f"No context available for {self.__class__.__name__}. Either initialize MCPApp first or pass context explicitly.") from e
 
     @contextmanager
     def use_context(self, context: "Context"):