PyPI - nao-core - Versions diffs - 0.0.28__py3-none-any.whl → 0.0.30__py3-none-any.whl - Mend

nao-core 0.0.28py3-none-any.whl → 0.0.30py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (46) hide show

nao_core/__init__.py +1 -1
nao_core/bin/fastapi/main.py +15 -2
nao_core/bin/fastapi/test_main.py +156 -0
nao_core/bin/migrations-postgres/0004_input_and_output_tokens.sql +8 -0
nao_core/bin/migrations-postgres/meta/0004_snapshot.json +847 -0
nao_core/bin/migrations-postgres/meta/_journal.json +7 -0
nao_core/bin/migrations-sqlite/0004_input_and_output_tokens.sql +8 -0
nao_core/bin/migrations-sqlite/meta/0004_snapshot.json +819 -0
nao_core/bin/migrations-sqlite/meta/_journal.json +7 -0
nao_core/bin/nao-chat-server +0 -0
nao_core/bin/public/assets/{code-block-F6WJLWQG-EQr6mTlQ.js → code-block-F6WJLWQG-z4zcca7w.js} +3 -3
nao_core/bin/public/assets/index-ClduEZSo.css +1 -0
nao_core/bin/public/assets/index-DhhS7iVA.js +473 -0
nao_core/bin/public/index.html +2 -2
nao_core/commands/debug.py +10 -6
nao_core/commands/init.py +39 -23
nao_core/commands/sync/accessors.py +2 -3
nao_core/commands/sync/databases.py +243 -1
nao_core/config/__init__.py +14 -1
nao_core/config/databases/__init__.py +32 -11
nao_core/config/databases/base.py +6 -0
nao_core/config/databases/bigquery.py +29 -1
nao_core/config/databases/databricks.py +69 -0
nao_core/config/databases/duckdb.py +33 -0
nao_core/config/databases/snowflake.py +115 -0
nao_core/config/exceptions.py +7 -0
{nao_core-0.0.28.dist-info → nao_core-0.0.30.dist-info}/METADATA +5 -4
{nao_core-0.0.28.dist-info → nao_core-0.0.30.dist-info}/RECORD +31 -35
nao_core/bin/public/assets/_chat-layout-BTlqRUE5.js +0 -1
nao_core/bin/public/assets/_chat-layout.index-DOARokp1.js +0 -1
nao_core/bin/public/assets/agentProvider-C6dGIy-H.js +0 -1
nao_core/bin/public/assets/button-By_1dzVx.js +0 -1
nao_core/bin/public/assets/folder-DnRS5rg3.js +0 -1
nao_core/bin/public/assets/index-CElAN2SH.css +0 -1
nao_core/bin/public/assets/index-ZTHASguQ.js +0 -59
nao_core/bin/public/assets/input-CUQA5tsi.js +0 -1
nao_core/bin/public/assets/login-BUQDum3t.js +0 -1
nao_core/bin/public/assets/mermaid-FSSLJTFX-Dc6ZvCPw.js +0 -427
nao_core/bin/public/assets/sidebar-bgEk7Xg8.js +0 -1
nao_core/bin/public/assets/signinForm-CGAhnAkv.js +0 -1
nao_core/bin/public/assets/signup-D2n11La3.js +0 -1
nao_core/bin/public/assets/user-CYl8Tly2.js +0 -1
nao_core/bin/public/assets/utils-DzJYey0s.js +0 -1
{nao_core-0.0.28.dist-info → nao_core-0.0.30.dist-info}/WHEEL +0 -0
{nao_core-0.0.28.dist-info → nao_core-0.0.30.dist-info}/entry_points.txt +0 -0
{nao_core-0.0.28.dist-info → nao_core-0.0.30.dist-info}/licenses/LICENSE +0 -0

nao_core/bin/public/index.html CHANGED Viewed

@@ -9,8 +9,8 @@
 		<link rel="apple-touch-icon" href="/logo192.png" />
 		<link rel="manifest" href="/manifest.json" />
 		<title>nao — Chat with your data</title>
