Function main

mcp_demo.py:171–290

()

async def main():
    print("\n" + "="*60)
    print("MyAgent Demo - Function-based + MCP Tools")
    print("="*60)

    # Switch between different configs to test:
    # - CONFIG: both local and remote tools
    # - CONFIG_NO_LOCAL: only remote MCP tools
    # - CONFIG_LOCAL_ONLY: only local function tools
    active_config = CONFIG_LOCAL_ONLY  # Change this to test different configurations

    try:
        # Step 1: Initialize tools from config
        print("\n" + "="*60)
        print("TOOL INITIALIZATION")
        print("="*60)

        tools_config = active_config.get("tools", {})

        # Initialize local function-based tools if enabled
        if tools_config.get("local", True):
            print("\n[✓] Loading local function-based tools...")
            init_tools()
            print("    Local tools initialized successfully")
        else:
            print("\n[✗] Local tools DISABLED by configuration")

        # Initialize remote MCP tools if configured
        remote_servers = tools_config.get("remote", [])
        if remote_servers:
            print(f"\n[✓] Found {len(remote_servers)} remote MCP server(s) in config:")
            for server in remote_servers:
                print(f"    - {server['id']}: {server['url']}")
            await init_mcp_tools(remote_servers=remote_servers)
        else:
            print("\n[✗] No remote MCP servers configured")

        # Get registry statistics
        registry = get_registry()
        print("\n" + "="*60)
        print("TOOL REGISTRY STATISTICS")
        print("="*60)
        print(f"Total tools loaded: {registry.total_tool_count()}")
        print(f"  - Function tools: {len(registry)}")
        print(f"  - MCP tools: {registry.total_tool_count() - len(registry)}")
        print(f"\nRegistered tool names:")
        for name in sorted(registry.get_all_names()):
            print(f"  - {name}")
        
        # Step 2: Initialize model
        print("\n" + "="*60)
        print("MODEL & AGENT INITIALIZATION")
        print("="*60)
        print("\nInitializing model...")
        model = OpenAIModel(
            api_key=os.getenv("OPENAI_API_KEY"),
            model_name="gpt-4o",
            temperature=0.7