fix: timeout

Author: echarles

dev/content/notebook.ipynb

@@ -3,7 +3,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "510c04f2-5e6e-4a91-8af5-f1be55e74312",
+   "id": "29a9d2af-9c57-449d-bbcb-2df77dfe9dfe",
    "metadata": {},
    "outputs": [],
    "source": []

examples/jupyter-repl/Makefile

@@ -0,0 +1,35 @@
+.PHONY: help jupyterlab repl clean
+
+# Default target
+# anthropic:claude-sonnet-4-20250514
+# azure-openai:gpt-4o-mini
+help:
+	@echo "Jupyter MCP Example - Makefile"
+	@echo ""
+	@echo "Available targets:"
+	@echo "  jupyterlab  - Start JupyterLab server on http://localhost:8888"
+	@echo "  repl        - Start jupyter-ai-agents REPL connected to Jupyter MCP server"
+	@echo "  clean       - Clean up temporary files"
+	@echo ""
+	@echo "Usage:"
+	@echo "  Terminal 1: make jupyterlab    # Start JupyterLab with MCP server"
+	@echo "  Terminal 2: make repl          # Connect REPL to Jupyter MCP server"
+	@echo ""
+	@echo "Note: The REPL connects to jupyter-mcp-server at http://localhost:8888/mcp"
+	@echo ""
+
+jupyterlab: ## jupyterlab
+	jupyter lab \
+		--port 8888 \
+		--ServerApp.root_dir ./../../dev/content \
+		--IdentityProvider.token=
+
+# Start REPL connected to MCP Jupyter
+repl:
+	@echo "🤖 Starting jupyter-ai-agents REPL connected to Jupyter Server"
+	@echo "Make sure the Jupyter server is running first (make jupyterlab in another terminal)"
+	@echo ""
+	jupyter-ai-agents repl \
+	  --verbose \
+	  --mcp-servers http://localhost:8888/mcp \
+	  --model anthropic:claude-sonnet-4-20250514

examples/jupyter-repl/README.md

