tai-sql 0.7.13__tar.gz → 0.7.14__tar.gz

{tai_sql-0.7.13 → tai_sql-0.7.14}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tai-sql
-Version: 0.7.13
+Version: 0.7.14
 Summary: SQL database management and code generation tool
 License-Expression: MIT
 License-File: LICENSE

{tai_sql-0.7.13 → tai_sql-0.7.14}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "tai-sql"
-version = "0.7.13"
+version = "0.7.14"
 description = "SQL database management and code generation tool"
 license = "MIT"
 license-files = ["LICENSE"]

{tai_sql-0.7.13 → tai_sql-0.7.14}/tai_sql/cli/commands/cmd_schema/schemas/chatbot.py

@@ -59,9 +59,12 @@ class ChatbotSchema(SchemaTemplate):
                 title: col[str] = column(description='Título de la conversación')
                 username: col[str] = column(description='ID del usuario propietario del chat')
                 is_active: col[bool] = column(default=True, description='Estado activo del chat')
+
+                visible: col[bool] = column(default=True, index=True, description='Si no, la conversación no se lista')
                 last_message_timestamp: col[datetime | None] = column(description='Timestamp del último mensaje del chat')
 
                 messages: onetomany[Mensaje] # Relación one-to-many con la tabla Mensaje
+                token_usage: onetomany[TokenUsage] # El consumo de la conversación, tenga mensaje o no
 
 
             class Feedback(Enum):
@@ -118,11 +121,13 @@ class ChatbotSchema(SchemaTemplate):
 
                 username: col[str] = column(index=True, description='Usuario que consumió')
                 message_id: col[bigint | None] = column(description='Mensaje que lo generó; vacío si se borró la conversación')
+                chat_id: col[bigint | None] = column(index=True, description='Conversación en la que se consumió; vacío si no es de ninguna')
 
                 tarifa_id: col[bigint | None] = column(index=True, description='Tarifa con la que se valora')
                 bolsa_id: col[bigint | None] = column(index=True, description='Bolsa a la que se carga; vacío si no hay bolsas')
 
                 message: manytoone[Mensaje] = relation(fields=['message_id'], references=['id'], backref='token_usage', onDelete='set null')
+                chat: manytoone[Chat] = relation(fields=['chat_id'], references=['id'], backref='token_usage', onDelete='set null')
 
 
             class ToolCall(Table):
@@ -351,10 +356,10 @@ class ChatbotSchema(SchemaTemplate):
                 con_tarifa: col[bool]
 
 
-            class UsuarioFacturacion(View):
-                """Vista que muestra los usuarios a los que se factura"""
-                __tablename__ = "usuario_facturacion"
-                __query__ = query('usuario_facturacion.sql')
+            class UsuarioConsumo(View):
+                """Vista que muestra los usuarios a los que se mide el consumo"""
+                __tablename__ = "usuario_consumo"
+                __query__ = query('usuario_consumo.sql')
 
                 username: col[str]
                 realm_name: col[str]
@@ -434,7 +439,7 @@ class ChatbotSchema(SchemaTemplate):
                 SUM(ROUND(((tu.prompt_tokens * t.precio_input_eur_millon
                           + tu.completion_tokens * t.precio_output_eur_millon)
                           * t.margen / 1000000)::numeric, 8))::double precision as total_cost_eur,
-                COUNT(DISTINCT me.chat_id) as chat_count,
+                COUNT(DISTINCT COALESCE(tu.chat_id, me.chat_id)) as chat_count,
                 COUNT(*) as message_count
             FROM token_usage tu
             JOIN tarifa t ON t.id = tu.tarifa_id
@@ -609,8 +614,8 @@ class ChatbotSchema(SchemaTemplate):
                 tm.active_days;
         ''').rstrip()
 
-        usuario_facturacion_sql = dedent('''\
-            -- Vista con los usuarios a los que se factura. Es la que leen las pantallas de facturación:
+        usuario_consumo_sql = dedent('''\
+            -- Vista con los usuarios a los que se mide el consumo. Es la que leen sus pantallas:
             -- /rbac/{realm}/user pide administración del realm y la contraseña no sale de aquí
             SELECT
                 u.username,
@@ -634,4 +639,4 @@ class ChatbotSchema(SchemaTemplate):
         (views_dir / 'tarifa_vigente.sql').write_text(tarifa_vigente_sql, encoding='utf-8')
         (views_dir / 'tokens_mensaje.sql').write_text(tokens_mensaje_sql, encoding='utf-8')
         (views_dir / 'user_stats.sql').write_text(user_stats_sql, encoding='utf-8')
-        (views_dir / 'usuario_facturacion.sql').write_text(usuario_facturacion_sql, encoding='utf-8')
+        (views_dir / 'usuario_consumo.sql').write_text(usuario_consumo_sql, encoding='utf-8')