glaip-sdk 0.0.3__py3-none-any.whl → 0.0.4__py3-none-any.whl

glaip_sdk/cli/commands/configure.py

@@ -11,10 +11,12 @@ from pathlib import Path
 import click
 import yaml
 from rich.console import Console
-from rich.panel import Panel
 from rich.table import Table
+from rich.text import Text
 
 from glaip_sdk import Client
+from glaip_sdk._version import __version__ as _SDK_VERSION
+from glaip_sdk.branding import AIPBranding
 
 console = Console()
 
@@ -79,7 +81,7 @@ def list_config():
             table.add_row(key, str(value))
 
     console.print(table)
-    console.print(f"\n📁 Config file: {CONFIG_FILE}")
+    console.print(Text(f"\n📁 Config file: {CONFIG_FILE}"))
 
 
 @config_group.command("set")
@@ -102,9 +104,9 @@ def set_config(key, value):
 
     if key == "api_key":
         masked_value = "***" + value[-4:] if len(value) > 4 else "***"
-        console.print(f"✅ Set {key} = {masked_value}")
+        console.print(Text(f"✅ Set {key} = {masked_value}"))
     else:
-        console.print(f"✅ Set {key} = {value}")
+        console.print(Text(f"✅ Set {key} = {value}"))
 
 
 @config_group.command("get")
@@ -115,7 +117,7 @@ def get_config(key):
     config = load_config()
 
     if key not in config:
-        console.print(f"[yellow]Configuration key '{key}' not found.[/yellow]")
+        console.print(Text(f"[yellow]Configuration key '{key}' not found.[/yellow]"))
         raise click.ClickException(f"Configuration key not found: {key}")
 
     value = config[key]
@@ -136,13 +138,13 @@ def unset_config(key):
     config = load_config()
 
     if key not in config:
-        console.print(f"[yellow]Configuration key '{key}' not found.[/yellow]")
+        console.print(Text(f"[yellow]Configuration key '{key}' not found.[/yellow]"))
         return
 
     del config[key]
     save_config(config)
 
-    console.print(f"✅ Removed {key} from configuration")
+    console.print(Text(f"✅ Removed {key} from configuration"))
 
 
 @config_group.command("reset")
@@ -168,13 +170,11 @@ def reset_config(force):
 
 def _configure_interactive():
     """Shared configuration logic for both configure commands."""
