tai-sql 0.8.2__tar.gz → 0.8.3__tar.gz

{tai_sql-0.8.2 → tai_sql-0.8.3}/PKG-INFO

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

{tai_sql-0.8.2 → tai_sql-0.8.3}/pyproject.toml

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

{tai_sql-0.8.2 → tai_sql-0.8.3}/tai_sql/cli/commands/cmd_schema/schemas/chatbot.py

@@ -6,8 +6,8 @@ from .base import SchemaTemplate
 
 
 class ChatbotSchema(SchemaTemplate):
-    """Plantilla chatbot: Chat, Mensaje, TokenUsage, TokenUsageParte, ToolCall, Sugerencia, Modelo, Tarifa, Bolsa,
-    LimitePresupuesto y vistas"""
+    """Plantilla chatbot: Chat, Mensaje, TokenUsage, TokenUsageParte, ToolCall, Sugerencia, InformeConversacion, Modelo,
+    Tarifa, Bolsa, LimitePresupuesto y vistas"""
 
     def get_content(self) -> str:
         return dedent(f'''
@@ -57,7 +57,7 @@ class ChatbotSchema(SchemaTemplate):
 
                 id: col[bigint] = column(primary_key=True, autoincrement=True, description='UUID del chat')
                 title: col[str] = column(description='Título de la conversación')
-                username: col[str] = column(description='ID del usuario propietario del chat')
+                username: col[str] = column(index=True, 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')
@@ -70,6 +70,7 @@ class ChatbotSchema(SchemaTemplate):
 
                 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
+                informes: onetomany[InformeConversacion] # Sus informes: los resúmenes que se pidieron de ella
 
 
             class Feedback(Enum):
@@ -93,7 +94,7 @@ class ChatbotSchema(SchemaTemplate):
                 content: col[str] = column(description='Contenido del mensaje')
                 role: col[MessageRole] = column(description='Rol del mensaje (user, assistant, system)')
                 timestamp: col[datetime] = column(default=datetime.now, description='Timestamp del mensaje')
-                feedback: col[Feedback] = column(default=Feedback.GOOD, description='Feedback del mensaje')
+                feedback: col[Feedback | None] = column(description='La valoración del usuario (good o bad); vacía si no la valoró')
                 feedback_comment: col[str | None] = column(description='Comentario adicional del feedback')
                 reasoning: col[str | None] = column(description='Texto de razonamiento del agente antes de usar herramientas')
                 agent_iterations: col[int] = column(default=0, description='Número de iteraciones del agente')
@@ -101,11 +102,12 @@ class ChatbotSchema(SchemaTemplate):
                 model_name: col[str | None] = column(description='Modelo con el que se respondió')
                 provider: col[str | None] = column(description='Proveedor del modelo con el que se respondió')
 
-                chat_id: col[bigint] = column(description='ID del chat al que pertenece el mensaje')
+                chat_id: col[bigint] = column(index=True, description='ID del chat al que pertenece el mensaje')
 
                 token_usage: onetomany[TokenUsage] # Relación one-to-many con la tabla TokenUsage
                 tool_calls: onetomany[ToolCall] # Relación one-to-many con la tabla ToolCall
                 sugerencias: onetomany[Sugerencia] # Lo que se sugiere preguntar tras esta respuesta
+                informes: onetomany[InformeConversacion] # El informe que entrega este mensaje de sistema
 
                 chat: manytoone[Chat] = relation(fields=['chat_id'], references=['id'], backref='messages') # Relación many-to-one con Chat
 
@@ -123,10 +125,10 @@ class ChatbotSchema(SchemaTemplate):
                 prompt_tokens: col[int] = column(default=0, description='Tokens consumidos en el prompt')
                 completion_tokens: col[int] = column(default=0, description='Tokens consumidos en la respuesta')
                 total_tokens: col[int] = column(default=0, description='Total de tokens consumidos')
-                timestamp: col[datetime] = column(default=datetime.now, description='Timestamp del consumo')
+                timestamp: col[datetime] = column(default=datetime.now, index=True, description='Timestamp del consumo')
 
                 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')
+                message_id: col[bigint | None] = column(index=True, 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')
@@ -150,6 +152,10 @@ class ChatbotSchema(SchemaTemplate):
             class ToolCall(Table):
                 """Tabla que almacena las llamadas a herramientas realizadas por el agente durante la generación de respuestas"""
                 __tablename__ = "tool_call"
+                # El call_id es único dentro de su mensaje (el del modelo, no el de la fila): así una llamada de
+                # un sub-agente puede apuntar a la que lo creó. El índice de esta restricción sirve también para
+                # buscar las llamadas de un mensaje, que empieza por message_id
+                __constraints__ = unique_constraint("message_id", "call_id")
 
                 id: col[bigint] = column(primary_key=True, autoincrement=True, description='ID de la llamada a herramienta')
                 tool_name: col[str] = column(description='Nombre de la herramienta invocada')
@@ -157,10 +163,18 @@ class ChatbotSchema(SchemaTemplate):
                 tool_result: col[dict | None] = column(description='Resultado de la herramienta en formato JSON')
                 duration_ms: col[int | None] = column(description='Duración de la ejecución en milisegundos')
                 success: col[bool] = column(default=True, description='Si la llamada fue exitosa')
+                call_id: col[str | None] = column(description='El id de la llamada que hizo el modelo')
+                padre: col[str | None] = column(description='El call_id de la herramienta que creó el sub-agente que hizo esta llamada; vacío en las del agente principal')
                 timestamp: col[datetime] = column(default=datetime.now, description='Timestamp de la llamada')
                 message_id: col[bigint] = column(description='ID del mensaje asociado')
 
                 message: manytoone[Mensaje] = relation(fields=['message_id'], references=['id'], backref='tool_calls') # Relación many-to-one con Mensaje
+                # La llamada que creó el sub-agente que hizo esta: del mismo mensaje, por la clave compuesta
+                # (message_id, padre) → (message_id, call_id). Por el id del modelo y no por el de la fila: las
+                # filas se crean anidadas en el mensaje, y el id de la del padre todavía no existe. Con padre
+                # vacío (las del agente principal) la clave no se comprueba. Borrar una se lleva sus hijas
+                llamada_padre: manytoone[ToolCall] = relation(fields=['message_id', 'padre'], references=['message_id', 'call_id'], backref='hijas', onDelete='cascade')
+                hijas: onetomany[ToolCall]
 
 
             class TokenUsageParte(Table):
@@ -191,6 +205,33 @@ class ChatbotSchema(SchemaTemplate):
                 message: manytoone[Mensaje] = relation(fields=['message_id'], references=['id'], backref='sugerencias', onDelete='cascade')
 
 
+            class InformeConversacion(Table):
+                """Tabla con los informes de una conversación: su resumen, redactado por el modelo; se descarga en DOCX o en PDF"""
+                __tablename__ = "informe_conversacion"
+
+                id: col[bigint] = column(primary_key=True, autoincrement=True)
+                extension: col[str] = column(description="'corto', 'medio' o 'largo'")
+                indicaciones: col[str | None] = column(description='Lo que pidió quien lo pidió (centrarse en algo, las tablas…); vacío si nada. No sale en el documento')
+                estado: col[str] = column(index=True, description="'preparando', 'listo' o 'error'")
+                error: col[str | None] = column(description='Por qué no se pudo preparar')
+                titulo: col[str | None] = column(description='El título que le puso el modelo')
+                contenido: col[str | None] = column(description='El informe, en markdown: el documento se pinta al descargarlo')
+                hasta_mensaje_id: col[bigint | None] = column(description='El último mensaje que resume: con el mismo, se reutiliza sin llamar al modelo')
+                mensajes: col[int] = column(default=0, description='Cuántos mensajes resume')
+                compactado: col[bool] = column(default=False, description='Si la conversación no cabía y se resumió por tandas antes de redactarlo')
+                reutilizado_de: col[bigint | None] = column(description='El informe del que se copió el contenido, sin llamar al modelo')
+                model_name: col[str | None] = column(description='Modelo con el que se redactó')
+                provider: col[str | None] = column(description='Proveedor del modelo con el que se redactó')
+                timestamp: col[datetime] = column(default=datetime.now, description='Cuándo se pidió')
+                chat_id: col[bigint] = column(index=True, description='La conversación que resume')
+                message_id: col[bigint | None] = column(index=True, description='El mensaje de sistema que lo entrega; vacío mientras se prepara')
+
+                # `chat` y `message`, como en TokenUsage: los triggers en cascada comparan las relaciones
+                # por su nombre, y estos no repiten el de ningún padre
+                chat: manytoone[Chat] = relation(fields=['chat_id'], references=['id'], backref='informes', onDelete='cascade')
+                message: manytoone[Mensaje] = relation(fields=['message_id'], references=['id'], backref='informes', onDelete='set null')
+
+
             class Modelo(Table):
                 """Tabla que almacena los modelos con los que puede responder el agente"""
                 __tablename__ = "modelo"
@@ -278,6 +319,8 @@ class ChatbotSchema(SchemaTemplate):
                 total_completion_tokens: col[int]
                 total_tokens: col[int]
                 total_cost_eur: col[float]
+                total_cache_read: col[int | None]
+                total_cache_write: col[int | None]
                 chat_count: col[int]
                 message_count: col[int]
 
@@ -295,6 +338,8 @@ class ChatbotSchema(SchemaTemplate):
                 total_completion_tokens: col[int]
                 total_tokens: col[int]
                 coste_eur: col[float]
+                total_cache_read: col[int | None]
+                total_cache_write: col[int | None]
                 message_count: col[int]
 
 
@@ -379,7 +424,7 @@ class ChatbotSchema(SchemaTemplate):
                 content: col[str]
                 # 'good' o 'bad'
                 feedback: col[str]
-                feedback_comment: col[str]
+                feedback_comment: col[str | None]
                 total_duration_ms: col[int | None]
                 model_name: col[str | None]
                 chat_id: col[bigint]
@@ -517,6 +562,10 @@ class ChatbotSchema(SchemaTemplate):
                               + COALESCE(tu.cache_write, 0) * (COALESCE(t.precio_cache_write_eur_millon, t.precio_input_eur_millon) - t.precio_input_eur_millon)
                               + tu.completion_tokens * t.precio_output_eur_millon)
                               * t.margen / 1000000)::numeric, 8))::double precision as total_cost_eur,
+                    -- Lo leído de la caché del proveedor y lo escrito en ella: parte de la entrada. Sin COALESCE: un
+                    -- día o un mes sin ningún apunte que lo diga (anteriores a apuntarlo) queda vacío, no a 0
+                    SUM(tu.cache_read) as total_cache_read,
+                    SUM(tu.cache_write) as total_cache_write,
                     COUNT(DISTINCT COALESCE(tu.chat_id, me.chat_id)) as chat_count
                 FROM token_usage tu
                 JOIN tarifa t ON t.id = tu.tarifa_id
@@ -557,6 +606,8 @@ class ChatbotSchema(SchemaTemplate):
                 COALESCE(co.total_completion_tokens, 0) as total_completion_tokens,
                 COALESCE(co.total_tokens, 0) as total_tokens,
                 COALESCE(co.total_cost_eur, 0) as total_cost_eur,
+                co.total_cache_read,
+                co.total_cache_write,
                 COALESCE(co.chat_count, 0) as chat_count,
                 COALESCE(e.message_count, 0) as message_count
             FROM consumo co
@@ -586,7 +637,11 @@ class ChatbotSchema(SchemaTemplate):
                               + COALESCE(tu.cache_read, 0) * (COALESCE(t.precio_cache_input_eur_millon, t.precio_input_eur_millon) - t.precio_input_eur_millon)
                               + COALESCE(tu.cache_write, 0) * (COALESCE(t.precio_cache_write_eur_millon, t.precio_input_eur_millon) - t.precio_input_eur_millon)
                               + tu.completion_tokens * t.precio_output_eur_millon)
-                              * t.margen / 1000000)::numeric, 8))::double precision as coste_eur
+                              * t.margen / 1000000)::numeric, 8))::double precision as coste_eur,
+                    -- Lo leído de la caché del proveedor y lo escrito en ella: parte de la entrada. Sin COALESCE: un
+                    -- día o un mes sin ningún apunte que lo diga (anteriores a apuntarlo) queda vacío, no a 0
+                    SUM(tu.cache_read) as total_cache_read,
+                    SUM(tu.cache_write) as total_cache_write
                 FROM token_usage tu
                 JOIN tarifa t ON t.id = tu.tarifa_id
                 JOIN modelo m ON m.id = t.modelo_id
@@ -625,6 +680,8 @@ class ChatbotSchema(SchemaTemplate):
                 COALESCE(co.total_completion_tokens, 0) as total_completion_tokens,
                 COALESCE(co.total_tokens, 0) as total_tokens,
                 COALESCE(co.coste_eur, 0) as coste_eur,
+                co.total_cache_read,
+                co.total_cache_write,
                 COALESCE(e.message_count, 0) as message_count
             FROM consumo co
             FULL JOIN enviados e
@@ -716,9 +773,8 @@ class ChatbotSchema(SchemaTemplate):
         ''').rstrip()
 
         valoracion_sql = dedent('''\
-            -- Vista con las respuestas que el usuario valoró con un comentario, con su conversación
-            -- y lo que se apuntó con ellas. Solo las que tienen comentario: la columna feedback nace
-            -- 'good', así que una valoración sin texto no se distingue de un mensaje sin valorar
+            -- Vista con las respuestas que el usuario valoró (con comentario o sin él), con su
+            -- conversación y lo que se apuntó con ellas. Un mensaje sin valorar tiene feedback vacío
             SELECT
                 m.id as message_id,
                 m.timestamp,
@@ -734,8 +790,7 @@ class ChatbotSchema(SchemaTemplate):
             FROM mensaje m
             JOIN chat c ON c.id = m.chat_id
             WHERE m.role = 'asistente'
-              AND m.feedback_comment IS NOT NULL
-              AND trim(m.feedback_comment) <> ''
+              AND m.feedback IS NOT NULL
             ORDER BY m.timestamp DESC;
         ''').rstrip()
 