-		<script type="module" crossorigin src="/assets/index-ZTHASguQ.js"></script>
-		<link rel="stylesheet" crossorigin href="/assets/index-CElAN2SH.css">
+		<script type="module" crossorigin src="/assets/index-DhhS7iVA.js"></script>
+		<link rel="stylesheet" crossorigin href="/assets/index-ClduEZSo.css">
 	</head>
 	<body>
 		<div id="app"></div>

nao_core/commands/debug.py CHANGED Viewed

@@ -4,11 +4,12 @@ from rich.console import Console
 from rich.table import Table
 from nao_core.config import NaoConfig
+from nao_core.config.databases import AnyDatabaseConfig
 console = Console()
-def test_database_connection(db_config) -> tuple[bool, str]:
+def test_database_connection(db_config: AnyDatabaseConfig) -> tuple[bool, str]:
     """Test connectivity to a database.
     Returns:
@@ -17,16 +18,19 @@ def test_database_connection(db_config) -> tuple[bool, str]:
     try:
         conn = db_config.connect()
         # Run a simple query to verify the connection works
-        if db_config.dataset_id:
+        if hasattr(db_config, "dataset_id") and 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)"
+        elif list_databases := getattr(conn, "list_databases", None):
+            # If no dataset, list schemas in the database instead
+            schemas = list_databases()
+            schema_count = len(schemas)
+            return True, f"Connected successfully ({schema_count} schemas 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)"
+            # Fallback for backends that don't support list_tables and list_databases
+            return True, "Connected but unable to list neither datasets nor schemas"
     except Exception as e:
         return False, str(e)

nao_core/commands/init.py CHANGED Viewed

@@ -7,18 +7,24 @@ from rich.console import Console
 from rich.panel import Panel
 from rich.prompt import Confirm, Prompt
-from nao_core.config import AnyDatabaseConfig, BigQueryConfig, DatabaseType, LLMConfig, LLMProvider, NaoConfig, SlackConfig
+from nao_core.config import (
+    AnyDatabaseConfig,
+    BigQueryConfig,
+    DatabaseType,
+    DatabricksConfig,
+    DuckDBConfig,
+    LLMConfig,
+    LLMProvider,
+    NaoConfig,
+    SlackConfig,
+    SnowflakeConfig,
+)
+from nao_core.config.exceptions import InitError
 from nao_core.config.repos import RepoConfig
 console = Console()
-class InitError(Exception):
-    """Base exception for init command errors."""
-    pass
 class EmptyProjectNameError(InitError):
     """Raised when project name is empty."""
@@ -77,27 +83,22 @@ def setup_project_name(force: bool = False) -> tuple[str, Path]:
 def setup_bigquery() -> BigQueryConfig:
     """Setup a BigQuery database configuration."""
-    console.print("\n[bold cyan]BigQuery Configuration[/bold cyan]\n")
+    return BigQueryConfig.promptConfig()
-    name = Prompt.ask("[bold]Connection name[/bold]", default="bigquery-prod")
+def setup_duckdb() -> DuckDBConfig:
+    """Setup a DuckDB database configuration."""
+    return DuckDBConfig.promptConfig()
-    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="")
+def setup_databricks() -> DatabricksConfig:
+    """Setup a Databricks database configuration."""
+    return DatabricksConfig.promptConfig()
-    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_snowflake() -> SnowflakeConfig:
+    """Setup a Snowflake database configuration."""
+    return SnowflakeConfig.promptConfig()
 def setup_databases() -> list[AnyDatabaseConfig]:
@@ -124,6 +125,21 @@ def setup_databases() -> list[AnyDatabaseConfig]:
             databases.append(db_config)
             console.print(f"\n[bold green]✓[/bold green] Added database [cyan]{db_config.name}[/cyan]")
+        elif db_type == DatabaseType.DUCKDB.value:
+            db_config = setup_duckdb()
+            databases.append(db_config)
+            console.print(f"\n[bold green]✓[/bold green] Added database [cyan]{db_config.name}[/cyan]")
+        elif db_type == DatabaseType.DATABRICKS.value:
+            db_config = setup_databricks()
+            databases.append(db_config)
+            console.print(f"\n[bold green]✓[/bold green] Added database [cyan]{db_config.name}[/cyan]")
+        elif db_type == DatabaseType.SNOWFLAKE.value:
+            db_config = setup_snowflake()
+            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

nao_core/commands/sync/accessors.py CHANGED Viewed

@@ -31,8 +31,7 @@ class DataAccessor(ABC):
     def get_table(self, conn: BaseBackend, dataset: str, table: str):
         """Helper to get an Ibis table reference."""
-        full_table_name = f"{dataset}.{table}"
-        return conn.table(full_table_name)
+        return conn.table(table, database=dataset)
 def truncate_middle(text: str, max_length: int) -> str:
@@ -57,7 +56,6 @@ class ColumnsAccessor(DataAccessor):
         try:
             t = self.get_table(conn, dataset, table)
             schema = t.schema()
             columns = list(schema.items())
             lines = [
@@ -79,6 +77,7 @@ class ColumnsAccessor(DataAccessor):
             return "\n".join(lines)
         except Exception as e:
+            print(e)
             return f"# {table}\n\nError fetching schema: {e}"

nao_core/commands/sync/databases.py CHANGED Viewed

@@ -85,6 +85,234 @@ def sync_bigquery(
     return datasets_synced, tables_synced
+def sync_duckdb(
+    db_config,
+    base_path: Path,
+    progress: Progress,
+    accessors: list[DataAccessor],
+) -> tuple[int, int]:
+    """Sync DuckDB database schema to markdown files.
+    Args:
+            db_config: The database configuration
+            base_path: Base output path
+            progress: Rich progress instance
+            accessors: List of data accessors to run
+    Returns:
+            Tuple of (schemas_synced, tables_synced)
+    """
+    conn = db_config.connect()
+    # Derive database name from path
+    if db_config.path == ":memory:":
+        db_name = "memory"
+    else:
+        db_name = Path(db_config.path).stem
+    db_path = base_path / "type=duckdb" / f"database={db_name}"
+    schemas_synced = 0
+    tables_synced = 0
+    # List all schemas in DuckDB
+    schemas = conn.list_databases()
+    schema_task = progress.add_task(
+        f"[dim]{db_config.name}[/dim]",
+        total=len(schemas),
+    )
+    for schema in schemas:
+        try:
+            all_tables = conn.list_tables(database=schema)
+        except Exception:
+            progress.update(schema_task, advance=1)
+            continue
+        # Filter tables based on include/exclude patterns
+        tables = [t for t in all_tables if db_config.matches_pattern(schema, t)]
+        # Skip schema if no tables match
+        if not tables:
+            progress.update(schema_task, advance=1)
+            continue
+        schema_path = db_path / f"schema={schema}"
+        schema_path.mkdir(parents=True, exist_ok=True)
+        schemas_synced += 1
+        table_task = progress.add_task(
+            f"  [cyan]{schema}[/cyan]",
+            total=len(tables),
+        )
+        for table in tables:
+            table_path = schema_path / f"table={table}"
+            table_path.mkdir(parents=True, exist_ok=True)
+            for accessor in accessors:
+                content = accessor.generate(conn, schema, table)
+                output_file = table_path / accessor.filename
+                output_file.write_text(content)
+            tables_synced += 1
+            progress.update(table_task, advance=1)
+        progress.update(schema_task, advance=1)
+    return schemas_synced, tables_synced
+def sync_databricks(
+    db_config,
+    base_path: Path,
+    progress: Progress,
+    accessors: list[DataAccessor],
+) -> tuple[int, int]:
+    """Sync Databricks database schema to markdown files.
+    Args:
+            db_config: The database configuration
+            base_path: Base output path
+            progress: Rich progress instance
+            accessors: List of data accessors to run
+    Returns:
+            Tuple of (schemas_synced, tables_synced)
+    """
+    conn = db_config.connect()
+    catalog = db_config.catalog or "main"
+    db_path = base_path / "type=databricks" / f"database={catalog}"
+    schemas_synced = 0
+    tables_synced = 0
+    if db_config.schema:
+        schemas = [db_config.schema]
+    else:
+        schemas = conn.list_databases()
+    schema_task = progress.add_task(
+        f"[dim]{db_config.name}[/dim]",
+        total=len(schemas),
+    )
+    for schema in schemas:
+        try:
+            all_tables = conn.list_tables(database=schema)
+        except Exception:
+            progress.update(schema_task, advance=1)
+            continue
+        # Filter tables based on include/exclude patterns
+        tables = [t for t in all_tables if db_config.matches_pattern(schema, t)]
+        # Skip schema if no tables match
+        if not tables:
+            progress.update(schema_task, advance=1)
+            continue
+        schema_path = db_path / f"schema={schema}"
+        schema_path.mkdir(parents=True, exist_ok=True)
+        schemas_synced += 1
+        table_task = progress.add_task(
+            f"  [cyan]{schema}[/cyan]",
+            total=len(tables),
+        )
+        for table in tables:
+            table_path = schema_path / f"table={table}"
+            table_path.mkdir(parents=True, exist_ok=True)
+            for accessor in accessors:
+                content = accessor.generate(conn, schema, table)
+                output_file = table_path / accessor.filename
+                output_file.write_text(content)
+            tables_synced += 1
+            progress.update(table_task, advance=1)
+        progress.update(schema_task, advance=1)
+    return schemas_synced, tables_synced
+def sync_snowflake(
+    db_config,
+    base_path: Path,
+    progress: Progress,
+    accessors: list[DataAccessor],
+) -> tuple[int, int]:
+    """Sync Snowflake database schema to markdown files.
+    Args:
+            db_config: The database configuration
+            base_path: Base output path
+            progress: Rich progress instance
+            accessors: List of data accessors to run
+    Returns:
+            Tuple of (schemas_synced, tables_synced)
+    """
+    conn = db_config.connect()
+    db_path = base_path / "type=snowflake" / f"database={db_config.database}"
+    schemas_synced = 0
+    tables_synced = 0
+    if db_config.schema:
+        schemas = [db_config.schema]
+    else:
+        schemas = conn.list_databases()
+    schema_task = progress.add_task(
+        f"[dim]{db_config.name}[/dim]",
+        total=len(schemas),
+    )
+    for schema in schemas:
+        try:
+            all_tables = conn.list_tables(database=schema)
+        except Exception:
+            progress.update(schema_task, advance=1)
+            continue
+        # Filter tables based on include/exclude patterns
+        tables = [t for t in all_tables if db_config.matches_pattern(schema, t)]
+        # Skip schema if no tables match
+        if not tables:
+            progress.update(schema_task, advance=1)
+            continue
+        schema_path = db_path / f"schema={schema}"
+        schema_path.mkdir(parents=True, exist_ok=True)
+        schemas_synced += 1
+        table_task = progress.add_task(
+            f"  [cyan]{schema}[/cyan]",
+            total=len(tables),
+        )
+        for table in tables:
+            table_path = schema_path / f"table={table}"
+            table_path.mkdir(parents=True, exist_ok=True)
+            for accessor in accessors:
+                content = accessor.generate(conn, schema, table)
+                output_file = table_path / accessor.filename
+                output_file.write_text(content)
+            tables_synced += 1
+            progress.update(table_task, advance=1)
+        progress.update(schema_task, advance=1)
+    return schemas_synced, tables_synced
 def sync_databases(databases: list, base_path: Path) -> tuple[int, int]:
     """Sync all configured databases.
