nao-core 0.0.9__tar.gz → 0.0.11__tar.gz

nao_core-0.0.11/PKG-INFO

@@ -0,0 +1,130 @@
+Metadata-Version: 2.4
+Name: nao-core
+Version: 0.0.11
+Summary: nao Core is your analytics context builder with the best chat interface.
+Project-URL: Homepage, https://getnao.io
+Project-URL: Repository, https://github.com/naolabs/chat
+Author: nao Labs
+License-Expression: MIT
+License-File: LICENSE
+Keywords: ai,analytics,chat
+Classifier: Development Status :: 3 - Alpha
+Classifier: Environment :: Console
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: MacOS
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Requires-Python: >=3.10
+Requires-Dist: cyclopts>=4.4.4
+Requires-Dist: ibis-framework[bigquery]>=9.0.0
+Requires-Dist: openai>=1.0.0
+Requires-Dist: pydantic>=2.10.0
+Requires-Dist: pyyaml>=6.0.0
+Requires-Dist: rich>=14.0.0
+Description-Content-Type: text/markdown
+
+# nao CLI
+
+Command-line interface for nao chat.
+
+## Installation
+
+```bash
+pip install nao-core
+```
+
+## Usage
+
+```bash
+nao --help
+Usage: nao COMMAND
+
+╭─ Commands ────────────────────────────────────────────────────────────────╮
+│ chat         Start the nao chat UI.                                       │
+│ init         Initialize a new nao project.                                │
+│ --help (-h)  Display this message and exit.                               │
+│ --version    Display application version.                                 │
+╰───────────────────────────────────────────────────────────────────────────╯
+```
+
+### Initialize a new nao project
+
+```bash
+nao init
+```
+
+This will create a new nao project in the current directory. It will prompt you for a project name and ask you if you want to set up an LLM configuration.
+
+### Start the nao chat UI
+
+```bash
+nao chat
+```
+
+This will start the nao chat UI. It will open the chat interface in your browser at `http://localhost:5005`.
+
+## Development
+
+### Building the package
+
+```bash
+cd cli
+python build.py --help
+Usage: build.py [OPTIONS]
+
+Build and package nao-core CLI.
+
+╭─ Commands ────────────────────────────────────────────────────────────────────────────────────────────────╮
+│ --help (-h)  Display this message and exit.                                                               │
+│ --version    Display application version.                                                                 │
+╰───────────────────────────────────────────────────────────────────────────────────────────────────────────╯
+╭─ Parameters ──────────────────────────────────────────────────────────────────────────────────────────────╮
+│ --force -f --no-force              Force rebuild the server binary [default: False]                       │
+│ --skip-server -s --no-skip-server  Skip server build, only build Python package [default: False]          │
+│ --bump                             Bump version before building (patch, minor, major) [choices: patch,    │
+│                                    minor, major]                                                          │
+╰───────────────────────────────────────────────────────────────────────────────────────────────────────────╯
+```
+
+This will:
+1. Build the frontend with Vite
+2. Compile the backend with Bun into a standalone binary
+3. Bundle everything into a Python wheel in `dist/`
+
+Options:
+- `--force` / `-f`: Force rebuild the server binary
+- `--skip-server`: Skip server build, only build Python package
+- `--bump`: Bump version before building (patch, minor, major)
+
+### Installing for development
+
+```bash
+cd cli
+pip install -e .
+```
+
+### Publishing to PyPI
+
+```bash
+# Build first
+python build.py
+
+# Publish
+uv publish dist/*
+```
+
+## Architecture
+
+```
+nao chat (CLI command)
+    ↓ spawns
+nao-chat-server (Bun-compiled binary)
+    ↓ serves
+Backend API + Frontend Static Files
+    ↓
+Browser at http://localhost:5005
+```

nao_core-0.0.11/README.md

