nucliadb-agentic-api 1.0.0.post279__tar.gz → 1.0.0.post286__tar.gz

{nucliadb_agentic_api-1.0.0.post279 → nucliadb_agentic_api-1.0.0.post286}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nucliadb_agentic_api
-Version: 1.0.0.post279
+Version: 1.0.0.post286
 Summary: NucliaDB Agentic API for Hyperforge
 Author: AI Data Team
 Author-email: AI Data Team <learning@nuclia.com>

nucliadb_agentic_api-1.0.0.post286/VERSION

@@ -0,0 +1 @@
+1.0.0.post286

nucliadb_agentic_api-1.0.0.post286/alembic/versions/0fe6c7b5f33d_migrate_sonnet_and_gpt_5_6_models.py

@@ -0,0 +1,35 @@
+"""migrate sonnet and gpt 5 6 models
+
+Revision ID: 0fe6c7b5f33d
+Revises: c4d5e6f7a8b9
+Create Date: 2026-09-21 10:28:51.980835
+
+"""
+
+from typing import Sequence, Union
+
+from alembic import op
+from nucliadb_agentic_api.db.llm_migration_utils import migrate_llm_models
+
+# revision identifiers, used by Alembic.
+revision: str = "0fe6c7b5f33d"
+down_revision: Union[str, Sequence[str], None] = "c4d5e6f7a8b9"
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    migrate_llm_models(
+        op.get_bind(),
+        {
+            "claude-4-5-sonnet": "claude-5-sonnet",
+            "chatgpt-azure-4o": "chatgpt-azure-5.6-terra",
+            "chatgpt-azure-4o-mini": "chatgpt-azure-5.6-luna",
+            "chatgpt-azure-o3-mini": "chatgpt-azure-5.6-luna",
+        },
+    )
+
+
+def downgrade() -> None:
+    """Downgrade schema."""
+    pass

{nucliadb_agentic_api-1.0.0.post279 → nucliadb_agentic_api-1.0.0.post286}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "uv_build"
 
 [project]
 name = "nucliadb_agentic_api"
-version = "1.0.0.post279"
+version = "1.0.0.post286"
 description = "NucliaDB Agentic API for Hyperforge"
 readme = "README.md"
 requires-python = ">=3.12"

{nucliadb_agentic_api-1.0.0.post279 → nucliadb_agentic_api-1.0.0.post286}/pyproject.toml.orig

@@ -4,7 +4,7 @@ build-backend = "uv_build"
 
 [project]
 name = "nucliadb_agentic_api"
-version = "1.0.0.post279"
+version = "1.0.0.post286"
 authors = [{ name = "AI Data Team", email = "learning@nuclia.com" }]
 description = "NucliaDB Agentic API for Hyperforge"
 readme = "README.md"

nucliadb_agentic_api-1.0.0.post286/src/nucliadb_agentic_api/db/llm_migration_utils.py

@@ -0,0 +1,64 @@
+import json
+from typing import Any
+
+from sqlalchemy import text
+from sqlalchemy.engine import Connection
+
+AGENTIC_LLM_TABLES = [
+    ("agentic_config_table", "config", ("account", "kbid", "agentic_id")),
+    ("sources_table", "config", ("account", "kbid", "source_id")),
+]
+
+
+def walk_and_replace(obj: Any, replacements: dict[str, str]) -> bool:
+    """Replace exact model IDs in recursively nested LLMConfig objects."""
+    modified = False
+    if isinstance(obj, dict):
+        model_id = obj.get("model_id")
+        if (
+            obj.get("_type") == "llm_config"
+            and isinstance(model_id, str)
+            and "/" not in model_id
+            and model_id in replacements
+        ):
+            obj["model_id"] = replacements[model_id]
+            modified = True
+
+        for value in obj.values():
+            modified = walk_and_replace(value, replacements) or modified
+    elif isinstance(obj, list):
+        for value in obj:
+            modified = walk_and_replace(value, replacements) or modified
+    return modified
+
+
+def migrate_llm_models(conn: Connection, replacements: dict[str, str]) -> int:
+    """Migrate LLMConfig model IDs in agentic and source JSONB configs."""
+    modified_rows = 0
+    for table, column, key_columns in AGENTIC_LLM_TABLES:
+        selected_columns = ", ".join((*key_columns, column))
+        rows = conn.execute(
+            text(
+                f"SELECT {selected_columns} FROM {table} "
+                f'WHERE {column}::text LIKE \'%"_type": "llm_config"%\''
+            )
+        ).fetchall()
+        for row in rows:
+            config = row[-1]
+            if isinstance(config, str):
+                config = json.loads(config)
+            if config is None or not walk_and_replace(config, replacements):
+                continue
+
+            where = " AND ".join(f"{key} = :{key}" for key in key_columns)
+            params = dict(zip(key_columns, row[:-1], strict=True))
+            params["config"] = json.dumps(config)
+            conn.execute(
+                text(
+                    f"UPDATE {table} SET {column} = CAST(:config AS jsonb) "
+                    f"WHERE {where}"
+                ),
+                params,
+            )
+            modified_rows += 1
+    return modified_rows

nucliadb_agentic_api-1.0.0.post279/VERSION

@@ -1 +0,0 @@
-1.0.0.post279