19da099583
4 feste Farbstufen (primary/secondary/tertiary/mentioned) mit variablen Labels pro Lage, die von Haiku generiert werden. - DB: category_labels Spalte in incidents, alte Kategorien migriert (target->primary, response/retaliation->secondary, actor->tertiary) - Geoparsing: generate_category_labels() + neuer Prompt mit neuen Keys - QC: Kategorieprüfung auf neue Keys umgestellt - Orchestrator: Tuple-Rückgabe + Labels in DB speichern - API: category_labels im Locations- und Lagebild-Response - Frontend: Dynamische Legende aus API-Labels mit Fallback-Defaults - Migrationsskript für bestehende Lagen Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
74 Zeilen
2.4 KiB
Python
74 Zeilen
2.4 KiB
Python
#!/usr/bin/env python3
|
|
"""Einmaliges Migrationsskript: Generiert Haiku-Labels fuer alle bestehenden Lagen.
|
|
|
|
Ausfuehrung auf dem Monitor-Server:
|
|
cd /home/claude-dev/AegisSight-Monitor
|
|
.venvs_run: /home/claude-dev/.venvs/osint/bin/python migrate_category_labels.py
|
|
"""
|
|
import asyncio
|
|
import json
|
|
import logging
|
|
import os
|
|
import sys
|
|
|
|
# Projektpfad setzen damit imports funktionieren
|
|
sys.path.insert(0, os.path.join(os.path.dirname(__file__), 'src'))
|
|
|
|
logging.basicConfig(
|
|
level=logging.INFO,
|
|
format='%(asctime)s [%(name)s] %(levelname)s: %(message)s',
|
|
)
|
|
logger = logging.getLogger("migrate_labels")
|
|
|
|
|
|
async def main():
|
|
from database import get_db
|
|
from agents.geoparsing import generate_category_labels
|
|
|
|
db = await get_db()
|
|
try:
|
|
# Alle Incidents ohne category_labels laden
|
|
cursor = await db.execute(
|
|
"SELECT id, title, description FROM incidents WHERE category_labels IS NULL"
|
|
)
|
|
incidents = [dict(row) for row in await cursor.fetchall()]
|
|
|
|
if not incidents:
|
|
logger.info("Keine Incidents ohne Labels gefunden. Nichts zu tun.")
|
|
return
|
|
|
|
logger.info(f"{len(incidents)} Incidents ohne Labels gefunden. Starte Generierung...")
|
|
|
|
success = 0
|
|
for inc in incidents:
|
|
incident_id = inc["id"]
|
|
context = f"{inc['title']} - {inc.get('description') or ''}"
|
|
logger.info(f"Generiere Labels fuer Incident {incident_id}: {inc['title'][:60]}...")
|
|
|
|
try:
|
|
labels = await generate_category_labels(context)
|
|
if labels:
|
|
await db.execute(
|
|
"UPDATE incidents SET category_labels = ? WHERE id = ?",
|
|
(json.dumps(labels, ensure_ascii=False), incident_id),
|
|
)
|
|
await db.commit()
|
|
success += 1
|
|
logger.info(f" -> Labels: {labels}")
|
|
else:
|
|
logger.warning(f" -> Keine Labels generiert")
|
|
except Exception as e:
|
|
logger.error(f" -> Fehler: {e}")
|
|
|
|
# Kurze Pause um Rate-Limits zu vermeiden
|
|
await asyncio.sleep(0.5)
|
|
|
|
logger.info(f"\nMigration abgeschlossen: {success}/{len(incidents)} Incidents mit Labels versehen.")
|
|
|
|
finally:
|
|
await db.close()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(main())
|