@@ -0,0 +1,101 @@
+# nao CLI
+
+Command-line interface for nao chat.
+
+## Installation
+
+```bash
+pip install nao-core
+```
+
+## Usage
+
+```bash
+nao --help
+Usage: nao COMMAND
+
+╭─ Commands ────────────────────────────────────────────────────────────────╮
+│ chat         Start the nao chat UI.                                       │
+│ init         Initialize a new nao project.                                │
+│ --help (-h)  Display this message and exit.                               │
+│ --version    Display application version.                                 │
+╰───────────────────────────────────────────────────────────────────────────╯
+```
+
+### Initialize a new nao project
+
+```bash
+nao init
+```
+
+This will create a new nao project in the current directory. It will prompt you for a project name and ask you if you want to set up an LLM configuration.
+
+### Start the nao chat UI
+
+```bash
+nao chat
+```
+
+This will start the nao chat UI. It will open the chat interface in your browser at `http://localhost:5005`.
+
+## Development
+
+### Building the package
+
+```bash
+cd cli
+python build.py --help
+Usage: build.py [OPTIONS]
+
+Build and package nao-core CLI.
+
+╭─ Commands ────────────────────────────────────────────────────────────────────────────────────────────────╮
+│ --help (-h)  Display this message and exit.                                                               │
+│ --version    Display application version.                                                                 │
+╰───────────────────────────────────────────────────────────────────────────────────────────────────────────╯
+╭─ Parameters ──────────────────────────────────────────────────────────────────────────────────────────────╮
+│ --force -f --no-force              Force rebuild the server binary [default: False]                       │
+│ --skip-server -s --no-skip-server  Skip server build, only build Python package [default: False]          │
+│ --bump                             Bump version before building (patch, minor, major) [choices: patch,    │
+│                                    minor, major]                                                          │
+╰───────────────────────────────────────────────────────────────────────────────────────────────────────────╯
+```
+
+This will:
+1. Build the frontend with Vite
+2. Compile the backend with Bun into a standalone binary
+3. Bundle everything into a Python wheel in `dist/`
+
+Options:
+- `--force` / `-f`: Force rebuild the server binary
+- `--skip-server`: Skip server build, only build Python package
+- `--bump`: Bump version before building (patch, minor, major)
+
+### Installing for development
+
+```bash
+cd cli
+pip install -e .
+```
+
+### Publishing to PyPI
+
+```bash
+# Build first
+python build.py
+
+# Publish
+uv publish dist/*
+```
+
+## Architecture
+
+```
+nao chat (CLI command)
+    ↓ spawns
+nao-chat-server (Bun-compiled binary)
+    ↓ serves
+Backend API + Frontend Static Files
+    ↓
+Browser at http://localhost:5005
+```

{nao_core-0.0.9 → nao_core-0.0.11}/nao_core/__init__.py

@@ -1,3 +1,2 @@
 # nao Core CLI package
-__version__ = "0.0.9"
-
+__version__ = "0.0.11"

nao_core-0.0.11/nao_core/commands/__init__.py

@@ -0,0 +1,6 @@
+from nao_core.commands.chat import chat
+from nao_core.commands.debug import debug
+from nao_core.commands.init import init
+from nao_core.commands.sync import sync
+
+__all__ = ["chat", "debug", "init", "sync"]

nao_core-0.0.11/nao_core/commands/chat.py