-    console.print(
-        Panel(
-            "[bold cyan]AIP Configuration[/bold cyan]\nConfigure your AIP CLI settings.",
-            title="🔧 Configuration Setup",
-            border_style="cyan",
-        )
+    # Display AIP welcome banner
+    branding = AIPBranding.create_from_sdk(
+        sdk_version=_SDK_VERSION, package_name="glaip-sdk"
     )
+    branding.display_welcome_panel(title="🔧 AIP Configuration")
 
     # Load existing config
     config = load_config()
@@ -208,7 +208,7 @@ def _configure_interactive():
     # Save configuration
     save_config(config)
 
-    console.print(f"\n✅ Configuration saved to: {CONFIG_FILE}")
+    console.print(Text(f"\n✅ Configuration saved to: {CONFIG_FILE}"))
 
     # Test the new configuration
     console.print("\n🔌 Testing connection...")
@@ -219,9 +219,9 @@ def _configure_interactive():
         # Try to list resources to test connection
         try:
             agents = client.list_agents()
-            console.print(f"✅ Connection successful! Found {len(agents)} agents")
+            console.print(Text(f"✅ Connection successful! Found {len(agents)} agents"))
         except Exception as e:
-            console.print(f"⚠️  Connection established but API call failed: {e}")
+            console.print(Text(f"⚠️  Connection established but API call failed: {e}"))
             console.print(
                 "   You may need to check your API permissions or network access"
             )
@@ -229,7 +229,7 @@ def _configure_interactive():
         client.close()
 
     except Exception as e:
-        console.print(f"❌ Connection failed: {e}")
+        console.print(Text(f"❌ Connection failed: {e}"))
         console.print("   Please check your API URL and key")
         console.print("   You can run 'aip status' later to test again")
 

glaip_sdk/cli/commands/init.py

@@ -11,32 +11,28 @@ from pathlib import Path
 import click
 import yaml
 from rich.console import Console
-from rich.panel import Panel
+from rich.text import Text
 
 from glaip_sdk import Client
+from glaip_sdk._version import __version__ as _SDK_VERSION
+from glaip_sdk.branding import AIPBranding
 
 console = Console()
 
 
 @click.command()
-@click.option("--no-scaffold", is_flag=True, help="Skip creating sample agent and tool")
-@click.option("--no-demo", is_flag=True, help="Don't launch interactive demo")
-def init_command(no_scaffold, no_demo):
+def init_command():
     """Initialize AIP project configuration."""
-
-    console.print(
-        Panel(
-            "[bold cyan]Welcome to AIP![/bold cyan]\nLet's set up your project.",
-            title="🚀 AIP Initialization",
-            border_style="cyan",
-        )
+    # Display AIP welcome banner
+    branding = AIPBranding.create_from_sdk(
+        sdk_version=_SDK_VERSION, package_name="glaip-sdk"
     )
+    branding.display_welcome_panel(title="🚀 AIP Initialization")
 
-    # Get configuration with better formatting
-    console.print("\n[bold]Step 1:[/bold] API Configuration")
+    # Get configuration
+    console.print("\n[bold]API Configuration[/bold]")
     console.print("─" * 50)
 
-    # Use built-in input for better control
     console.print(
         "\n[cyan]AIP API URL[/cyan] (default: https://your-aip-instance.com):"
     )
@@ -44,134 +40,54 @@ def init_command(no_scaffold, no_demo):
     if not api_url:
         api_url = "https://your-aip-instance.com"
 
-    console.print()  # Add spacing
-    console.print("[cyan]AIP API Key[/cyan]:")
+    console.print("\n[cyan]AIP API Key[/cyan]:")
     api_key = getpass.getpass("> ")
 
-    # Project configuration removed - not needed for AIP CLI
-
     # Create config directory
     config_dir = Path.home() / ".aip"
     try:
         config_dir.mkdir(exist_ok=True)
     except Exception as e:
-        console.print(f"⚠️  Warning: Could not create config directory: {e}")
+        console.print(Text(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"
     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}")
+        console.print(Text(f"⚠️  Warning: Could not save configuration: {e}"))
         return
 
     # Set secure file permissions (0600) - best effort on all platforms
     try:
         os.chmod(config_file, 0o600)
     except Exception:
-        # Best effort - may fail on Windows or other non-POSIX systems
-        pass
-
-    console.print(f"\n✅ Configuration saved to: {config_file}")
-
-    # Create sample agent and tool if requested
-    if not no_scaffold:
-        console.print("\n[bold]Step 2:[/bold] Sample Resources")
-        console.print("─" * 50)
-
-        console.print("\n[cyan]Create sample agent & tool?[/cyan] (Y/n):")
-        create_sample_resources = input("> ").strip().lower()
-        if create_sample_resources in ["", "y", "yes"]:
-            try:
-                # Set environment variables
-                os.environ["AIP_API_URL"] = api_url
-                os.environ["AIP_API_KEY"] = api_key
-
-                client = Client()
-
-                # Create sample agent
-                agent = client.create_agent(
-                    name="hello-world",
-                    instruction="You are a helpful AI assistant that says hello",
-                )
-                console.print(f"✅ Created sample agent: {agent.name} (id: {agent.id})")
-
-                # Create sample tool
-                tool_content = '''def hello_tool(name="World"):
-    """A simple hello tool."""
-    return f"Hello, {name}!"
-'''
-                with open("greeting_tool.py", "w") as f:
-                    f.write(tool_content)
-
-                tool = client.create_tool("greeting_tool.py", framework="langchain")
-                console.print(f"✅ Created sample tool: {tool.name} (id: {tool.id})")
-
-            except Exception as e:
-                console.print(f"⚠️  Warning: Could not create sample resources: {e}")
-
-    # Launch interactive demo if requested
-    if not no_demo:
-        console.print("\n[bold]Step 3:[/bold] Interactive Demo")
-        console.print("─" * 50)
-
-        console.print("\n[cyan]Start an interactive demo now?[/cyan] (Y/n):")
-        launch_demo = input("> ").strip().lower()
-        if launch_demo in ["", "y", "yes"]:
-            launch_interactive_demo()
-
-    console.print("\n🎉 [bold green]AIP initialization complete![/bold green]")
-    console.print(f"📁 Configuration: {config_file}")
-    console.print("💡 Next steps:")
-    console.print("   • Run 'aip agents list' to see your agents")
-    console.print("   • Run 'aip tools list' to see your tools")
-    console.print("   • Run 'aip status' to check connection")
+        pass  # Ignore permission errors
 
+    console.print(Text(f"\n✅ Configuration saved to: {config_file}"))
 
-def launch_interactive_demo():
-    """Launch interactive demo with sample agent."""
+    # Test the new configuration
+    console.print("\n🔌 Testing connection...")
     try:
-        console.print(
-            Panel(
-                "[bold green]Interactive Demo[/bold green]\nType to talk to hello-world. Ctrl+C to exit.",
-                title="🎮 Demo Mode",
-                border_style="green",
-            )
-        )
+        # Set environment variables
+        os.environ["AIP_API_URL"] = api_url
+        os.environ["AIP_API_KEY"] = api_key
 
         client = Client()
-        agents = client.find_agents(name="hello-world")
-        if not agents:
-            console.print("❌ Sample agent not found")
-            return
-
-        agent = agents[0]  # Use the first (and should be only) agent
-
-        while True:
-            try:
-                console.print("\n> ", end="")
-                user_input = input().strip()
-                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}")
-
-            except KeyboardInterrupt:
-                break
-            except Exception as e:
-                console.print(f"❌ Error: {e}")
-
+        agents = client.list_agents()
+        console.print(Text(f"✅ Connection successful! Found {len(agents)} agents"))
     except Exception as e:
-        console.print(f"⚠️  Could not launch demo: {e}")
-
+        console.print(Text(f"⚠️  Connection established but API call failed: {e}"))
+        console.print(
+            Text("   You may need to check your API permissions or network access")
+        )
 
-# Note: The init command should be registered in main.py, not here
-# This prevents circular imports
+    console.print("\n🎉 [bold green]AIP initialization complete![/bold green]")
+    console.print(f"📁 Configuration: {config_file}")
+    console.print("💡 Next steps:")
+    console.print("   • Run 'aip agents list' to see your agents")
+    console.print("   • Run 'aip tools list' to see your tools")
+    console.print("   • Run 'aip status' to check connection")

glaip_sdk/cli/commands/mcps.py

@@ -10,6 +10,7 @@ import click
 from rich.console import Console
 from rich.panel import Panel
 from rich.table import Table
+from rich.text import Text
 
 from ..utils import (
     coerce_to_row,
@@ -125,7 +126,7 @@ def create(ctx, name, transport, description, config):
         if view == "json":
             click.echo(json.dumps({"error": str(e)}, indent=2))
         else:
-            console.print(f"[red]Error creating MCP: {e}[/red]")
+            console.print(Text(f"[red]Error creating MCP: {e}[/red]"))
         raise click.ClickException(str(e))
 
 
@@ -251,7 +252,7 @@ def tools_from_config(ctx, config_file):
                     )
                 console.print(table)
             else:
-                console.print("[yellow]No tools found in MCP config[/yellow]")
+                console.print(Text("[yellow]No tools found in MCP config[/yellow]"))
 
     except Exception as e:
         raise click.ClickException(str(e))
@@ -278,7 +279,9 @@ def test_connection(ctx, config_file):
         view = (ctx.obj or {}).get("view", "rich")
         if view != "json":
             console.print(
-                f"[yellow]Testing MCP connection with config from {config_file}...[/yellow]"
+                Text(
+                    f"[yellow]Testing MCP connection with config from {config_file}...[/yellow]"
+                )
             )
 
         # Test connection using config
@@ -338,7 +341,7 @@ def update(ctx, mcp_ref, name, description, config):
             click.echo(json.dumps(updated_mcp.model_dump(), indent=2))
         else:
             console.print(
-                f"[green]✅ MCP '{updated_mcp.name}' updated successfully[/green]"
+                Text(f"[green]✅ MCP '{updated_mcp.name}' updated successfully[/green]")
             )
 
     except Exception as e:
@@ -346,7 +349,7 @@ def update(ctx, mcp_ref, name, description, config):
         if view == "json":
             click.echo(json.dumps({"error": str(e)}, indent=2))
         else:
-            console.print(f"[red]Error updating MCP: {e}[/red]")
+            console.print(Text(f"[red]Error updating MCP: {e}[/red]"))
         raise click.ClickException(str(e))
 
 
@@ -369,7 +372,7 @@ def delete(ctx, mcp_ref, yes):
         ):
             view = (ctx.obj or {}).get("view", "rich")
             if view != "json":
-                console.print("Deletion cancelled.")
+                console.print(Text("Deletion cancelled."))
             return
 
         client.mcps.delete_mcp(mcp.id)
@@ -382,12 +385,14 @@ def delete(ctx, mcp_ref, yes):
                 )
             )
         else:
-            console.print(f"[green]✅ MCP '{mcp.name}' deleted successfully[/green]")
+            console.print(
+                Text(f"[green]✅ MCP '{mcp.name}' deleted successfully[/green]")
+            )
 
     except Exception as e:
         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]")
+            console.print(Text(f"[red]Error deleting MCP: {e}[/red]"))
         raise click.ClickException(str(e))

glaip_sdk/cli/commands/tools.py

@@ -10,6 +10,7 @@ import re
 import click
 from rich.console import Console
 from rich.panel import Panel
+from rich.text import Text
 
 from ..utils import (
     coerce_to_row,
@@ -209,7 +210,7 @@ def create(ctx, file_arg, file, name, description, tags):
         if ctx.obj.get("view") == "json":
             click.echo(json.dumps({"error": str(e)}, indent=2))
         else:
-            console.print(f"[red]Error creating tool: {e}[/red]")
+            console.print(Text(f"[red]Error creating tool: {e}[/red]"))
         raise click.ClickException(str(e))
 
 
@@ -276,15 +277,15 @@ def update(ctx, tool_id, file, description, tags):
             # Update code
             updated_tool = tool.update(file_path=file)
             if ctx.obj.get("view") != "json":
-                console.print(f"[green]✓[/green] Tool code updated from {file}")
+                console.print(Text(f"[green]✓[/green] Tool code updated from {file}"))
         elif update_data:
             # Update metadata
             updated_tool = tool.update(**update_data)
             if ctx.obj.get("view") != "json":
-                console.print("[green]✓[/green] Tool metadata updated")
+                console.print(Text("[green]✓[/green] Tool metadata updated"))
         else:
             if ctx.obj.get("view") != "json":
-                console.print("[yellow]No updates specified[/yellow]")
+                console.print(Text("[yellow]No updates specified[/yellow]"))
             return
 
         if ctx.obj.get("view") == "json":
@@ -298,7 +299,7 @@ def update(ctx, tool_id, file, description, tags):
         if ctx.obj.get("view") == "json":
             click.echo(json.dumps({"error": str(e)}, indent=2))
         else:
-            console.print(f"[red]Error updating tool: {e}[/red]")
+            console.print(Text(f"[red]Error updating tool: {e}[/red]"))
         raise click.ClickException(str(e))
 
 
@@ -323,7 +324,7 @@ def delete(ctx, tool_id, yes):
             f"Are you sure you want to delete tool '{tool.name}'?"
         ):
             if ctx.obj.get("view") != "json":
