tai-sql 0.7.17__tar.gz → 0.7.18__tar.gz

{tai_sql-0.7.17 → tai_sql-0.7.18}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tai-sql
-Version: 0.7.17
+Version: 0.7.18
 Summary: SQL database management and code generation tool
 License-Expression: LicenseRef-Proprietary
 License-File: LICENSE

{tai_sql-0.7.17 → tai_sql-0.7.18}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "tai-sql"
-version = "0.7.17"
+version = "0.7.18"
 description = "SQL database management and code generation tool"
 license = "LicenseRef-Proprietary"
 license-files = ["LICENSE"]

{tai_sql-0.7.17 → tai_sql-0.7.18}/tai_sql/cli/commands/cmd_schema/schemas/chatbot.py

@@ -285,6 +285,9 @@ class ChatbotSchema(SchemaTemplate):
 
                 message_id: col[bigint]
                 chat_id: col[bigint]
+                # El dueño del chat. Una vista no tiene relaciones por las que llegar a `chat`, así que
+                # una regla de propietario (tai-api `owner()`) necesita la columna aquí
+                username: col[str]
                 model_name: col[str | None]
                 provider: col[str | None]
                 prompt_tokens: col[int]
@@ -568,6 +571,7 @@ class ChatbotSchema(SchemaTemplate):
             SELECT
                 tu.message_id,
                 m.chat_id,
+                c.username,
                 m.model_name,
                 m.provider,
                 SUM(tu.prompt_tokens) as prompt_tokens,
@@ -575,7 +579,8 @@ class ChatbotSchema(SchemaTemplate):
                 SUM(tu.total_tokens) as total_tokens
             FROM token_usage tu
             JOIN mensaje m ON m.id = tu.message_id
-            GROUP BY tu.message_id, m.chat_id, m.model_name, m.provider;
+            JOIN chat c ON c.id = m.chat_id
+            GROUP BY tu.message_id, m.chat_id, c.username, m.model_name, m.provider;
         ''').rstrip()
 
         user_stats_sql = dedent('''\