@@ -0,0 +1,132 @@
+import os
+import subprocess
+import sys
+import webbrowser
+from pathlib import Path
+from time import sleep
+
+from rich.console import Console
+
+from nao_core.config import NaoConfig
+
+console = Console()
+
+# Default port for the nao chat server
+SERVER_PORT = 5005
+
+
+def get_server_binary_path() -> Path:
+    """Get the path to the bundled nao-chat-server binary."""
+    # The binary is in the bin folder relative to this file
+    cli_dir = Path(__file__).parent.parent
+    bin_dir = cli_dir / "bin"
+    binary_path = bin_dir / "nao-chat-server"
+
+    if not binary_path.exists():
+        console.print(f"[bold red]✗[/bold red] Server binary not found at {binary_path}")
+        console.print("[dim]Make sure you've built the server with ./scripts/build-server.sh[/dim]")
+        sys.exit(1)
+
+    return binary_path
+
+
+def wait_for_server(port: int, timeout: int = 30) -> bool:
+    """Wait for the server to be ready."""
+    import socket
+
+    for _ in range(timeout * 10):  # Check every 100ms
+        try:
+            with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as sock:
+                sock.settimeout(0.1)
+                result = sock.connect_ex(("localhost", port))
+                if result == 0:
+                    return True
+        except OSError:
+            pass
+        sleep(0.1)
+    return False
+
+
+def chat():
+    """Start the nao chat UI.
+
+    Launches the nao chat server and opens the web interface in your browser.
+    """
+    console.print("\n[bold cyan]💬 Starting nao chat...[/bold cyan]\n")
+
+    binary_path = get_server_binary_path()
+    bin_dir = binary_path.parent
+
+    console.print(f"[dim]Server binary: {binary_path}[/dim]")
+    console.print(f"[dim]Working directory: {bin_dir}[/dim]")
+
+    # Try to load nao config from current directory
+    config = NaoConfig.try_load()
+    if config:
+        console.print(f"[bold green]✓[/bold green] Loaded config from {Path.cwd() / 'nao_config.yaml'}")
+    else:
+        console.print("[dim]No nao_config.yaml found in current directory[/dim]")
+
+    # Start the server process
+    process = None
+    try:
+        # Set up environment - inherit from parent but ensure we're in the bin dir
+        # so the server can find the public folder
+        env = os.environ.copy()
+
+        # Set LLM API key from config if available
+        if config and config.llm:
+            env_var_name = f"{config.llm.provider.upper()}_API_KEY"
+            env[env_var_name] = config.llm.api_key
+            console.print(f"[bold green]✓[/bold green] Set {env_var_name} from config")
+
+        process = subprocess.Popen(
+            [str(binary_path)],
+            cwd=str(bin_dir),
+            env=env,
+            stdout=subprocess.PIPE,
+            stderr=subprocess.STDOUT,
+            text=True,
+        )
+
+        console.print("[bold green]✓[/bold green] Server starting...")
+
+        # Wait for the server to be ready
+        if wait_for_server(SERVER_PORT):
+            url = f"http://localhost:{SERVER_PORT}"
+            console.print(f"[bold green]✓[/bold green] Server ready at {url}")
+            console.print("\n[bold]Opening browser...[/bold]")
+            webbrowser.open(url)
+            console.print("\n[dim]Press Ctrl+C to stop the server[/dim]\n")
+        else:
+            console.print("[bold yellow]⚠[/bold yellow] Server is taking longer than expected to start...")
+            console.print(f"[dim]Check http://localhost:{SERVER_PORT} manually[/dim]")
+
+        # Stream server output to console
+        if process.stdout:
+            for line in process.stdout:
+                # Filter out some of the verbose logging if needed
+                console.print(f"[dim]{line.rstrip()}[/dim]")
+
+        # Wait for process to complete
+        process.wait()
+
+    except KeyboardInterrupt:
+        console.print("\n[bold yellow]Shutting down...[/bold yellow]")
+        if process:
+            # Send SIGTERM for graceful shutdown
+            process.terminate()
+            try:
+                process.wait(timeout=5)
+            except subprocess.TimeoutExpired:
+                # Force kill if it doesn't respond
+                process.kill()
+                process.wait()
+        console.print("[bold green]✓[/bold green] Server stopped")
+        sys.exit(0)
+
+    except Exception as e:
+        console.print(f"[bold red]✗[/bold red] Failed to start server: {e}")
+        if process:
+            process.kill()
+        sys.exit(1)

nao_core-0.0.11/nao_core/commands/debug.py

