tai-sql 0.8.0__tar.gz → 0.8.1__tar.gz

{tai_sql-0.8.0 → tai_sql-0.8.1}/PKG-INFO

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

{tai_sql-0.8.0 → tai_sql-0.8.1}/pyproject.toml

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

{tai_sql-0.8.0 → tai_sql-0.8.1}/tai_sql/generators/python_client/templates/domain/dtos.py.jinja2

@@ -221,10 +221,28 @@ class {{ model.name }}Read(PrettyModel):
 
     {% if not model.is_view %}
     @classmethod
+{% if _all_vector_cols %}
+    def from_created_instance(
+        cls,
+        instance: {{ model.name }},
+        included: set[str],
+        excluded: str=None,
+        include_vectors: bool = False,
+    ) -> {{ model.name }}Read:
+{% else %}
     def from_created_instance(cls, instance: {{ model.name }}, included: set[str], excluded: str=None) -> {{ model.name }}Read:
+{% endif %}
+        {% if _all_vector_cols %}
+        # Como `from_instance`: los vectores solo si se piden. Son miles de números por fila, y
+        # lo que se acaba de crear se devuelve entero (`create_many` de mil filas eran millones).
+        # Las filas anidadas los dejan fuera siempre, como las relaciones de `from_instance`.
+        {% endif %}
         dto_data = {
             {% for column in model.columns %}
-            {% if not column.hash %}
+            {% if column.hash %}
+            {% elif column.type == 'vector' %}
+            '{{ column.name }}': instance.{{ column.name }} if include_vectors else None,
+            {% else %}
             '{{ column.name }}': instance.{{ column.name }},
             {% endif %}
             {% endfor %}

{tai_sql-0.8.0 → tai_sql-0.8.1}/tai_sql/generators/python_client/templates/macros/write.jinja2

@@ -139,11 +139,15 @@
         return len({{ await_keyword -}} cls.create_many([{{ model.name }}Create.from_dict(record) for record in records_data]))
         {% endif %}
 
+{% set _created_vector_cols = model.columns | selectattr('type', 'equalto', 'vector') | list %}
     @classmethod
     @{{ error_handler_name }}
     {{ def_keyword }} create(
         cls, 
         {{ model.tablename }}: {{ model.name }}Create,
+{% if _created_vector_cols %}
+        include_vectors: bool = False,
+{% endif %}
         session: Optional[{{ session_type }}] = None
     ) -> {{ model.name }}Read:
         """
@@ -151,6 +155,10 @@
         
         Args:
             {{ model.tablename }}: Datos del {{ model.tablename }} a crear
+{% if _created_vector_cols %}
+            include_vectors: Si True, el registro devuelto lleva sus columnas vectoriales. Por
+                defecto no, como en `find`: se guardan igual, solo no se devuelven.
+{% endif %}
             session: Sesión existente (opcional)
             
         Returns:
@@ -173,7 +181,11 @@
 {% endif %}
             {{ await_keyword -}} session.flush()
             {{ await_keyword -}} cls._on_create_after(new_instance, included, session)
+{% if _created_vector_cols %}
+            return {{ model.name }}Read.from_created_instance(new_instance, included, include_vectors=include_vectors)
+{% else %}
             return {{ model.name }}Read.from_created_instance(new_instance, included)
+{% endif %}
 
         if session is not None:
             data = {{ await_keyword -}} _execute_create(session)
@@ -199,6 +211,9 @@
     {{ def_keyword }} create_many(
         cls, 
         records: List[{{ model.name }}Create], 
+{% if _created_vector_cols %}
+        include_vectors: bool = False,
+{% endif %}
         session: Optional[{{ session_type }}] = None
     ) -> List[{{ model.name }}Read]:
         """
@@ -206,6 +221,10 @@
         
         Args:
             records: Lista de {{ model.name }}Create con los datos de los registros
+{% if _created_vector_cols %}
+            include_vectors: Si True, los registros devueltos llevan sus columnas vectoriales. Por
+                defecto no, como en `find_many`: con mil filas serían millones de números.
+{% endif %}
             session: Sesión existente (opcional)
             
         Returns:
@@ -240,7 +259,11 @@
             # Los DTO se construyen después de los triggers `after`, como en `create`: lo que
             # escriban en la fila tiene que salir en lo que se devuelve
             return [
+{% if _created_vector_cols %}
+                {{ model.name }}Read.from_created_instance(instance, included, include_vectors=include_vectors)
+{% else %}
                 {{ model.name }}Read.from_created_instance(instance, included)
+{% endif %}
                 for instance, included in creadas
             ]