@@ -119,11 +347,25 @@ def sync_databases(databases: list, base_path: Path) -> tuple[int, int]:
             accessor_names = [a.filename.replace(".md", "") for a in db_accessors]
             try:
+                console.print(f"[dim]{db.name} accessors:[/dim] {', '.join(accessor_names)}")
                 if db.type == "bigquery":
-                    console.print(f"[dim]{db.name} accessors:[/dim] {', '.join(accessor_names)}")
                     datasets, tables = sync_bigquery(db, base_path, progress, db_accessors)
                     total_datasets += datasets
                     total_tables += tables
+                elif db.type == "duckdb":
+                    schemas, tables = sync_duckdb(db, base_path, progress, db_accessors)
+                    total_datasets += schemas
+                    total_tables += tables
+                elif db.type == "databricks":
+                    console.print(f"[dim]{db.name} accessors:[/dim] {', '.join(accessor_names)}")
+                    schemas, tables = sync_databricks(db, base_path, progress, db_accessors)
+                    total_datasets += schemas
+                    total_tables += tables
+                elif db.type == "snowflake":
+                    console.print(f"[dim]{db.name} accessors:[/dim] {', '.join(accessor_names)}")
+                    schemas, tables = sync_snowflake(db, base_path, progress, db_accessors)
+                    total_datasets += schemas
+                    total_tables += tables
                 else:
                     console.print(f"[yellow]⚠ Unsupported database type: {db.type}[/yellow]")
             except Exception as e:

nao_core/config/__init__.py CHANGED Viewed

@@ -1,5 +1,14 @@
 from .base import NaoConfig
-from .databases import AccessorType, AnyDatabaseConfig, BigQueryConfig, DatabaseType
+from .databases import (
+    AccessorType,
+    AnyDatabaseConfig,
+    BigQueryConfig,
+    DatabaseType,
+    DatabricksConfig,
+    DuckDBConfig,
+    SnowflakeConfig,
+)
+from .exceptions import InitError
 from .llm import LLMConfig, LLMProvider
 from .slack import SlackConfig
@@ -8,8 +17,12 @@ __all__ = [
     "AccessorType",
     "AnyDatabaseConfig",
     "BigQueryConfig",
+    "DuckDBConfig",
+    "DatabricksConfig",
+    "SnowflakeConfig",
     "DatabaseType",
     "LLMConfig",
     "LLMProvider",
     "SlackConfig",
+    "InitError",
 ]

nao_core/config/databases/__init__.py CHANGED Viewed

@@ -1,20 +1,26 @@
+from typing import Annotated, Union
+from pydantic import Discriminator, Tag
 from .base import AccessorType, DatabaseConfig, DatabaseType
 from .bigquery import BigQueryConfig
+from .databricks import DatabricksConfig
+from .duckdb import DuckDBConfig
+from .snowflake import SnowflakeConfig
 # =============================================================================
 # Database Config Registry
 # =============================================================================