@@ -0,0 +1,11 @@
+<!--
+  ~ Copyright (c) 2023-2024 Datalayer, Inc.
+  ~
+  ~ BSD 3-Clause License
+-->
+
+[![Datalayer](https://assets.datalayer.tech/datalayer-25.svg)](https://datalayer.ai)
+
+[![Become a Sponsor](https://img.shields.io/static/v1?label=Become%20a%20Sponsor&message=%E2%9D%A4&logo=GitHub&style=flat&color=1ABC9C)](https://github.com/sponsors/datalayer)
+
+# Jupyter REPL Example

jupyter_ai_agents/cli/app.py

@@ -372,10 +372,8 @@ async def list_tools_async():
             from jupyter_ai_agents.tools import MCPServerStreamableHTTP
 
             server_urls = [s.strip() for s in mcp_servers.split(',')]
-            typer.echo("\n🔧 Available MCP Tools:")
 
             for server_url in server_urls:
-                typer.echo(f"\n   Connecting to: {server_url}")
                 try:
                     mcp_client = MCPServerStreamableHTTP(server_url)
 
@@ -384,32 +382,33 @@ async def list_tools_async():
                         tools = await mcp_client.list_tools()
 
                         if not tools or len(tools) == 0:
-                            typer.echo("     No tools available")
+                            typer.echo("\n  No tools available")
                             continue
 
-                        typer.echo(f"     Found {len(tools)} tools:")
+                        typer.echo(f"\n  Available Tools ({len(tools)}):")
                         for tool in tools:
                             name = tool.name
                             description = tool.description or ""
                             schema = tool.inputSchema
 
+                            # Build parameter list
                             params = []
                             if schema and "properties" in schema:
                                 for param_name, param_info in schema["properties"].items():
                                     param_type = param_info.get("type", "any")
                                     params.append(f"{param_name}: {param_type}")
 
                             param_str = f"({', '.join(params)})" if params else "()"
-                            desc_first_line = description.split('\n')[0] if description else ""
-                            typer.echo(f"     • {name}{param_str} - {desc_first_line}")
+                            desc_first_line = description.split('\n')[0] if description else "No description"
+                            typer.echo(f"    • {name}{param_str} - {desc_first_line}")
                 except Exception as e:
                     logger.warning(f"Could not connect to {server_url}: {e}")
-                    typer.echo(f"     ⚠️  Could not connect: {e}")
+                    typer.echo(f"\n  ⚠️  Could not list tools from {server_url}")
 
         except Exception as e:
             logger.warning(f"Could not list tools: {e}")
-            typer.echo(f"\n⚠️  Could not list tools: {e}")
-            typer.echo("   The agent will still work with available tools\n")
+            typer.echo(f"\n  ⚠️  Could not list tools: {e}")
+
 
     try:
         from pydantic_ai import Agent
@@ -460,6 +459,12 @@ async def list_tools_async():
                 )
                 model_display_name = model  # azure-openai:deployment-name
                 logger.info(f"Using Azure OpenAI deployment: {deployment_name}")
+            elif model.startswith('anthropic:'):
+                # Parse anthropic:model-name format and use create_model_with_provider
+                model_name_part = model.split(':', 1)[1]
+                model_obj = create_model_with_provider('anthropic', model_name_part, timeout)
+                model_display_name = model
+                logger.info(f"Using Anthropic model: {model_name_part} (timeout: {timeout}s)")
             else:
                 model_obj = model
                 model_display_name = model
@@ -485,7 +490,17 @@ async def list_tools_async():
             mcp_client = MCPServerStreamableHTTP(server_url)
             toolsets.append(mcp_client)
 
-        # List tools before starting the agent (separate asyncio.run call)
+        # Display welcome message
+        typer.echo("="*70)
+        typer.echo("🪐 ✨ Jupyter AI Agents - Interactive REPL")
+        typer.echo("="*70)
+        typer.echo(f"Model: {model_display_name}")
+        
+        typer.echo(f"MCP Servers: {len(server_urls)} connected")
+        for server_url in server_urls:
+            typer.echo(f"  - {server_url}")
+        
+        # List tools inline in welcome message
         asyncio.run(list_tools_async())
 
         # Create default system prompt if not provided
@@ -506,17 +521,6 @@ async def list_tools_async():
             system_prompt=instructions,
         )
 
-        # Display welcome message
-        typer.echo("="*70)
-        typer.echo("🪐 ✨ Jupyter AI Agents - Interactive REPL")
-        typer.echo("="*70)
-        typer.echo(f"Model: {model_display_name}")
-        
-        server_urls = [s.strip() for s in mcp_servers.split(',')]
-        typer.echo(f"MCP Servers: {len(server_urls)} connected")
-        for server_url in server_urls:
-            typer.echo(f"  - {server_url}")
-        
         typer.echo("="*70)
         typer.echo("\nSpecial commands:")
         typer.echo("  /exit       - Exit the session")

jupyter_ai_agents/utils/model.py

@@ -114,17 +114,22 @@ def create_model_with_provider(
         return OpenAIChatModel(model_name, provider=azure_provider_with_timeout)
     elif model_provider.lower() == 'anthropic':
         from pydantic_ai.models.anthropic import AnthropicModel
+        from pydantic_ai.providers.anthropic import AnthropicProvider
         from anthropic import AsyncAnthropic
 
-        # Create Anthropic client with custom timeout
-        client = AsyncAnthropic(
-            timeout=http_timeout,
+        # Create Anthropic client with custom timeout and longer connect timeout
+        # Note: Many corporate networks block Anthropic API, use Azure/OpenAI if connection fails
+        anthropic_client = AsyncAnthropic(
+            timeout=httpx.Timeout(timeout, connect=60.0),  # Longer connect timeout for slow/restricted networks
             max_retries=2
         )
 
+        # Wrap in AnthropicProvider
+        anthropic_provider = AnthropicProvider(anthropic_client=anthropic_client)
+        
         return AnthropicModel(
             model_name,
-            provider=client
+            provider=anthropic_provider
         )
     elif model_provider.lower() in ['openai', 'github-copilot']:
         from pydantic_ai.models.openai import OpenAIChatModel