scripts/bootstrap_qdrant_plans.py aktualisiert

scripts/bootstrap_qdrant_plans.py

@@ -1,21 +1,24 @@
 #!/usr/bin/env python3
 # -*- coding: utf-8 -*-
 """
-Bootstrap (idempotent) für Qdrant-Collections rund um Pläne – v1.1.0
+Bootstrap (idempotent) für Qdrant-Collections rund um Pläne – v1.2.1
 
-- Legt fehlende Payload-Indizes als KEYWORD an (wo sinnvoll)
+Änderungen ggü. v1.1.0:
-- Nutzt materialisierte Facettenfelder auf Templates (section_*),
+- NEU: KEYWORD-Index `plan_section_names` in der `plans`-Collection (robuste Section-Filter).
-  um verschachtelte Pfade stabil zu indizieren
+- Beibehalt: materialisierte Template-Facetten (section_*), Sessions-Indizes.
-- Collections selbst werden NICHT erstellt; das passiert beim ersten Upsert
+- Idempotent: existierende Indizes werden sauber übersprungen.
 
-Start:
+Hinweis:
-  pip install qdrant-client --upgrade
+- Das Script erstellt KEINE Collections; sie werden beim ersten Upsert durch die Router angelegt.
-  python3 scripts/bootstrap_qdrant_plans.py
+- Läuft gefahrlos mehrfach.
 """
 import os
 from qdrant_client import QdrantClient
 from qdrant_client.models import PayloadSchemaType
 
+# -------------------------
+# ENV / Defaults
+# -------------------------
 QDRANT_HOST = os.getenv("QDRANT_HOST", "localhost")
 QDRANT_PORT = int(os.getenv("QDRANT_PORT", "6333"))
 
@@ -23,6 +26,9 @@ PLANS = os.getenv("PLAN_COLLECTION") or os.getenv("QDRANT_COLLECTION_PLANS", "pl
 TEMPLATES = os.getenv("PLAN_TEMPLATE_COLLECTION", "plan_templates")
 SESSIONS  = os.getenv("PLAN_SESSION_COLLECTION", "plan_sessions")
 
+# -------------------------
+# Index-Spezifikation (KEYWORD)
+# -------------------------
 INDEX_SPECS = {
     # Templates: nutzen materialisierte Felder für stabile Indizes
     TEMPLATES: [
@@ -36,19 +42,20 @@ INDEX_SPECS = {
         ("section_forbid_keywords",     PayloadSchemaType.KEYWORD),
         ("goals",                       PayloadSchemaType.KEYWORD),
     ],
-    # Plans: belassen verschachtelte Felder wie bisher; falls deine Qdrant-Version das
+    # Plans: zusätzlich zum verschachtelten Pfad auch materialisierte Section-Namen
-    # nicht unterstützt, können wir analog materialisieren (section_names) und hier ergänzen.
     PLANS: [
         ("discipline",          PayloadSchemaType.KEYWORD),
         ("age_group",           PayloadSchemaType.KEYWORD),
         ("target_group",        PayloadSchemaType.KEYWORD),
-        ("sections.name", PayloadSchemaType.KEYWORD),
+        ("sections.name",       PayloadSchemaType.KEYWORD),   # Alt; belassen für Kompatibilität
+        ("plan_section_names",  PayloadSchemaType.KEYWORD),   # Neu & robust
         ("goals",               PayloadSchemaType.KEYWORD),
         ("created_by",          PayloadSchemaType.KEYWORD),
         ("created_at",          PayloadSchemaType.KEYWORD),
         ("fingerprint",         PayloadSchemaType.KEYWORD),
         ("title",               PayloadSchemaType.KEYWORD),
     ],
+    # Sessions
     SESSIONS: [
         ("plan_id",     PayloadSchemaType.KEYWORD),
         ("executed_at", PayloadSchemaType.KEYWORD),
@@ -57,32 +64,43 @@ INDEX_SPECS = {
     ],
 }
 
-
+# -------------------------
+# Helpers
+# -------------------------
 def _create_indexes(client: QdrantClient, collection: str, specs):
-    # Prüfe, ob Collection existiert (wir erwarten sie). Falls nicht, Hinweis, aber nicht fatal.
+    """Erzeugt fehlende Payload-Indizes als KEYWORD. Überspringt existierende."""
+    # Existenz der Collection prüfen (nicht fatal, falls fehlend)
     try:
         client.get_collection(collection)
-        print(f"[Bootstrap v1.1.0] Collection '{collection}' ok.")
+        print(f"[Bootstrap v1.2.1] Collection '{collection}' ok.")
     except Exception as e:
-        print(f"[Bootstrap v1.1.0] WARN: Collection '{collection}' nicht gefunden (wird beim ersten Upsert erstellt). Details: {e}")
+        print(f"[Bootstrap v1.2.1] WARN: Collection '{collection}' nicht gefunden (wird beim ersten Upsert erstellt). Details: {e}")
         return
 
     for field, schema in specs:
         try:
-            client.create_payload_index(collection_name=collection, field_name=field, field_schema=schema)
+            client.create_payload_index(
-            print(f"[Bootstrap v1.1.0] Index created: {collection}.{field} ({schema})")
+                collection_name=collection,
+                field_name=field,
+                field_schema=schema,
+            )
+            print(f"[Bootstrap v1.2.1] Index created: {collection}.{field} ({schema.name.lower()})")
         except Exception as e:
-            print(f"[Bootstrap v1.1.0] Index exists or skipped: {collection}.{field} -> {e}")
+            # Meist: Already exists -> überspringen
-
+            print(f"[Bootstrap v1.2.1] Index exists or skipped: {collection}.{field} -> {e}")
 
+# -------------------------
+# Main
+# -------------------------
 def main():
-    print(f"[Bootstrap v1.1.0] Qdrant @ {QDRANT_HOST}:{QDRANT_PORT}")
+    print(f"[Bootstrap v1.2.1] Qdrant @ {QDRANT_HOST}:{QDRANT_PORT}")
-    print(f"[Bootstrap v1.1.0] Collections: TEMPLATES={TEMPLATES} PLANS={PLANS} SESSIONS={SESSIONS}")
+    print(f"[Bootstrap v1.2.1] Collections: TEMPLATES={TEMPLATES} PLANS={PLANS} SESSIONS={SESSIONS}")
     client = QdrantClient(host=QDRANT_HOST, port=QDRANT_PORT)
+
     for coll, specs in INDEX_SPECS.items():
         _create_indexes(client, coll, specs)
-    print("[Bootstrap v1.1.0] done.")
 
+    print("[Bootstrap v1.2.1] done.")
 
 if __name__ == "__main__":
     main()