Lars
70821bae5a
All checks were successful
Deploy mindnet to llm-node / deploy (push) Successful in 3s
105 lines
2.7 KiB
Python
105 lines
2.7 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
"""
|
|
Kleine Diagnosehilfe für mindnet_edges.
|
|
|
|
Nutzung (aus Projektwurzel):
|
|
|
|
(.venv) python tests/diag_edges_for_note.py \
|
|
--note-id 20251130-relshow-embeddings-101
|
|
|
|
Optional:
|
|
|
|
--prefix mindnet
|
|
|
|
Ziel:
|
|
- Zeigt alle Edges aus der Collection <prefix>_edges,
|
|
die sich auf die angegebene note_id beziehen:
|
|
* note_id == NOTE
|
|
* oder source_id == NOTE
|
|
* oder target_id == NOTE
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import argparse
|
|
import os
|
|
import sys
|
|
from typing import List
|
|
|
|
# Projekt-Root (Ordner mit "app/") zum sys.path hinzufügen,
|
|
# damit "import app.core.qdrant" funktioniert, auch wenn das Script
|
|
# direkt aus tests/ gestartet wird.
|
|
THIS_DIR = os.path.dirname(os.path.abspath(__file__))
|
|
PROJECT_ROOT = os.path.dirname(THIS_DIR)
|
|
if PROJECT_ROOT not in sys.path:
|
|
sys.path.insert(0, PROJECT_ROOT)
|
|
|
|
from qdrant_client.http import models as rest # type: ignore
|
|
|
|
from app.core.qdrant import QdrantConfig, get_client, collection_names
|
|
|
|
|
|
def fetch_edges_for_note(note_id: str, prefix: str | None, limit: int = 512) -> List[dict]:
|
|
cfg = QdrantConfig.from_env()
|
|
if prefix:
|
|
cfg.prefix = prefix
|
|
|
|
client = get_client(cfg)
|
|
_, _, edges_col = collection_names(cfg.prefix)
|
|
|
|
conditions = []
|
|
for field in ("note_id", "source_id", "target_id"):
|
|
conditions.append(
|
|
rest.FieldCondition(
|
|
key=field,
|
|
match=rest.MatchValue(value=str(note_id)),
|
|
)
|
|
)
|
|
|
|
flt = rest.Filter(
|
|
should=conditions,
|
|
)
|
|
|
|
points, _ = client.scroll(
|
|
collection_name=edges_col,
|
|
scroll_filter=flt,
|
|
with_payload=True,
|
|
with_vectors=False,
|
|
limit=limit,
|
|
)
|
|
|
|
out: List[dict] = []
|
|
for p in points or []:
|
|
pl = dict(p.payload or {})
|
|
if pl:
|
|
out.append(pl)
|
|
return out
|
|
|
|
|
|
def main() -> None:
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("--note-id", required=True, help="Note-ID, z. B. 20251130-relshow-embeddings-101")
|
|
parser.add_argument("--prefix", default=None, help="Qdrant-Prefix (Default: aus ENV/QdrantConfig)")
|
|
args = parser.parse_args()
|
|
|
|
note_id = args.note_id
|
|
prefix = args.prefix
|
|
|
|
edges = fetch_edges_for_note(note_id, prefix)
|
|
print(f"Prefix: {prefix or '(aus ENV)'}")
|
|
print(f"Note-ID: {note_id}")
|
|
print(f"Gefundene Edges: {len(edges)}")
|
|
print("-" * 60)
|
|
|
|
for i, e in enumerate(edges, start=1):
|
|
print(f"[{i}] kind={e.get('kind')} confidence={e.get('confidence')}")
|
|
print(f" note_id = {e.get('note_id')}")
|
|
print(f" source_id= {e.get('source_id')}")
|
|
print(f" target_id= {e.get('target_id')}")
|
|
print()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|