airweave-cli 0.1.0__tar.gz

airweave_cli-0.1.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Airweave
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

airweave_cli-0.1.0/PKG-INFO

@@ -0,0 +1,194 @@
+Metadata-Version: 2.1
+Name: airweave-cli
+Version: 0.1.0
+Summary: The Airweave CLI for developers and AI agents.
+License: MIT
+Keywords: airweave,cli,search,ai,agents
+Author: Airweave
+Author-email: hello@airweave.ai
+Requires-Python: >=3.9,<4.0
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Classifier: Typing :: Typed
+Requires-Dist: airweave-sdk (>=0.9)
+Requires-Dist: rich (>=13,<14)
+Requires-Dist: typer[all] (>=0.12,<0.13)
+Description-Content-Type: text/markdown
+
+# Airweave CLI
+
+[![pypi](https://img.shields.io/pypi/v/airweave-cli)](https://pypi.python.org/pypi/airweave-cli)
+
+A command-line interface for [Airweave](https://airweave.ai) — search across your connected sources from any terminal. Built for developers and AI agents.
+
+## Installation
+
+```sh
+pip install airweave-cli
+```
+
+This installs the `airweave` binary.
+
+## Quick Start
+
+```sh
+# Set your API key
+export AIRWEAVE_API_KEY="your-api-key"
+
+# Search a collection
+airweave search "quarterly revenue figures" --collection finance-data
+
+# List collections
+airweave collections list
+```
+
+## Authentication
+
+The CLI resolves credentials in this order:
+
+1. `AIRWEAVE_API_KEY` environment variable
+2. `~/.airweave/config.json` (saved via `airweave auth login`)
+
+For interactive setup:
+
+```sh
+airweave auth login
+```
+
+This prompts for your API key, validates it, and saves it locally.
+
+### Self-hosted
+
+Point the CLI at your own Airweave instance:
+
+```sh
+export AIRWEAVE_BASE_URL="https://airweave.internal.corp.com"
+```
+
+Or set it during `airweave auth login`.
+
+## Commands
+
+### Search
+
+```sh
+airweave search "<query>" --collection <id> --top-k 5 --format json
+```
+
+| Flag | Short | Default | Description |
+|------|-------|---------|-------------|
+| `--collection` | `-c` | `$AIRWEAVE_COLLECTION` | Collection readable ID |
+| `--top-k` | `-k` | `10` | Number of results |
+| `--format` | `-f` | `json` | Output format: `json` or `text` |
+
+JSON output (default) writes pure JSON to stdout — nothing else. Pipe it:
+
+```sh
+airweave search "query" -c my-collection | jq '.results[0].md_content'
+```
+
+### Auth
+
+```sh
+airweave auth login     # interactive setup
+airweave auth status    # show current auth state
+airweave auth logout    # clear saved credentials
+```
+
+### Collections
+
+```sh
+airweave collections list                           # list all
+airweave collections create --name "My Data"        # create
+airweave collections get my-data-x7k9m              # get details
+```
+
+### Sources
+
+```sh
+airweave sources list --collection my-data          # list source connections
+airweave sources sync <source-connection-id>        # trigger sync
+airweave sources sync <id> --force                  # full re-sync
+```
+
+## Agent Usage
+
+The CLI is designed for non-interactive use by AI agents. Every command:
+
+- Outputs clean JSON to stdout (default)
+- Sends all errors to stderr
+- Uses correct exit codes (0 = success, 1 = error)
+
+### Environment variables
+
+```sh
+export AIRWEAVE_API_KEY="sk-..."
+export AIRWEAVE_COLLECTION="my-knowledge-base"
+export AIRWEAVE_BASE_URL="https://api.airweave.ai"  # optional
+```
+
+### Piping
+
+```sh
+# Get the top result's content
+airweave search "how to reset password" | jq -r '.results[0].md_content'
+
+# Get the AI-generated answer
+airweave search "what is our refund policy" | jq -r '.completion'
+
+# List collection IDs
+airweave collections list | jq -r '.[].readable_id'
+```
+
+### CI / Cloud Functions
+
+```yaml
+# GitHub Actions example
+- name: Search Airweave
+  env:
+    AIRWEAVE_API_KEY: ${{ secrets.AIRWEAVE_API_KEY }}
+    AIRWEAVE_COLLECTION: docs
+  run: |
+    result=$(airweave search "deployment guide")
+    echo "$result" | jq '.results[:3]'
+```
+
+No SDK imports, no MCP server, no config files — just an env var and a shell.
+
+## Configuration
+
+Config file location: `~/.airweave/config.json`
+
+```json
+{
+  "api_key": "sk-...",
+  "base_url": "https://api.airweave.ai",
+  "collection": "my-default-collection"
+}
+```
+
+Resolution order for all settings:
+
+1. CLI flag (e.g. `--collection`)
+2. Environment variable (e.g. `AIRWEAVE_COLLECTION`)
+3. Config file
+4. Default / error
+
+## Contributing
+
+```sh
+git clone https://github.com/airweave-ai/cli.git
+cd cli
+poetry install
+poetry run airweave --help
+```
+

airweave_cli-0.1.0/README.md

@@ -0,0 +1,167 @@
+# Airweave CLI
+
+[![pypi](https://img.shields.io/pypi/v/airweave-cli)](https://pypi.python.org/pypi/airweave-cli)
+
+A command-line interface for [Airweave](https://airweave.ai) — search across your connected sources from any terminal. Built for developers and AI agents.
+
+## Installation
+
+```sh
+pip install airweave-cli
+```
+
+This installs the `airweave` binary.
+
+## Quick Start
+
+```sh
+# Set your API key
+export AIRWEAVE_API_KEY="your-api-key"
+
+# Search a collection
+airweave search "quarterly revenue figures" --collection finance-data
+
+# List collections
+airweave collections list
+```
+
+## Authentication
+
+The CLI resolves credentials in this order:
+
+1. `AIRWEAVE_API_KEY` environment variable
+2. `~/.airweave/config.json` (saved via `airweave auth login`)
+
+For interactive setup:
+
+```sh
+airweave auth login
+```
+
+This prompts for your API key, validates it, and saves it locally.
+
+### Self-hosted
+
+Point the CLI at your own Airweave instance:
+
+```sh
+export AIRWEAVE_BASE_URL="https://airweave.internal.corp.com"
+```
+
+Or set it during `airweave auth login`.
+
+## Commands
+
+### Search
+
+```sh
+airweave search "<query>" --collection <id> --top-k 5 --format json
+```
+
+| Flag | Short | Default | Description |
+|------|-------|---------|-------------|
+| `--collection` | `-c` | `$AIRWEAVE_COLLECTION` | Collection readable ID |
+| `--top-k` | `-k` | `10` | Number of results |
+| `--format` | `-f` | `json` | Output format: `json` or `text` |
+
+JSON output (default) writes pure JSON to stdout — nothing else. Pipe it:
+
+```sh
+airweave search "query" -c my-collection | jq '.results[0].md_content'
+```
+
+### Auth
+
+```sh
+airweave auth login     # interactive setup
+airweave auth status    # show current auth state
+airweave auth logout    # clear saved credentials
+```
+
+### Collections
+
+```sh
+airweave collections list                           # list all
+airweave collections create --name "My Data"        # create
+airweave collections get my-data-x7k9m              # get details
+```
+
+### Sources
+
+```sh
+airweave sources list --collection my-data          # list source connections
+airweave sources sync <source-connection-id>        # trigger sync
+airweave sources sync <id> --force                  # full re-sync
+```
+
+## Agent Usage
+
+The CLI is designed for non-interactive use by AI agents. Every command:
+
+- Outputs clean JSON to stdout (default)
+- Sends all errors to stderr
+- Uses correct exit codes (0 = success, 1 = error)
+
+### Environment variables
+
+```sh
+export AIRWEAVE_API_KEY="sk-..."
+export AIRWEAVE_COLLECTION="my-knowledge-base"
+export AIRWEAVE_BASE_URL="https://api.airweave.ai"  # optional
+```
+
+### Piping
+
+```sh
+# Get the top result's content
+airweave search "how to reset password" | jq -r '.results[0].md_content'
+
+# Get the AI-generated answer
+airweave search "what is our refund policy" | jq -r '.completion'
+
+# List collection IDs
+airweave collections list | jq -r '.[].readable_id'
+```
+
+### CI / Cloud Functions
+
+```yaml
+# GitHub Actions example
+- name: Search Airweave
+  env:
+    AIRWEAVE_API_KEY: ${{ secrets.AIRWEAVE_API_KEY }}
+    AIRWEAVE_COLLECTION: docs
+  run: |
+    result=$(airweave search "deployment guide")
+    echo "$result" | jq '.results[:3]'
+```
+
+No SDK imports, no MCP server, no config files — just an env var and a shell.
+
+## Configuration
+
+Config file location: `~/.airweave/config.json`
+
+```json
+{
+  "api_key": "sk-...",
+  "base_url": "https://api.airweave.ai",
+  "collection": "my-default-collection"
+}
+```
+
+Resolution order for all settings:
+
+1. CLI flag (e.g. `--collection`)
+2. Environment variable (e.g. `AIRWEAVE_COLLECTION`)
+3. Config file
+4. Default / error
+
+## Contributing
+
+```sh
+git clone https://github.com/airweave-ai/cli.git
+cd cli
+poetry install
+poetry run airweave --help
+```

airweave_cli-0.1.0/airweave_cli/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.1.0"

airweave_cli-0.1.0/airweave_cli/commands/auth.py

@@ -0,0 +1,105 @@
+from __future__ import annotations
+
+import json
+import os
+from enum import Enum
+
+import typer
+from rich.console import Console
+
+from airweave_cli.config import (
+    clear_config,
+    load_config,
+    save_config,
+)
+
+app = typer.Typer(name="auth", help="Manage authentication.", no_args_is_help=True)
+
+stderr = Console(stderr=True)
+stdout = Console()
+
+
+class OutputFormat(str, Enum):
+    json = "json"
+    text = "text"
+
+
+@app.command()
+def login() -> None:
+    """Interactive login: save your API key to ~/.airweave/config.json."""
+    api_key = typer.prompt("API key", hide_input=True)
+    base_url = typer.prompt(
+        "Base URL",
+        default="https://api.airweave.ai",
+        show_default=True,
+    )
+    collection = typer.prompt(
+        "Default collection (readable_id)",
+        default="",
+        show_default=False,
+    )
+
+    # Validate the key
+    try:
+        from airweave import AirweaveSDK
+
+        client = AirweaveSDK(api_key=api_key, base_url=base_url)
+        client.collections.list(limit=1)
+    except Exception as exc:
+        stderr.print(f"[red]Authentication failed:[/red] {exc}")
+        raise typer.Exit(code=1)
+
+    cfg = load_config()
+    cfg["api_key"] = api_key
+    if base_url and base_url != "https://api.airweave.ai":
+        cfg["base_url"] = base_url
+    if collection:
+        cfg["collection"] = collection
+    save_config(cfg)
+
+    stderr.print("[green]Logged in.[/green] Config saved to ~/.airweave/config.json")
+
+
+@app.command()
+def status(
+    format: OutputFormat = typer.Option(
+        OutputFormat.text, "--format", "-f", help="Output format."
+    ),
+) -> None:
+    """Show current authentication state."""
+    env_key = os.environ.get("AIRWEAVE_API_KEY")
+    cfg = load_config()
+
+    info = {
+        "api_key_source": "env" if env_key else ("config" if cfg.get("api_key") else "none"),
+        "api_key_set": bool(env_key or cfg.get("api_key")),
+        "base_url": os.environ.get("AIRWEAVE_BASE_URL")
+        or cfg.get("base_url", "https://api.airweave.ai"),
+        "base_url_source": "env"
+        if os.environ.get("AIRWEAVE_BASE_URL")
+        else ("config" if cfg.get("base_url") else "default"),
+        "collection": os.environ.get("AIRWEAVE_COLLECTION") or cfg.get("collection"),
+        "collection_source": "env"
+        if os.environ.get("AIRWEAVE_COLLECTION")
+        else ("config" if cfg.get("collection") else "none"),
+    }
+
+    if format == OutputFormat.json:
+        typer.echo(json.dumps(info, indent=2))
+        return
+
+    stdout.print()
+    key_display = "****" if info["api_key_set"] else "[red]not set[/red]"
+    stdout.print(f"  API key:     {key_display}  [dim]({info['api_key_source']})[/dim]")
+    stdout.print(f"  Base URL:    {info['base_url']}  [dim]({info['base_url_source']})[/dim]")
+    coll_display = info["collection"] or "[red]not set[/red]"
+    coll_source = f"  [dim]({info['collection_source']})[/dim]" if info["collection"] else ""
+    stdout.print(f"  Collection:  {coll_display}{coll_source}")
+    stdout.print()
+
+
+@app.command()
+def logout() -> None:
+    """Clear saved credentials from ~/.airweave/config.json."""
+    clear_config()
+    stderr.print("Logged out. Config cleared.")

airweave_cli-0.1.0/airweave_cli/commands/collections.py

@@ -0,0 +1,116 @@
+from __future__ import annotations
+
+import json
+from enum import Enum
+from typing import Optional
+
+import typer
+from rich.console import Console
+from rich.table import Table
+
+from airweave_cli.config import get_client, serialize
+
+app = typer.Typer(name="collections", help="Manage collections.", no_args_is_help=True)
+
+stderr = Console(stderr=True)
+stdout = Console()
+
+
+class OutputFormat(str, Enum):
+    json = "json"
+    text = "text"
+
+
+@app.command("list")
+def list_collections(
+    format: OutputFormat = typer.Option(
+        OutputFormat.json, "--format", "-f", help="Output format."
+    ),
+) -> None:
+    """List all collections."""
+    client = get_client()
+    try:
+        collections = client.collections.list()
+    except Exception as exc:
+        stderr.print(f"[red]Error:[/red] {exc}")
+        raise typer.Exit(code=1)
+
+    if format == OutputFormat.json:
+        typer.echo(json.dumps(serialize(collections), indent=2, default=str))
+        return
+
+    if not collections:
+        stderr.print("[yellow]No collections found.[/yellow]")
+        return
+
+    table = Table(title="Collections")
+    table.add_column("Name", style="bold")
+    table.add_column("Readable ID")
+    table.add_column("Status")
+    table.add_column("Created")
+
+    for c in collections:
+        table.add_row(
+            c.name,
+            c.readable_id,
+            str(c.status) if c.status else "-",
+            c.created_at.strftime("%Y-%m-%d %H:%M") if c.created_at else "-",
+        )
+
+    stdout.print(table)
+
+
+@app.command()
+def create(
+    name: str = typer.Option(..., "--name", "-n", help="Collection name."),
+    readable_id: Optional[str] = typer.Option(
+        None, "--readable-id", "-r", help="Custom readable ID (auto-generated if omitted)."
+    ),
+    format: OutputFormat = typer.Option(
+        OutputFormat.json, "--format", "-f", help="Output format."
+    ),
+) -> None:
+    """Create a new collection."""
+    client = get_client()
+    kwargs = {"name": name}
+    if readable_id:
+        kwargs["readable_id"] = readable_id
+
+    try:
+        collection = client.collections.create(**kwargs)
+    except Exception as exc:
+        stderr.print(f"[red]Error:[/red] {exc}")
+        raise typer.Exit(code=1)
+
+    if format == OutputFormat.json:
+        typer.echo(json.dumps(serialize(collection), indent=2, default=str))
+        return
+
+    stdout.print(f"[green]Created:[/green] {collection.name} ({collection.readable_id})")
+
+
+@app.command()
+def get(
+    readable_id: str = typer.Argument(..., help="Collection readable_id."),
+    format: OutputFormat = typer.Option(
+        OutputFormat.json, "--format", "-f", help="Output format."
+    ),
+) -> None:
+    """Get details of a collection."""
+    client = get_client()
+    try:
+        collection = client.collections.get(readable_id)
+    except Exception as exc:
+        stderr.print(f"[red]Error:[/red] {exc}")
+        raise typer.Exit(code=1)
+
+    if format == OutputFormat.json:
+        typer.echo(json.dumps(serialize(collection), indent=2, default=str))
+        return
+
+    stdout.print(f"[bold]{collection.name}[/bold]  ({collection.readable_id})")
+    stdout.print(f"  ID:        {collection.id}")
+    stdout.print(f"  Status:    {collection.status}")
+    stdout.print(f"  Vectors:   {collection.vector_size}d ({collection.embedding_model_name})")
+    stdout.print(f"  Created:   {collection.created_at}")
+    stdout.print(f"  Modified:  {collection.modified_at}")

airweave_cli-0.1.0/airweave_cli/commands/search.py

@@ -0,0 +1,87 @@
+from __future__ import annotations
+
+import json
+from enum import Enum
+from typing import Optional
+
+import typer
+from rich.console import Console
+from rich.markdown import Markdown
+from rich.panel import Panel
+from rich.text import Text
+
+from airweave_cli.config import get_client, resolve_collection, serialize
+
+stderr = Console(stderr=True)
+stdout = Console()
+
+
+class OutputFormat(str, Enum):
+    json = "json"
+    text = "text"
+
+
+def search(
+    query: str = typer.Argument(..., help="Search query."),
+    collection: Optional[str] = typer.Option(
+        None, "--collection", "-c", help="Collection readable_id."
+    ),
+    top_k: int = typer.Option(10, "--top-k", "-k", help="Number of results to return."),
+    format: OutputFormat = typer.Option(
+        OutputFormat.json, "--format", "-f", help="Output format."
+    ),
+) -> None:
+    """Search a collection. This is the core command.
+
+    Pipe JSON output directly:  airweave search "query" | jq '.results[0]'
+    """
+    coll = resolve_collection(collection)
+    client = get_client()
+
+    try:
+        from airweave import SearchRequest
+
+        response = client.collections.search(
+            coll,
+            request=SearchRequest(query=query, limit=top_k),
+        )
+    except Exception as exc:
+        stderr.print(f"[red]Error:[/red] {exc}")
+        raise typer.Exit(code=1)
+
+    if format == OutputFormat.json:
+        typer.echo(json.dumps(serialize(response), indent=2, default=str))
+        return
+
+    # Rich text output
+    results = response.results
+    if not results:
+        stderr.print("[yellow]No results found.[/yellow]")
+        raise typer.Exit(code=0)
+
+    if response.completion:
+        stdout.print(Panel(Markdown(response.completion), title="Answer", border_style="green"))
+        stdout.print()
+
+    for i, result in enumerate(results, 1):
+        meta = result.get("system_metadata") or {}
+        source = meta.get("source_name") or result.get("source_name", "unknown")
+        score = result.get("score")
+        name = result.get("name", "")
+        content = result.get("textual_representation") or result.get("md_content", "")
+        source_fields = result.get("source_fields") or {}
+        url = source_fields.get("web_url") or source_fields.get("url") or result.get("url")
+
+        title_parts = [f"[bold]#{i}[/bold]  {source}"]
+        if score is not None:
+            title_parts.append(f"[dim]score={score:.3f}[/dim]")
+        title = Text.from_markup("  ".join(title_parts))
+
+        snippet = content[:500] + ("..." if len(content) > 500 else "") if content else ""
+        body = Markdown(snippet) if snippet else Text.from_markup("[dim]no content[/dim]")
+
+        subtitle = url or None
+        if name and not snippet:
+            body = Text(name)
+
+        stdout.print(Panel(body, title=title, subtitle=subtitle, border_style="blue"))

airweave_cli-0.1.0/airweave_cli/commands/sources.py

@@ -0,0 +1,96 @@
+from __future__ import annotations
+
+import json
+from enum import Enum
+from typing import Optional
+
+import typer
+from rich.console import Console
+from rich.table import Table
+
+from airweave_cli.config import get_client, resolve_collection, serialize
+
+app = typer.Typer(name="sources", help="Manage source connections.", no_args_is_help=True)
+
+stderr = Console(stderr=True)
+stdout = Console()
+
+
+class OutputFormat(str, Enum):
+    json = "json"
+    text = "text"
+
+
+@app.command("list")
+def list_sources(
+    collection: Optional[str] = typer.Option(
+        None, "--collection", "-c", help="Collection readable_id."
+    ),
+    format: OutputFormat = typer.Option(
+        OutputFormat.json, "--format", "-f", help="Output format."
+    ),
+) -> None:
+    """List source connections for a collection."""
+    coll = resolve_collection(collection)
+    client = get_client()
+
+    try:
+        sources = client.source_connections.list(collection=coll)
+    except Exception as exc:
+        stderr.print(f"[red]Error:[/red] {exc}")
+        raise typer.Exit(code=1)
+
+    if format == OutputFormat.json:
+        typer.echo(json.dumps(serialize(sources), indent=2, default=str))
+        return
+
+    if not sources:
+        stderr.print("[yellow]No source connections found.[/yellow]")
+        return
+
+    table = Table(title="Source Connections")
+    table.add_column("Name", style="bold")
+    table.add_column("Source")
+    table.add_column("ID")
+    table.add_column("Status")
+    table.add_column("Entities")
+
+    for s in sources:
+        table.add_row(
+            s.name,
+            s.short_name,
+            s.id,
+            str(s.status) if s.status else "-",
+            str(s.entity_count) if s.entity_count is not None else "-",
+        )
+
+    stdout.print(table)
+
+
+@app.command()
+def sync(
+    source_connection_id: str = typer.Argument(..., help="Source connection ID (UUID)."),
+    force: bool = typer.Option(
+        False, "--force", help="Force a full re-sync instead of incremental."
+    ),
+    format: OutputFormat = typer.Option(
+        OutputFormat.json, "--format", "-f", help="Output format."
+    ),
+) -> None:
+    """Trigger a sync for a source connection."""
+    client = get_client()
+
+    try:
+        job = client.source_connections.run(
+            source_connection_id,
+            force_full_sync=force if force else None,
+        )
+    except Exception as exc:
+        stderr.print(f"[red]Error:[/red] {exc}")
+        raise typer.Exit(code=1)
+
+    if format == OutputFormat.json:
+        typer.echo(json.dumps(serialize(job), indent=2, default=str))
+        return
+
+    stdout.print(f"[green]Sync started.[/green]  Job ID: {job.id}  Status: {job.status}")

airweave_cli-0.1.0/airweave_cli/config.py

@@ -0,0 +1,105 @@
+from __future__ import annotations
+
+import json
+import os
+from pathlib import Path
+from typing import Any, Dict, Optional
+
+import typer
+
+CONFIG_DIR = Path.home() / ".airweave"
+CONFIG_PATH = CONFIG_DIR / "config.json"
+
+DEFAULT_BASE_URL = "https://api.airweave.ai"
+
+
+def load_config() -> Dict[str, Any]:
+    if not CONFIG_PATH.exists():
+        return {}
+    try:
+        return json.loads(CONFIG_PATH.read_text())
+    except (json.JSONDecodeError, OSError):
+        return {}
+
+
+def save_config(data: Dict[str, Any]) -> None:
+    CONFIG_DIR.mkdir(parents=True, exist_ok=True)
+    CONFIG_DIR.chmod(0o700)
+    CONFIG_PATH.write_text(json.dumps(data, indent=2) + "\n")
+
+
+def clear_config() -> None:
+    if CONFIG_PATH.exists():
+        CONFIG_PATH.unlink()
+
+
+def _fail(message: str) -> None:
+    typer.echo(message, err=True)
+    raise typer.Exit(code=1)
+
+
+def resolve_api_key() -> str:
+    key = os.environ.get("AIRWEAVE_API_KEY")
+    if key:
+        return key
+
+    cfg = load_config()
+    key = cfg.get("api_key")
+    if key:
+        return key
+
+    _fail(
+        "No API key found. Set AIRWEAVE_API_KEY or run: airweave auth login"
+    )
+    return ""  # unreachable, keeps type checkers happy
+
+
+def resolve_base_url() -> str:
+    url = os.environ.get("AIRWEAVE_BASE_URL")
+    if url:
+        return url
+
+    cfg = load_config()
+    url = cfg.get("base_url")
+    if url:
+        return url
+
+    return DEFAULT_BASE_URL
+
+
+def resolve_collection(flag: Optional[str] = None) -> str:
+    if flag:
+        return flag
+
+    coll = os.environ.get("AIRWEAVE_COLLECTION")
+    if coll:
+        return coll
+
+    cfg = load_config()
+    coll = cfg.get("collection")
+    if coll:
+        return coll
+
+    _fail(
+        "No collection specified. Use --collection, set AIRWEAVE_COLLECTION, "
+        "or run: airweave auth login"
+    )
+    return ""
+
+
+def get_client():
+    from airweave import AirweaveSDK
+
+    return AirweaveSDK(
+        api_key=resolve_api_key(),
+        base_url=resolve_base_url(),
+    )
+
+
+def serialize(obj: Any) -> Any:
+    """Convert an SDK model (or list of models) to a JSON-serializable dict."""
+    if isinstance(obj, list):
+        return [serialize(item) for item in obj]
+    if hasattr(obj, "dict"):
+        return obj.dict()
+    return obj

airweave_cli-0.1.0/airweave_cli/main.py

@@ -0,0 +1,45 @@
+from __future__ import annotations
+
+import typer
+
+from airweave_cli import __version__
+from airweave_cli.commands.auth import app as auth_app
+from airweave_cli.commands.collections import app as collections_app
+from airweave_cli.commands.search import search
+from airweave_cli.commands.sources import app as sources_app
+
+app = typer.Typer(
+    name="airweave",
+    help="Airweave CLI — search across your connected sources.",
+    no_args_is_help=True,
+)
+
+
+def _version_callback(value: bool) -> None:
+    if value:
+        typer.echo(f"airweave-cli {__version__}")
+        raise typer.Exit()
+
+
+@app.callback()
+def main(
+    version: bool = typer.Option(
+        False,
+        "--version",
+        "-v",
+        help="Show version and exit.",
+        callback=_version_callback,
+        is_eager=True,
+    ),
+) -> None:
+    pass
+
+
+app.add_typer(auth_app, name="auth")
+app.add_typer(collections_app, name="collections")
+app.add_typer(sources_app, name="sources")
+app.command()(search)
+
+
+if __name__ == "__main__":
+    app()

airweave_cli-0.1.0/pyproject.toml

@@ -0,0 +1,57 @@
+[project]
+name = "airweave-cli"
+
+[tool.poetry]
+name = "airweave-cli"
+version = "0.1.0"
+description = "The Airweave CLI for developers and AI agents."
+readme = "README.md"
+license = "MIT"
+authors = ["Airweave <hello@airweave.ai>"]
+keywords = ["airweave", "cli", "search", "ai", "agents"]
+
+classifiers = [
+    "Intended Audience :: Developers",
+    "Programming Language :: Python",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.9",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+    "Programming Language :: Python :: 3.13",
+    "Operating System :: OS Independent",
+    "Topic :: Software Development :: Libraries :: Python Modules",
+    "Typing :: Typed",
+]
+packages = [
+    { include = "airweave_cli" }
+]
+
+[project.urls]
+Repository = "https://github.com/airweave-ai/cli"
+
+[tool.poetry.dependencies]
+python = "^3.9"
+typer = {version = "^0.12", extras = ["all"]}
+rich = "^13"
+airweave-sdk = ">=0.9"
+
+[tool.poetry.group.dev.dependencies]
+pytest = "^8.0"
+ruff = ">=0.11"
+
+[tool.poetry.scripts]
+airweave = "airweave_cli.main:app"
+
+[tool.ruff]
+line-length = 100
+
+[tool.ruff.lint]
+select = ["E", "F", "I"]
+
+[tool.pytest.ini_options]
+testpaths = ["tests"]
+
+[build-system]
+requires = ["poetry-core"]
+build-backend = "poetry.core.masonry.api"