chatsbom 0.2.2__py3-none-any.whl → 0.2.4__py3-none-any.whl

chatsbom/commands/index.py

@@ -244,29 +244,42 @@ def main(
 ):
     """Index SBOM data into the database."""
 
+    # Check ClickHouse connection and ensure database exists
+    from chatsbom.core.clickhouse import check_clickhouse_connection
+
+    # First check basic connectivity
+    check_clickhouse_connection(
+        host=host,
+        port=port,
+        user=user,
+        password=password,
+        database='default',
+        console=console,
+        require_database=False,
+    )
+
     # Ensure database exists
     try:
-        # Connect to default database first
         tmp_client = clickhouse_connect.get_client(
             host=host, port=port, username=user, password=password, database='default',
         )
         tmp_client.command(f"CREATE DATABASE IF NOT EXISTS {database}")
+        console.print(
+            f'[green]Database [cyan]{database}[/cyan] is ready.[/green]',
+        )
     except Exception as e:
         console.print(
-            f'[bold red]Error:[/] Failed to connect to ClickHouse at '
-            f'[cyan]{host}:{port}[/]\n\n'
-            f'Details: {e}\n\n'
-            'Please ensure:\n'
-            '  1. ClickHouse is running: [cyan]docker compose up -d[/]\n'
-            '  2. Host and port are correct\n'
-            '  3. User credentials are valid',
+            f'[bold red]Error:[/] Failed to create database [cyan]{database}[/]\n\n'
+            f'Details: {e}',
         )
         raise typer.Exit(1)
 
     try:
         client = get_client(host, port, user, password, database)
     except Exception as e:
-        console.print(f"[red]Failed to connect to ClickHouse: {e}[/red]")
+        console.print(
+            f"[red]Failed to connect to database {database}: {e}[/red]",
+        )
         raise typer.Exit(code=1)
 
     if clean:

chatsbom/commands/query.py