@@ -0,0 +1,144 @@
+import sys
+
+from rich.console import Console
+from rich.table import Table
+
+from nao_core.config import NaoConfig
+
+console = Console()
+
+
+def test_database_connection(db_config) -> tuple[bool, str]:
+    """Test connectivity to a database.
+
+    Returns:
+            Tuple of (success, message)
+    """
+    try:
+        conn = db_config.connect()
+        # Run a simple query to verify the connection works
+        if db_config.dataset_id:
+            # If dataset is specified, list tables in that dataset
+            tables = conn.list_tables()
+            table_count = len(tables)
+            return True, f"Connected successfully ({table_count} tables found)"
+        else:
+            # If no dataset, list datasets in the project instead
+            datasets = conn.list_databases()
+            dataset_count = len(datasets)
+            return True, f"Connected successfully ({dataset_count} datasets found)"
+    except Exception as e:
+        return False, str(e)
+
+
+def test_llm_connection(llm_config) -> tuple[bool, str]:
+    """Test connectivity to an LLM provider.
+
+    Returns:
+            Tuple of (success, message)
+    """
+    try:
+        if llm_config.provider.value == "openai":
+            import openai
+
+            client = openai.OpenAI(api_key=llm_config.api_key)
+            # Make a minimal API call to verify the key works
+            models = client.models.list()
+            # Just check we can iterate (don't need to consume all)
+            model_count = sum(1 for _ in models)
+            return True, f"Connected successfully ({model_count} models available)"
+        else:
+            return False, f"Unknown provider: {llm_config.provider}"
+    except Exception as e:
+        return False, str(e)
+
+
+def debug():
+    """Test connectivity to configured databases and LLMs.
+
+    Loads the nao configuration from the current directory and tests
+    connections to all configured databases and LLM providers.
+    """
+    console.print("\n[bold cyan]🔍 nao debug - Testing connections...[/bold cyan]\n")
+
+    # Load config
+    config = NaoConfig.try_load()
+    if not config:
+        console.print("[bold red]✗[/bold red] No nao_config.yaml found in current directory")
+        console.print("[dim]Run 'nao init' to create a configuration file[/dim]")
+        sys.exit(1)
+
+    console.print(f"[bold green]✓[/bold green] Loaded config: [cyan]{config.project_name}[/cyan]\n")
+
+    # Test databases
+    if config.databases:
+        console.print("[bold]Databases:[/bold]")
+        db_table = Table(show_header=True, header_style="bold")
+        db_table.add_column("Name")
+        db_table.add_column("Type")
+        db_table.add_column("Status")
+        db_table.add_column("Details")
+
+        for db in config.databases:
+            console.print(f"  Testing [cyan]{db.name}[/cyan]...", end=" ")
+            success, message = test_database_connection(db)
+
+            if success:
+                console.print("[bold green]✓[/bold green]")
+                db_table.add_row(
+                    db.name,
+                    db.type,
+                    "[green]Connected[/green]",
+                    message,
+                )
+            else:
+                console.print("[bold red]✗[/bold red]")
+                # Truncate long error messages
+                short_msg = message[:80] + "..." if len(message) > 80 else message
+                db_table.add_row(
+                    db.name,
+                    db.type,
+                    "[red]Failed[/red]",
+                    short_msg,
+                )
+
+        console.print()
+        console.print(db_table)
+    else:
+        console.print("[dim]No databases configured[/dim]")
+
+    console.print()
+
+    # Test LLM
+    if config.llm:
+        console.print("[bold]LLM Provider:[/bold]")
+        llm_table = Table(show_header=True, header_style="bold")
+        llm_table.add_column("Provider")
+        llm_table.add_column("Status")
+        llm_table.add_column("Details")
+
+        console.print(f"  Testing [cyan]{config.llm.provider.value}[/cyan]...", end=" ")
+        success, message = test_llm_connection(config.llm)
+
+        if success:
+            console.print("[bold green]✓[/bold green]")
+            llm_table.add_row(
+                config.llm.provider.value,
+                "[green]Connected[/green]",
+                message,
+            )
+        else:
+            console.print("[bold red]✗[/bold red]")
+            short_msg = message[:80] + "..." if len(message) > 80 else message
+            llm_table.add_row(
+                config.llm.provider.value,
+                "[red]Failed[/red]",
+                short_msg,
+            )
+
+        console.print()
+        console.print(llm_table)
+    else:
+        console.print("[dim]No LLM configured[/dim]")
+
+    console.print()

nao_core-0.0.11/nao_core/commands/init.py