-# When adding more backends, convert this to a discriminated union:
-# AnyDatabaseConfig = Annotated[
-#     Union[
-#         Annotated[BigQueryConfig, Tag("bigquery")],
-#         Annotated[PostgresConfig, Tag("postgres")],
-#     ],
-#     Discriminator(lambda x: x.get("type", "bigquery")),
-# ]
-AnyDatabaseConfig = BigQueryConfig
+AnyDatabaseConfig = Annotated[
+    Union[
+        Annotated[BigQueryConfig, Tag("bigquery")],
+        Annotated[DatabricksConfig, Tag("databricks")],
+        Annotated[SnowflakeConfig, Tag("snowflake")],
+        Annotated[DuckDBConfig, Tag("duckdb")],
+    ],
+    Discriminator("type"),
+]
 def parse_database_config(data: dict) -> DatabaseConfig:
@@ -22,8 +28,23 @@ def parse_database_config(data: dict) -> DatabaseConfig:
     db_type = data.get("type")
     if db_type == "bigquery":
         return BigQueryConfig.model_validate(data)
+    elif db_type == "duckdb":
+        return DuckDBConfig.model_validate(data)
+    elif db_type == "databricks":
+        return DatabricksConfig.model_validate(data)
+    elif db_type == "snowflake":
+        return SnowflakeConfig.model_validate(data)
     else:
         raise ValueError(f"Unknown database type: {db_type}")
-__all__ = ["AccessorType", "DatabaseConfig", "DatabaseType", "BigQueryConfig", "AnyDatabaseConfig"]
+__all__ = [
+    "AccessorType",
+    "AnyDatabaseConfig",
+    "BigQueryConfig",
+    "DuckDBConfig",
+    "DatabaseConfig",
+    "DatabaseType",
+    "DatabricksConfig",
+    "SnowflakeConfig",
+]

nao_core/config/databases/base.py CHANGED Viewed

@@ -4,12 +4,18 @@ from enum import Enum
 from ibis import BaseBackend
 from pydantic import BaseModel, Field