-                console.print("Deletion cancelled.")
+                console.print(Text("Deletion cancelled."))
             return
 
         tool.delete()
@@ -336,13 +337,15 @@ def delete(ctx, tool_id, yes):
                 )
             )
         else:
-            console.print(f"[green]✅ Tool '{tool.name}' deleted successfully[/green]")
+            console.print(
+                Text(f"[green]✅ Tool '{tool.name}' deleted successfully[/green]")
+            )
 
     except Exception as e:
         if ctx.obj.get("view") == "json":
             click.echo(json.dumps({"error": str(e)}, indent=2))
         else:
-            console.print(f"[red]Error deleting tool: {e}[/red]")
+            console.print(Text(f"[red]Error deleting tool: {e}[/red]"))
         raise click.ClickException(str(e))
 
 
@@ -441,5 +444,5 @@ def upload_update(ctx, tool_id, file, name, description, tags):
         if ctx.obj.get("view") == "json":
             click.echo(json.dumps({"error": str(e)}, indent=2))
         else:
-            console.print(f"[red]Error updating tool: {e}[/red]")
+            console.print(Text(f"[red]Error updating tool: {e}[/red]"))
         raise click.ClickException(str(e))

glaip_sdk/cli/main.py

@@ -15,6 +15,7 @@ from rich.table import Table
 
 from glaip_sdk import Client
 from glaip_sdk._version import __version__ as _SDK_VERSION
+from glaip_sdk.branding import AIPBranding
 from glaip_sdk.cli.commands.agents import agents_group
 from glaip_sdk.cli.commands.configure import (
     config_group,
@@ -49,11 +50,12 @@ def main(ctx, api_url, api_key, timeout, view, no_tty):
     agents, tools, MCPs, and more.
 
     Examples:
+        aip version                     # Show detailed version info
         aip configure                   # Configure credentials
+        aip init                         # Initialize configuration
         aip agents list                   # List all agents
         aip tools create my_tool.py      # Create a new tool
-        aip agents run my-agent "Hello"  # Run an agent
-        aip init                         # Initialize configuration
+        aip agents run my-agent "Hello world"  # Run an agent
     """
 
     # Store configuration in context
@@ -89,6 +91,12 @@ def status(ctx):
     try:
         console = Console()
 
+        # Display AIP status banner
+        branding = AIPBranding.create_from_sdk(
+            sdk_version=_SDK_VERSION, package_name="glaip-sdk"
+        )
+        branding.display_status_banner("ready")
+
         # Load config from file and merge with context
         file_config = load_config()
         context_config = ctx.obj or {}
@@ -195,7 +203,10 @@ def status(ctx):
 @main.command()
 def version():
     """Show version information."""
-    click.echo(f"aip version {_SDK_VERSION}")
+    branding = AIPBranding.create_from_sdk(
+        sdk_version=_SDK_VERSION, package_name="glaip-sdk"
+    )
+    branding.display_version_panel()
 
 
 @main.command()