glaip-sdk 0.0.1b10__py3-none-any.whl → 0.0.3__py3-none-any.whl

glaip_sdk/cli/commands/init.py

@@ -13,6 +13,8 @@ import yaml
 from rich.console import Console
 from rich.panel import Panel
 
+from glaip_sdk import Client
+
 console = Console()
 
 
@@ -50,14 +52,22 @@ def init_command(no_scaffold, no_demo):
 
     # Create config directory
     config_dir = Path.home() / ".aip"
-    config_dir.mkdir(exist_ok=True)
+    try:
+        config_dir.mkdir(exist_ok=True)
+    except Exception as e:
+        console.print(f"⚠️  Warning: Could not create config directory: {e}")
+        return
 
     # Save configuration
     config = {"api_url": api_url, "api_key": api_key}
 
     config_file = config_dir / "config.yaml"
-    with open(config_file, "w") as f:
-        yaml.dump(config, f, default_flow_style=False)
+    try:
+        with open(config_file, "w") as f:
+            yaml.dump(config, f, default_flow_style=False)
+    except Exception as e:
+        console.print(f"⚠️  Warning: Could not save configuration: {e}")
+        return
 
     # Set secure file permissions (0600) - best effort on all platforms
     try:
@@ -77,8 +87,6 @@ def init_command(no_scaffold, no_demo):
         create_sample_resources = input("> ").strip().lower()
         if create_sample_resources in ["", "y", "yes"]:
             try:
-                from glaip_sdk import Client
-
                 # Set environment variables
                 os.environ["AIP_API_URL"] = api_url
                 os.environ["AIP_API_KEY"] = api_key
@@ -127,8 +135,6 @@ def init_command(no_scaffold, no_demo):
 def launch_interactive_demo():
     """Launch interactive demo with sample agent."""
     try:
-        from glaip_sdk import Client
-
         console.print(
             Panel(
                 "[bold green]Interactive Demo[/bold green]\nType to talk to hello-world. Ctrl+C to exit.",
@@ -152,6 +158,9 @@ def launch_interactive_demo():
                 if user_input.lower() in ["exit", "quit", "bye"]:
                     break
 
+                if not user_input:  # Skip empty inputs
+                    continue
+
                 response = agent.run(user_input)
                 console.print(f"🤖 {response}")
 

glaip_sdk/cli/commands/mcps.py

@@ -11,14 +11,13 @@ from rich.console import Console
 from rich.panel import Panel
 from rich.table import Table
 
-from glaip_sdk.utils import is_uuid
-
 from ..utils import (
+    coerce_to_row,
     get_client,
-    handle_ambiguous_resource,
     output_flags,
     output_list,
     output_result,
+    resolve_resource,
 )
 
 console = Console()
@@ -32,25 +31,14 @@ def mcps_group():
 
 def _resolve_mcp(ctx, client, ref, select=None):
     """Resolve MCP reference (ID or name) with ambiguity handling."""
-    if is_uuid(ref):
-        return client.mcps.get_mcp_by_id(ref)
-
-    # Find MCPs by name
-    matches = client.mcps.find_mcps(name=ref)
-    if not matches:
-        raise click.ClickException(f"MCP '{ref}' not found")
-
-    if len(matches) == 1:
-        return matches[0]
-
-    # Multiple matches - handle ambiguity
-    if select:
-        idx = int(select) - 1
-        if not (0 <= idx < len(matches)):
-            raise click.ClickException(f"--select must be 1..{len(matches)}")
-        return matches[idx]
-
-    return handle_ambiguous_resource(ctx, "MCP", ref, matches)
+    return resolve_resource(
+        ctx,
+        ref,
+        get_by_id=client.mcps.get_mcp_by_id,
+        find_by_name=client.mcps.find_mcps,
+        label="MCP",
+        select=select,
+    )
 
 
 @mcps_group.command(name="list")
@@ -71,24 +59,17 @@ def list_mcps(ctx):
 
         # Transform function for safe dictionary access
         def transform_mcp(mcp):
-            # Handle both dict and object formats
-            if isinstance(mcp, dict):
-                return {
-                    "id": str(mcp.get("id", "N/A")),
-                    "name": mcp.get("name", "N/A"),
-                    "config": str(mcp.get("config", "N/A"))[:50] + "..."
-                    if mcp.get("config")
-                    else "N/A",
-                }
-            else:
-                # Fallback to attribute access
-                return {
-                    "id": str(getattr(mcp, "id", "N/A")),
-                    "name": getattr(mcp, "name", "N/A"),
-                    "config": str(getattr(mcp, "config", "N/A"))[:50] + "..."
-                    if getattr(mcp, "config", None)
-                    else "N/A",
-                }
+            row = coerce_to_row(mcp, ["id", "name", "config"])
+            # Ensure id is always a string
+            row["id"] = str(row["id"])
+            # Truncate config field for display
+            if row["config"] != "N/A":
+                row["config"] = (
+                    str(row["config"])[:50] + "..."
+                    if len(str(row["config"])) > 50
+                    else str(row["config"])
+                )
+            return row
 
         output_list(ctx, mcps, "🔌 Available MCPs", columns, transform_mcp)
 
@@ -118,13 +99,14 @@ def create(ctx, name, transport, description, config):
 
         mcp = client.mcps.create_mcp(
             name=name,
-            type="server",  # Always server type
+            type="server",  # MCPs are always server type
             transport=transport,
             description=description,
             config=mcp_config,
         )
 
-        if ctx.obj.get("view") == "json":
+        view = (ctx.obj or {}).get("view", "rich")
+        if view == "json":
             click.echo(json.dumps(mcp.model_dump(), indent=2))
         else:
             # Rich output
@@ -139,7 +121,8 @@ def create(ctx, name, transport, description, config):
             console.print(panel)
 
     except Exception as e:
-        if ctx.obj.get("view") == "json":
+        view = (ctx.obj or {}).get("view", "rich")
+        if view == "json":
             click.echo(json.dumps({"error": str(e)}, indent=2))
         else:
             console.print(f"[red]Error creating MCP: {e}[/red]")
@@ -155,25 +138,8 @@ def get(ctx, mcp_ref):
     try:
         client = get_client(ctx)
 
-        # Get MCP by ID or name (shows all matches for names)
-        if is_uuid(mcp_ref):
-            mcp = client.mcps.get_mcp_by_id(mcp_ref)
-        else:
-            # Find MCPs by name
-            matches = client.mcps.find_mcps(name=mcp_ref)
-            if not matches:
-                raise click.ClickException(f"MCP '{mcp_ref}' not found")
-            elif len(matches) == 1:
-                mcp = matches[0]
-            else:
-                # Show all matches
-                console.print(
-                    f"[yellow]Multiple MCPs found with name '{mcp_ref}':[/yellow]"
-                )
-                for i, match in enumerate(matches, 1):
-                    console.print(f"  {i}. {match.name} (ID: {match.id})")
-                console.print("[yellow]Use the ID for unambiguous operations.[/yellow]")
-                return
+        # Resolve MCP using helper function
+        mcp = _resolve_mcp(ctx, client, mcp_ref)
 
         # Create result data with actual available fields
         result_data = {
@@ -202,53 +168,32 @@ def list_tools(ctx, mcp_ref):
     try:
         client = get_client(ctx)
 
-        # Get MCP by ID or name (shows all matches for names)
-        if is_uuid(mcp_ref):
-            mcp = client.get_mcp_by_id(mcp_ref)
-        else:
-            # Find MCPs by name
-            matches = client.find_mcps(name=mcp_ref)
-            if not matches:
-                raise click.ClickException(f"MCP '{mcp_ref}' not found")
-            elif len(matches) == 1:
-                mcp = matches[0]
-            else:
-                # Show all matches
-                console.print(
-                    f"[yellow]Multiple MCPs found with name '{mcp_ref}':[/yellow]"
-                )
-                for i, match in enumerate(matches, 1):
-                    console.print(f"  {i}. {match.name} (ID: {match.id})")
-                console.print("[yellow]Use the ID for unambiguous operations.[/yellow]")
-                return
+        # Resolve MCP using helper function
+        mcp = _resolve_mcp(ctx, client, mcp_ref)
 
         # Get tools from MCP
         tools = client.mcps.get_mcp_tools(mcp.id)
 
-        if ctx.obj.get("view") == "json":
-            click.echo(json.dumps(tools, indent=2))
-        else:  # rich output
-            if tools:
-                table = Table(
-                    title=f"🔧 Tools from MCP: {mcp.name}",
-                    show_header=True,
-                    header_style="bold magenta",
-                )
-                table.add_column("Name", style="cyan", no_wrap=True)
-                table.add_column("Description", style="green")
-                table.add_column("Type", style="yellow")
+        # Define table columns: (data_key, header, style, width)
+        columns = [
+            ("name", "Name", "cyan", None),
+            ("description", "Description", "green", 50),
+            ("type", "Type", "yellow", None),
+        ]
 
-                for tool in tools:
-                    table.add_row(
-                        tool.get("name", "N/A"),
-                        tool.get("description", "N/A")[:50] + "..."
-                        if len(tool.get("description", "")) > 50
-                        else tool.get("description", "N/A"),
-                        tool.get("type", "N/A"),
-                    )
-                console.print(table)
-            else:
-                console.print(f"[yellow]No tools found in MCP '{mcp.name}'[/yellow]")
+        # Transform function for safe dictionary access
+        def transform_tool(tool):
+            return {
+                "name": tool.get("name", "N/A"),
+                "description": tool.get("description", "N/A")[:47] + "..."
+                if len(tool.get("description", "")) > 47
+                else tool.get("description", "N/A"),
+                "type": tool.get("type", "N/A"),
+            }
+
+        output_list(
+            ctx, tools, f"🔧 Tools from MCP: {mcp.name}", columns, transform_tool
+        )
 
     except Exception as e:
         raise click.ClickException(str(e))
@@ -273,15 +218,17 @@ def tools_from_config(ctx, config_file):
         with open(config_file) as f:
             config = json.load(f)
 
-        if ctx.obj.get("view") != "json":
+        view = (ctx.obj or {}).get("view", "rich")
+        if view != "json":
             console.print(
                 f"[yellow]Fetching tools from MCP config in {config_file}...[/yellow]"
             )
 
         # Get tools from MCP config
-        tools = client.mcps.get_tools_from_mcp_config(config)
+        tools = client.mcps.get_mcp_tools_from_config(config)
 
-        if ctx.obj.get("view") == "json":
+        view = (ctx.obj or {}).get("view", "rich")
+        if view == "json":
             click.echo(json.dumps(tools, indent=2))
         else:  # rich output
             if tools:
@@ -328,7 +275,8 @@ def test_connection(ctx, config_file):
         with open(config_file) as f:
             config = json.load(f)
 
-        if ctx.obj.get("view") != "json":
+        view = (ctx.obj or {}).get("view", "rich")
+        if view != "json":
             console.print(
                 f"[yellow]Testing MCP connection with config from {config_file}...[/yellow]"
             )
@@ -336,7 +284,8 @@ def test_connection(ctx, config_file):
         # Test connection using config
         result = client.mcps.test_mcp_connection_from_config(config)
 
-        if ctx.obj.get("view") == "json":
+        view = (ctx.obj or {}).get("view", "rich")
+        if view == "json":
             click.echo(json.dumps(result, indent=2))
         else:
             success_panel = Panel(
@@ -363,25 +312,8 @@ def update(ctx, mcp_ref, name, description, config):
     try:
         client = get_client(ctx)
 
-        # Get MCP by ID or name (shows all matches for names)
-        if is_uuid(mcp_ref):
-            mcp = client.mcps.get_mcp_by_id(mcp_ref)
-        else:
-            # Find MCPs by name
-            matches = client.mcps.find_mcps(name=mcp_ref)
-            if not matches:
-                raise click.ClickException(f"MCP '{mcp_ref}' not found")
-            elif len(matches) == 1:
-                mcp = matches[0]
-            else:
-                # Show all matches
-                console.print(
-                    f"[yellow]Multiple MCPs found with name '{mcp_ref}':[/yellow]"
-                )
-                for i, match in enumerate(matches, 1):
-                    console.print(f"  {i}. {match.name} (ID: {match.id})")
-                console.print("[yellow]Use the ID for unambiguous operations.[/yellow]")
-                return
+        # Resolve MCP using helper function
+        mcp = _resolve_mcp(ctx, client, mcp_ref)
 
         # Build update data
         update_data = {}
@@ -399,9 +331,10 @@ def update(ctx, mcp_ref, name, description, config):
             raise click.ClickException("No update fields specified")
 
         # Update MCP
-        updated_mcp = client.mcps.update_mcp(mcp.id, update_data)
+        updated_mcp = client.mcps.update_mcp(mcp.id, **update_data)
 
-        if ctx.obj.get("view") == "json":
+        view = (ctx.obj or {}).get("view", "rich")
+        if view == "json":
             click.echo(json.dumps(updated_mcp.model_dump(), indent=2))
         else:
             console.print(
@@ -409,7 +342,8 @@ def update(ctx, mcp_ref, name, description, config):
             )
 
     except Exception as e:
-        if ctx.obj.get("view") == "json":
+        view = (ctx.obj or {}).get("view", "rich")
+        if view == "json":
             click.echo(json.dumps({"error": str(e)}, indent=2))
         else:
             console.print(f"[red]Error updating MCP: {e}[/red]")
@@ -426,37 +360,22 @@ def delete(ctx, mcp_ref, yes):
     try:
         client = get_client(ctx)
 
-        # Get MCP by ID or name (shows all matches for names)
-        if is_uuid(mcp_ref):
-            mcp = client.mcps.get_mcp_by_id(mcp_ref)
-        else:
-            # Find MCPs by name
-            matches = client.mcps.find_mcps(name=mcp_ref)
-            if not matches:
-                raise click.ClickException(f"MCP '{mcp_ref}' not found")
-            elif len(matches) == 1:
-                mcp = matches[0]
-            else:
-                # Show all matches
-                console.print(
-                    f"[yellow]Multiple MCPs found with name '{mcp_ref}':[/yellow]"
-                )
-                for i, match in enumerate(matches, 1):
-                    console.print(f"  {i}. {match.name} (ID: {match.id})")
-                console.print("[yellow]Use the ID for unambiguous operations.[/yellow]")
-                return
+        # Resolve MCP using helper function
+        mcp = _resolve_mcp(ctx, client, mcp_ref)
 
         # Confirm deletion
         if not yes and not click.confirm(
             f"Are you sure you want to delete MCP '{mcp.name}'?"
         ):
-            if ctx.obj.get("view") != "json":
+            view = (ctx.obj or {}).get("view", "rich")
+            if view != "json":
                 console.print("Deletion cancelled.")
             return
 
         client.mcps.delete_mcp(mcp.id)
 
-        if ctx.obj.get("view") == "json":
+        view = (ctx.obj or {}).get("view", "rich")
+        if view == "json":
             click.echo(
                 json.dumps(
                     {"success": True, "message": f"MCP '{mcp.name}' deleted"}, indent=2
@@ -466,7 +385,8 @@ def delete(ctx, mcp_ref, yes):
             console.print(f"[green]✅ MCP '{mcp.name}' deleted successfully[/green]")
 
     except Exception as e:
-        if ctx.obj.get("view") == "json":
+        view = (ctx.obj or {}).get("view", "rich")
+        if view == "json":
             click.echo(json.dumps({"error": str(e)}, indent=2))
         else:
             console.print(f"[red]Error deleting MCP: {e}[/red]")