@@ -0,0 +1,175 @@
+from pathlib import Path
+from typing import Annotated
+
+from cyclopts import Parameter
+from rich.console import Console
+from rich.prompt import Confirm, Prompt
+
+from nao_core.config import BigQueryConfig, DatabaseConfig, DatabaseType, LLMConfig, LLMProvider, NaoConfig
+
+console = Console()
+
+
+class InitError(Exception):
+    """Base exception for init command errors."""
+
+    pass
+
+
+class EmptyProjectNameError(InitError):
+    """Raised when project name is empty."""
+
+    def __init__(self):
+        super().__init__("Project name cannot be empty.")
+
+
+class ProjectExistsError(InitError):
+    """Raised when project folder already exists."""
+
+    def __init__(self, project_name: str):
+        self.project_name = project_name
+        super().__init__(f"Folder '{project_name}' already exists.")
+
+
+class EmptyApiKeyError(InitError):
+    """Raised when API key is empty."""
+
+    def __init__(self):
+        super().__init__("API key cannot be empty.")
+
+
+def setup_project_name(force: bool = False) -> tuple[str, Path]:
+    """Setup the project name."""
+    project_name = Prompt.ask("[bold]Enter your project name[/bold]")
+
+    if not project_name:
+        raise EmptyProjectNameError()
+
+    project_path = Path(project_name)
+
+    if project_path.exists() and not force:
+        raise ProjectExistsError(project_name)
+
+    project_path.mkdir(parents=True, exist_ok=True)
+
+    return project_name, project_path
+
+
+def setup_bigquery() -> BigQueryConfig:
+    """Setup a BigQuery database configuration."""
+    console.print("\n[bold cyan]BigQuery Configuration[/bold cyan]\n")
+
+    name = Prompt.ask("[bold]Connection name[/bold]", default="bigquery-prod")
+
+    project_id = Prompt.ask("[bold]GCP Project ID[/bold]")
+    if not project_id:
+        raise InitError("GCP Project ID cannot be empty.")
+
+    dataset_id = Prompt.ask("[bold]Default dataset[/bold] [dim](optional, press Enter to skip)[/dim]", default="")
+
+    credentials_path = Prompt.ask(
+        "[bold]Service account JSON path[/bold] [dim](optional, uses ADC if empty)[/dim]",
+        default="",
+    )
+
+    return BigQueryConfig(
+        name=name,
+        project_id=project_id,
+        dataset_id=dataset_id or None,
+        credentials_path=credentials_path or None,
+    )
+
+
+def setup_databases() -> list[DatabaseConfig]:
+    """Setup database configurations."""
+    databases: list[DatabaseConfig] = []
+
+    should_setup = Confirm.ask("\n[bold]Set up database connections?[/bold]", default=True)
+
+    if not should_setup:
+        return databases
+
+    while True:
+        console.print("\n[bold cyan]Database Configuration[/bold cyan]\n")
+
+        db_type_choices = [t.value for t in DatabaseType]
+        db_type = Prompt.ask(
+            "[bold]Select database type[/bold]",
+            choices=db_type_choices,
+            default=db_type_choices[0],
+        )
+
+        if db_type == DatabaseType.BIGQUERY.value:
+            db_config = setup_bigquery()
+            databases.append(db_config)
+            console.print(f"\n[bold green]✓[/bold green] Added database [cyan]{db_config.name}[/cyan]")
+
+        add_another = Confirm.ask("\n[bold]Add another database?[/bold]", default=False)
+        if not add_another:
+            break
+
+    return databases
+
+
+def setup_llm() -> LLMConfig | None:
+    """Setup the LLM configuration."""
+    llm_config = None
+    should_setup = Confirm.ask("\n[bold]Set up LLM configuration?[/bold]", default=True)
+
+    if should_setup:
+        console.print("\n[bold cyan]LLM Configuration[/bold cyan]\n")
+
+        provider_choices = [p.value for p in LLMProvider]
+        llm_provider = Prompt.ask(
+            "[bold]Select LLM provider[/bold]",
+            choices=provider_choices,
+            default=provider_choices[0],
+        )
+
+        api_key = Prompt.ask(
+            f"[bold]Enter your {llm_provider.upper()} API key[/bold]",
+            password=True,
+        )
+
+        if not api_key:
+            raise EmptyApiKeyError()
+
+        llm_config = LLMConfig(
+            provider=LLMProvider(llm_provider),
+            api_key=api_key,
+        )
+
+    return llm_config
+
+
+def init(
+    *,
+    force: Annotated[bool, Parameter(name=["-f", "--force"])] = False,
+):
+    """Initialize a new nao project.
+
+    Creates a project folder with a nao_config.yaml configuration file.
+
+    Parameters
+    ----------
+    force : bool
+        Force re-initialization even if the folder already exists.
+    """
+    console.print("\n[bold cyan]🚀 nao project initialization[/bold cyan]\n")
+
+    try:
+        project_name, project_path = setup_project_name(force=force)
+        config = NaoConfig(
+            project_name=project_name,
+            databases=setup_databases(),
+            llm=setup_llm(),
+        )
+        config.save(project_path)
+
+        console.print()
+        console.print(f"[bold green]✓[/bold green] Created project [cyan]{project_name}[/cyan]")
+        console.print(f"[bold green]✓[/bold green] Created [dim]{project_path / 'nao_config.yaml'}[/dim]")
+        console.print()
+        console.print("[bold green]Done![/bold green] Your nao project is ready. 🎉")
+    except InitError as e:
+        console.print(f"[bold red]✗[/bold red] {e}")