+from rich.console import Console
+console = Console()
 class DatabaseType(str, Enum):
     """Supported database types."""
     BIGQUERY = "bigquery"
+    DUCKDB = "duckdb"
+    DATABRICKS = "databricks"
+    SNOWFLAKE = "snowflake"
 class AccessorType(str, Enum):

nao_core/config/databases/bigquery.py CHANGED Viewed

@@ -3,8 +3,11 @@ from typing import Literal
 import ibis
 from ibis import BaseBackend
 from pydantic import Field
+from rich.prompt import Prompt
-from .base import DatabaseConfig
+from nao_core.config.exceptions import InitError
+from .base import DatabaseConfig, console
 class BigQueryConfig(DatabaseConfig):
@@ -20,6 +23,31 @@ class BigQueryConfig(DatabaseConfig):
     sso: bool = Field(default=False, description="Use Single Sign-On (SSO) for authentication")
     location: str | None = Field(default=None, description="BigQuery location")
+    @classmethod
+    def promptConfig(cls) -> "BigQueryConfig":
+        """Interactively prompt the user for BigQuery 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 connect(self) -> BaseBackend:
         """Create an Ibis BigQuery connection."""
         kwargs: dict = {"project_id": self.project_id}

nao_core/config/databases/databricks.py ADDED Viewed

@@ -0,0 +1,69 @@
+from typing import Literal
+import ibis
+from ibis import BaseBackend
+from pydantic import Field
+from rich.prompt import Prompt
+from nao_core.config.exceptions import InitError
+from .base import DatabaseConfig, console
+class DatabricksConfig(DatabaseConfig):
+    """Databricks-specific configuration."""
+    type: Literal["databricks"] = "databricks"
+    server_hostname: str = Field(description="Databricks server hostname (e.g., 'adb-xxxx.azuredatabricks.net')")
+    http_path: str = Field(description="HTTP path to the SQL warehouse or cluster")
+    access_token: str = Field(description="Databricks personal access token")
+    catalog: str | None = Field(default=None, description="Unity Catalog name (optional)")
+    schema: str | None = Field(default=None, description="Default schema (optional)")
+    @classmethod
+    def promptConfig(cls) -> "DatabricksConfig":
+        """Interactively prompt the user for Databricks configuration."""
+        console.print("\n[bold cyan]Databricks Configuration[/bold cyan]\n")
+        name = Prompt.ask("[bold]Connection name[/bold]", default="databricks-prod")
+        server_hostname = Prompt.ask("[bold]Server hostname[/bold] [dim](e.g., adb-xxxx.azuredatabricks.net)[/dim]")
+        if not server_hostname:
+            raise InitError("Server hostname cannot be empty.")
+        http_path = Prompt.ask("[bold]HTTP path[/bold] [dim](e.g., /sql/1.0/warehouses/xxxx)[/dim]")
+        if not http_path:
+            raise InitError("HTTP path cannot be empty.")
+        access_token = Prompt.ask("[bold]Access token[/bold]", password=True)
+        if not access_token:
+            raise InitError("Access token cannot be empty.")
+        catalog = Prompt.ask("[bold]Catalog[/bold] [dim](optional, press Enter to skip)[/dim]", default=None)
+        schema = Prompt.ask("[bold]Default schema[/bold] [dim](optional, press Enter to skip)[/dim]", default=None)
+        return DatabricksConfig(
+            name=name,
+            server_hostname=server_hostname,
+            http_path=http_path,
+            access_token=access_token,
+            catalog=catalog,
+            schema=schema,
+        )
+    def connect(self) -> BaseBackend:
+        """Create an Ibis Databricks connection."""
+        kwargs: dict = {
+            "server_hostname": self.server_hostname,
+            "http_path": self.http_path,
+            "access_token": self.access_token,
+        }
+        if self.catalog:
+            kwargs["catalog"] = self.catalog
+        if self.schema:
+            kwargs["schema"] = self.schema
+        return ibis.databricks.connect(**kwargs)

nao-core 0.0.28__py3-none-any.whl → 0.0.30__py3-none-any.whl

nao-core 0.0.28py3-none-any.whl → 0.0.30py3-none-any.whl