{tai_sql-0.8.2 → tai_sql-0.8.3}/tai_sql/sync/README.md

@@ -94,18 +94,23 @@ En [plan/planner.py](tai_sql/sync/plan/planner.py), y es:
  4. DROP FK .............. las que el schema ya no declara
  5. DROP FK .............. las que rompe un cambio de tipo
  6. columnas ............. ADD / DROP / ALTER
- 7. ADD FK ............... de tablas nuevas y de tablas existentes
- 8. ALTER FK ............. cambio de acción en cascada
- 9. ADD FK ............... recreación tras el cambio de tipo
-10. DROP CONSTRAINT UNIQUE  antes del cambio de PK, que puede necesitar sus columnas
-11. ALTER PRIMARY KEY .... DROP de la vieja + ADD de la nueva
-12. ADD CONSTRAINT UNIQUE
+ 7. DROP CONSTRAINT UNIQUE  antes del cambio de PK, que puede necesitar sus columnas
+ 8. ALTER PRIMARY KEY .... DROP de la vieja + ADD de la nueva
+ 9. ADD CONSTRAINT UNIQUE
+10. ADD FK ............... de tablas nuevas y de tablas existentes
+11. ALTER FK ............. cambio de acción en cascada
+12. ADD FK ............... recreación tras el cambio de tipo
 13. DROP INDEX ........... solo los que creó tai-sql
 14. CREATE INDEX ......... de tablas nuevas y de tablas existentes
 15. redefinir vista ...... en orden de dependencias
 16. CREATE VIEW .......... en orden de dependencias
 ```
 
+La unicidad y la clave primaria van antes de las FKs: una FK solo se crea hacia columnas con una
+restricción única o la clave primaria, y una FK hacia una restricción nueva del mismo push (una FK
+compuesta a la propia tabla) fallaba por llegar antes. En una tabla nueva no pasaba, porque la
+UNIQUE va dentro del `CREATE TABLE`.
+
 Antes el orden real no era el documentado: la generación del `CREATE TABLE` invocaba a la de
 claves foráneas en su última línea, así que las FKs de las tablas nuevas quedaban antes de los
 DROP y de los ALTER.

{tai_sql-0.8.2 → tai_sql-0.8.3}/tai_sql/sync/plan/planner.py

@@ -19,17 +19,24 @@ El orden es:
     4. DROP FK ...................... las que el schema ya no declara
     5. DROP FK ...................... las que rompe un cambio de tipo
     6. columnas ..................... ADD / DROP / ALTER
-    7. ADD FK ....................... de tablas nuevas y de tablas existentes
-    8. ALTER FK ..................... cambio de acción en cascada
-    9. ADD FK ....................... recreación tras el cambio de tipo
-   10. DROP CONSTRAINT UNIQUE ....... antes del cambio de PK, que puede necesitar sus columnas
-   11. ALTER PRIMARY KEY ............ DROP de la vieja + ADD de la nueva
-   12. ADD CONSTRAINT UNIQUE
+    7. DROP CONSTRAINT UNIQUE ....... antes del cambio de PK, que puede necesitar sus columnas
+    8. ALTER PRIMARY KEY ............ DROP de la vieja + ADD de la nueva
+    9. ADD CONSTRAINT UNIQUE
+   10. ADD FK ....................... de tablas nuevas y de tablas existentes
+   11. ALTER FK ..................... cambio de acción en cascada
+   12. ADD FK ....................... recreación tras el cambio de tipo
    13. DROP INDEX ................... solo los que creó tai-sql
    14. CREATE INDEX ................. de tablas nuevas y de tablas existentes
    15. redefinir vista ............. en orden de dependencias
    16. CREATE VIEW ................. en orden de dependencias
 
+La unicidad y la clave primaria van **antes** de las FKs: una FK solo se puede crear hacia
+columnas con una restricción única o la clave primaria, y si esa restricción es nueva en el
+mismo push —una FK compuesta a la propia tabla, como `(message_id, padre)` →
+`(message_id, call_id)`— la FK fallaba por llegar antes («there is no unique constraint
+matching given keys»). En una tabla nueva no pasaba: la UNIQUE va dentro del CREATE TABLE.
+Lo que esos pasos quitan ya no tiene FKs que lo usen: las sueltan los pasos 4 y 5.
+
 El planner **no toca la base de datos**. Lo único que hace es pedirle al driver el texto de
 cada sentencia; las vistas dependientes que hay que soltar se las dan ya leídas.
 """
@@ -104,12 +111,13 @@ class DDLPlanner:
         statements.extend(self._drop_foreign_keys(report))
         statements.extend(self._drop_foreign_keys_for_type_change(report))
         statements.extend(self._column_changes(report))
-        statements.extend(self._add_foreign_keys(report))
-        statements.extend(self._alter_foreign_keys(report))
-        statements.extend(self._recreate_foreign_keys_for_type_change(report))
+        # Lo que una FK puede necesitar: la unicidad y la clave primaria, antes que ellas
         statements.extend(self._drop_unique_constraints(report))
         statements.extend(self._alter_primary_keys(report))
         statements.extend(self._add_unique_constraints(report))
+        statements.extend(self._add_foreign_keys(report))
+        statements.extend(self._alter_foreign_keys(report))
+        statements.extend(self._recreate_foreign_keys_for_type_change(report))
         statements.extend(self._drop_indexes(report))
         statements.extend(self._create_indexes(report))
         statements.extend(self._alter_views(report))
@@ -222,7 +230,7 @@ class DDLPlanner:
 
         return statements
 
-    # ── 7, 8 y 9. ADD y ALTER de claves foráneas ─────────────────────────
+    # ── 10, 11 y 12. ADD y ALTER de claves foráneas ──────────────────────
 
     def _add_foreign_keys(self, report: DriftReport) -> list[Statement]:
         """
@@ -282,7 +290,7 @@ class DDLPlanner:
             no_defer=no_defer,
         )
 
-    # ── 10, 11 y 12. Unicidad y clave primaria ───────────────────────────
+    # ── 7, 8 y 9. Unicidad y clave primaria ──────────────────────────────
 
     def _drop_unique_constraints(self, report: DriftReport) -> list[Statement]:
         """