@@ -68,8 +68,8 @@ def main(
 
     candidate_query = f"""
     SELECT a.name, count() as cnt
-    FROM {db_config.artifacts_table} a
-    JOIN {db_config.repositories_table} r ON a.repository_id = r.id
+    FROM {db_config.artifacts_table} FINAL a
+    JOIN {db_config.repositories_table} FINAL r ON a.repository_id = r.id
     WHERE a.name ILIKE {{pattern:String}} {lang_filter}
     GROUP BY a.name
     ORDER BY cnt DESC
@@ -109,8 +109,8 @@ def main(
     # Count total results first
     count_query = f"""
     SELECT count()
-    FROM {db_config.artifacts_table} a
-    JOIN {db_config.repositories_table} r ON a.repository_id = r.id
+    FROM {db_config.artifacts_table} FINAL a
+    JOIN {db_config.repositories_table} FINAL r ON a.repository_id = r.id
     WHERE a.name = {{library:String}} {lang_filter}
     """
 
@@ -139,8 +139,8 @@ def main(
         r.stars,
         a.version,
         r.url
-    FROM {db_config.artifacts_table} AS a
-    JOIN {db_config.repositories_table} AS r ON a.repository_id = r.id
+    FROM {db_config.artifacts_table} FINAL AS a
+    JOIN {db_config.repositories_table} FINAL AS r ON a.repository_id = r.id
     WHERE a.name = {{library:String}} {lang_filter}
     ORDER BY r.stars DESC
     LIMIT {{limit:UInt32}}

chatsbom/commands/status.py

@@ -34,11 +34,19 @@ def main(
         database=database or config.database.database,
     )
 
-    try:
-        repo = SBOMRepository(db_config)
-    except Exception as e:
-        console.print(f"[red]Failed to connect to ClickHouse: {e}[/red]")
-        raise typer.Exit(code=1)
+    # Check ClickHouse connection before proceeding
+    from chatsbom.core.clickhouse import check_clickhouse_connection
+    check_clickhouse_connection(
+        host=db_config.host,
+        port=db_config.port,
+        user=db_config.user,
+        password=db_config.password,
+        database=db_config.database,
+        console=console,
+        require_database=True,
+    )
+
+    repo = SBOMRepository(db_config)
 
     # 1. Total Counts
     try:
@@ -91,7 +99,7 @@ def main(
     # 4. Dependency Types Distribution
     try:
         type_res = repo.client.query(
-            f'SELECT type, count() as cnt FROM {db_config.artifacts_table} GROUP BY type ORDER BY cnt DESC',
+            f'SELECT type, count() as cnt FROM {db_config.artifacts_table} FINAL GROUP BY type ORDER BY cnt DESC',
         ).result_rows
     except Exception as e:
         console.print(
@@ -112,7 +120,7 @@ def main(
     try:
         lang_counts = {
             row[0]: row[1] for row in repo.client.query(
-                f'SELECT language, count() FROM {db_config.repositories_table} GROUP BY language',
+                f'SELECT language, count() FROM {db_config.repositories_table} FINAL GROUP BY language',
             ).result_rows
         }
     except Exception as e:
@@ -143,8 +151,8 @@ def main(
                 # Query count of projects in this language using this framework
                 fw_count_query = f"""
                 SELECT count(DISTINCT r.id)
-                FROM {db_config.repositories_table} r
-                JOIN {db_config.artifacts_table} a ON r.id = a.repository_id
+                FROM {db_config.repositories_table} FINAL r
+                JOIN {db_config.artifacts_table} FINAL a ON r.id = a.repository_id
                 WHERE r.language = {{lang:String}} AND a.name IN {{pkgs:Array(String)}}
                 """
                 fw_count = repo.client.query(
@@ -194,8 +202,8 @@ def main(
                 # Query top 3 repositories for this framework
                 top_repos_query = f"""
                 SELECT DISTINCT r.full_name, r.stars, r.url
-                FROM {db_config.repositories_table} r
-                JOIN {db_config.artifacts_table} a ON r.id = a.repository_id
+                FROM {db_config.repositories_table} FINAL r
+                JOIN {db_config.artifacts_table} FINAL a ON r.id = a.repository_id
                 WHERE r.language = {{lang:String}} AND a.name IN {{pkgs:Array(String)}}
                 ORDER BY r.stars DESC
                 LIMIT 3

chatsbom/core/repository.py

@@ -101,7 +101,7 @@ class SBOMRepository:
     def get_repository_count(self) -> int:
         """Get total number of repositories."""
         result = self.client.query(
-            f'SELECT count() FROM {self.config.repositories_table}',
+            f'SELECT count() FROM {self.config.repositories_table} FINAL',
         )
         return result.result_rows[0][0]
 
@@ -114,7 +114,7 @@ class SBOMRepository:
         """
         query = f"""
         SELECT language, count() as cnt
-        FROM {self.config.repositories_table}
+        FROM {self.config.repositories_table} FINAL
         GROUP BY language
         ORDER BY cnt DESC
         """
@@ -140,7 +140,7 @@ class SBOMRepository:
         where_clause = f"WHERE language = '{language}'" if language else ''
         query = f"""
         SELECT full_name, stars, language, description
-        FROM {self.config.repositories_table}
+        FROM {self.config.repositories_table} FINAL
         {where_clause}
         ORDER BY stars DESC
         LIMIT {limit}
@@ -201,7 +201,7 @@ class SBOMRepository:
     def get_artifact_count(self) -> int:
         """Get total number of artifacts."""
         result = self.client.query(
-            f'SELECT count() FROM {self.config.artifacts_table}',
+            f'SELECT count() FROM {self.config.artifacts_table} FINAL',
         )
         return result.result_rows[0][0]
 
@@ -232,8 +232,8 @@ class SBOMRepository:
             r.language,
             a.name,
             a.version
-        FROM {self.config.artifacts_table} a
-        JOIN {self.config.repositories_table} r ON a.repository_id = r.id
+        FROM {self.config.artifacts_table} FINAL a
+        JOIN {self.config.repositories_table} FINAL r ON a.repository_id = r.id
         WHERE a.name LIKE %(pattern)s
         {lang_filter}
         ORDER BY r.stars DESC
@@ -269,7 +269,7 @@ class SBOMRepository:
         if language:
             lang_filter = f"""
             WHERE repository_id IN (
-                SELECT id FROM {self.config.repositories_table}
+                SELECT id FROM {self.config.repositories_table} FINAL
                 WHERE language = '{language}'
             )
             """
@@ -278,7 +278,7 @@ class SBOMRepository:
         SELECT
             name,
             count(DISTINCT repository_id) as repo_count
-        FROM {self.config.artifacts_table}
+        FROM {self.config.artifacts_table} FINAL
         {lang_filter}
         GROUP BY name
         ORDER BY repo_count DESC
@@ -311,8 +311,8 @@ class SBOMRepository:
             a.name,
             count(DISTINCT a.repository_id) as repo_count,
             sum(r.stars) as total_stars
-        FROM {self.config.artifacts_table} a
-        JOIN {self.config.repositories_table} r ON a.repository_id = r.id
+        FROM {self.config.artifacts_table} FINAL a
+        JOIN {self.config.repositories_table} FINAL r ON a.repository_id = r.id
         WHERE a.name IN ('{packages_str}')
         GROUP BY a.name
         ORDER BY repo_count DESC

{chatsbom-0.2.2.dist-info → chatsbom-0.2.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: chatsbom
-Version: 0.2.2
+Version: 0.2.4
 Summary: ChatSBOM - Talk to your Supply Chain. Chat with SBOMs.
 Requires-Python: >=3.12
 Requires-Dist: claude-agent-sdk>=0.1.0
@@ -79,7 +79,7 @@ uvx chatsbom index --language go
 uvx chatsbom status
 
 # 6. Query dependencies
-uvx chatsbom query gin
+uvx chatsbom query gin --language go
 
 # 7. Launch AI chat interface
 uvx chatsbom chat

{chatsbom-0.2.2.dist-info → chatsbom-0.2.4.dist-info}/RECORD

@@ -5,20 +5,20 @@ chatsbom/commands/chat.py,sha256=ns9DFTjGJwVD_m6K2m0Flg-LQSmmhAMKjnkzPPfBYYg,109
 chatsbom/commands/collect.py,sha256=ApweQztO889o4X4lBrxlIwdYorDUSiE01M4-XMO0HUM,16536
 chatsbom/commands/convert.py,sha256=etc6s_tT4VAvjOCZs8dnHppbTYEeAHI6kHJikkUWD64,8249
 chatsbom/commands/download.py,sha256=I9TwEXO1zrh9-DwvUUMh4iZ1W5ZCXbtE4jcJiK2KqnY,10218
-chatsbom/commands/index.py,sha256=B5kweeKgcHJ1ICbZ7y1f4qk9EjYoaRsAwWaYsghjEEI,10391
-chatsbom/commands/query.py,sha256=9mVot94ZeZ-rFZRfM3bkcNQBhWAuWqDaMJ36JcKT0zQ,5661
-chatsbom/commands/status.py,sha256=g0kEn4H2LYrh62_HVW72EHtICa9UlNKxHAr2oVh0tjY,8274
+chatsbom/commands/index.py,sha256=DlK5FtwOFoJqy9gA6mYvfuVQLL9S15KRrH7_Qw7l0Fk,10633
+chatsbom/commands/query.py,sha256=NfsQKjLwV0uCDMG89FujcPC3vc-x4wtkWcC5-gkXUoE,5697
+chatsbom/commands/status.py,sha256=WbDFzIP0gC8AFrByRPvtpnEI2X8phoU5Sii4-dantys,8541
 chatsbom/core/__init__.py,sha256=x8qFrlBTgFn7TkmBN11ysPCUzRWBY350c5AN4u-3dWE,17
 chatsbom/core/clickhouse.py,sha256=sw4gWiaC2BE4QJlVWDsoDULKFB6YslB0W1FM630IY94,3530
 chatsbom/core/client.py,sha256=GSnOOAxoSUKU1pLXPX4W3cM5G4Iv7fQnknW-YkcX43Q,1400
 chatsbom/core/config.py,sha256=K4vDe387iVwwMY9aTDbDLlUPVOHCKLcA5ZMK000MQgk,3992
-chatsbom/core/repository.py,sha256=HUvwOAR7wFzP0Pqx3zS0sJtNIZOKBmeEcoh83sSPiUc,9765
+chatsbom/core/repository.py,sha256=Un788vLvVnCNqUn9EgTplQB0--_yCVjRNMCfTzwhSm4,9825
 chatsbom/core/schema.py,sha256=oZUNNjS3UerZtIMo9-KOL-2E3XwS8FT9JNQStWq4tYE,1267
 chatsbom/core/validation.py,sha256=v4boiht4hK73nuZNYpzLTkZKsNSoFPMUDIX-8sP2E9g,3927
 chatsbom/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 chatsbom/models/framework.py,sha256=Whlsbnn64YwftMifMEMVvhsWADC9uupBV4QN1i1BoNQ,2833
 chatsbom/models/language.py,sha256=RgbBQZfG630uNZ7w-BtB1LZ5B5Q_nEExGrwUadfpRqE,3601
-chatsbom-0.2.2.dist-info/METADATA,sha256=HC4d-l40-TAlbqxEXi69UjS6laqZe2i7yrQKMY3yiSU,4100
-chatsbom-0.2.2.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-chatsbom-0.2.2.dist-info/entry_points.txt,sha256=906Ig6u2FwWk3ftNi4mth03N1NRgP4-B2p9kpppcqWA,51
-chatsbom-0.2.2.dist-info/RECORD,,
+chatsbom-0.2.4.dist-info/METADATA,sha256=twAfr2y7Ix3kkFEF4Q8nuwQQ-JZ77fYSpK_1OWJWi_w,4114
+chatsbom-0.2.4.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+chatsbom-0.2.4.dist-info/entry_points.txt,sha256=906Ig6u2FwWk3ftNi4mth03N1NRgP4-B2p9kpppcqWA,51
+chatsbom-0.2.4.dist-info/RECORD,,