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

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

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

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

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

{tai_sql-0.7.12 → 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')

{tai_sql-0.7.12 → tai_sql-0.7.14}/tai_sql/generators/python_client/context.py

@@ -225,6 +225,12 @@ class TemplateContextBuilder:
             'is_operative': column.is_operative,
             'is_calculated': column.is_calculated,
             'self_reference': column.self_reference,
+            # La rellena la base de datos (NOW()): no es obligatoria al crear. Faltaba, y las
+            # plantillas que la miraban la leían siempre como falsa.
+            'server_now': column.server_now,
+            # El tamaño máximo de un texto (`column(length=n)`), para quien valide antes de
+            # escribir —los esquemas de formulario de tai-api—. `None` si no se declaró.
+            'max_length': column.length if column.type == 'str' else None,
         }
 
         if column.is_vector:

{tai_sql-0.7.12 → tai_sql-0.7.14}/tai_sql/generators/python_client/templates/domain/dtos.py.jinja2

@@ -247,7 +247,7 @@ class {{ model.name }}Read(PrettyModel):
 class {{ model.name }}Create(PrettyModel):
     """DTO de escritura para {{ model.name }}."""
     {% for column in model.columns -%}
-    {% if not column.constraints.get('autoincrement', False) and column.default is none and not column.nullable and not column.is_foreign_key and not column.is_operative and not column.has_encoder|default(false) and not column.is_calculated %}
+    {% if not column.constraints.get('autoincrement', False) and column.default is none and not column.nullable and not column.is_foreign_key and not column.is_operative and not column.has_encoder|default(false) and not column.is_calculated and not column.server_now %}
     {{ column.name }}: {{ column.pydantic_type }}
     {% endif %}
     {% endfor %}
@@ -257,6 +257,9 @@ class {{ model.name }}Create(PrettyModel):
     {{ column.name }}: Optional[{{ column.pydantic_type }}] = None
     {% elif column.is_foreign_key %}
     {{ column.name }}: Optional[{{ column.pydantic_type }}] = None
+    {% elif column.server_now %}
+    {# La rellena la base de datos con NOW() si no llega: no es obligatoria. #}
+    {{ column.name }}: Optional[{{ column.pydantic_type }}] = None
     {% elif column.default is none and column.nullable %}
     {{ column.name }}: Optional[{{ column.pydantic_type }}] = None
     {% elif column.default is not none %}
@@ -295,11 +298,18 @@ class {{ model.name }}Create(PrettyModel):
     def to_instance(self) -> {{ model.name }}:
         model = {{ model.name }}(
             {% for column in model.columns %}
-            {% if not column.constraints.get('autoincrement', False) and not column.is_operative and not column.is_calculated %}
+            {% if not column.constraints.get('autoincrement', False) and not column.is_operative and not column.is_calculated and not column.server_now %}
             {{ column.name }}=self.{{ column.name }},
             {% endif %}
             {% endfor %}
         )
+        {% for column in model.columns %}
+        {% if column.server_now and not column.is_operative and not column.is_calculated %}
+        # Solo si llega: asignarle None mandaría un NULL, y no dejaría a la base de datos poner NOW().
+        if self.{{ column.name }} is not None:
+            model.{{ column.name }} = self.{{ column.name }}
+        {% endif %}
+        {% endfor %}
         {% for col in operative_str_cols %}
         model.{{ col.name }} = get_username()
         {% endfor %}

{tai_sql-0.7.12 → tai_sql-0.7.14}/tai_sql/generators/python_client/templates/domain/model.py.jinja2

@@ -83,6 +83,9 @@ from pgvector.sqlalchemy import Vector
     {% if column.sa_type and column.sa_type not in sa_imports %}
         {% set _ = sa_imports.append(column.sa_type) %}
     {% endif %}
+    {% if column.server_now and 'func' not in sa_imports %}
+        {% set _ = sa_imports.append('func') %}
+    {% endif %}
 {% endfor %}
 {% if sa_imports %}
 from sqlalchemy import {{ sa_imports | sort | join(', ') }}
@@ -165,7 +168,7 @@ class {{ model.name }}(Base):
     {% set mapped_type = 'Optional[' + py_type + ']' if column.nullable else py_type %}
     {# Determine if mapped_column() call is needed #}
     {% set has_constraints = column.constraints.values() | select | list | length > 0 %}
-    {% set needs_mapped_column = has_constraints or column.encrypt or column.hash or column.is_calculated or column.sa_type is not none or column.type == 'vector' %}
+    {% set needs_mapped_column = has_constraints or column.encrypt or column.hash or column.is_calculated or column.sa_type is not none or column.type == 'vector' or column.server_now %}
     {% if needs_mapped_column %}
     {% set mc_args = [] %}
     {% if column.type == 'vector' %}{% set _ = mc_args.append('Vector(' + column.dimensions|string + ')') %}
@@ -185,6 +188,10 @@ class {{ model.name }}(Base):
     {% endfor %}
     {% endif %}
     {% if column.encrypt or column.hash %}{% set _ = mc_args.append('name="' + column.name + '"') %}{% endif %}
+    {# La pone la base de datos. Sin declararlo aquí, SQLAlchemy la enviaba como NULL —rellena
+       con None lo que no tiene valor para poder agrupar el INSERT— y la base de datos no llegaba
+       a poner NOW(). #}
+    {% if column.server_now %}{% set _ = mc_args.append('server_default=func.now()') %}{% endif %}
     {% if column.is_calculated %}{% set _ = mc_args.append('name="' + column.name + '"') %}{% endif %}
     {{ col_attr }}: Mapped[{{ mapped_type }}] = mapped_column({{ mc_args | join(', ') }})
     {% else %}

{tai_sql-0.7.12 → tai_sql-0.7.14}/tai_sql/rules/static/modelo.md

@@ -138,6 +138,9 @@ column(
 
 Una columna sin `column()` es una columna normal: `nombre: col[str]` basta.
 
+`server_now=True` la rellena la base de datos con `NOW()`: en el `Create` es opcional, y si no se
+envía, el cliente no la manda y el motor pone la fecha.
+
 ### Tipos disponibles
 
 `int`, `bigint`, `smallint`, `float`, `numeric`, `str`, `text`, `bool`, `datetime`,