scripts/import_markdown.py aktualisiert
Some checks failed
Deploy mindnet to llm-node / deploy (push) Failing after 1s

This commit is contained in:
Lars 2025-09-05 07:33:05 +02:00
parent 057c12c2d9
commit f7b1995b08

View File

@ -1,17 +1,41 @@
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Name: scripts/import_markdown.py
Version: v2.1.1 (2025-09-05)
Kurzbeschreibung:
Importiert Obsidian-Markdown-Notes in Qdrant (Notes/Chunks/Edges).
Leitet Wikilink-Edges (references/backlink/references_at) direkt beim Import
aus Volltext + echten Chunk-Texten ab. Keine Abhängigkeit zu edges.py (Legacy).
Aufruf (aus Projekt-Root, im venv):
python3 -m scripts.import_markdown --vault ./vault [--apply] [--note-id NOTE_ID] [--embed-note]
Parameter:
--vault Pfad zum Vault (z. B. ./vault)
--apply Führt Upserts in Qdrant aus (ohne Flag = Dry-Run mit JSON-Summaries)
--note-id Bearbeite nur eine konkrete Note-ID
--embed-note Optional: Note-Vektor (Volltext) zusätzlich einbetten
Umgebungsvariablen (optional):
QDRANT_URL, QDRANT_API_KEY, COLLECTION_PREFIX, VECTOR_DIM (Default 384)
Exitcodes:
0 = OK, 2 = keine Markdown-Dateien gefunden
Wichtige Hinweise:
- Verwendet ausschließlich app.core.derive_edges für Wikilink-Kanten.
- Chunk-Texte werden an derive_wikilink_edges übergeben; nur so entstehen references_at.
- Edges-IDs sind deterministisch (UUIDv5 in qdrant_points), idempotent.
- Edges-Collection bleibt 1D-Dummy-Vektor (DOT), reine Payload-Nutzung.
Changelog:
v2.1.1: Entfernt Abhängigkeit auf app.core.edges; direkte Nutzung von derive_wikilink_edges;
sichert Übergabe der Chunk-Texte references_at werden erzeugt.
v2.1.0: Note-Index über gesamten Vault; erste Fassung mit direkter Edge-Ableitung.
v2.0.x: Import Notes/Chunks/Embeddings, Edges via Backfill (separat).
"""
from __future__ import annotations
"""
WP-03 STEP 1
Ziel dieses Patches: Edges (references/backlink/references_at) entstehen direkt beim Import
auf Basis der **echten Chunks** und eines **Vorab-Note-Index**. Damit verschwinden
"dangling" references_at-Quellen aus dem Backfill-Ansatz.
Ausführung (aus Projekt-Root, im venv):
(.venv) python3 -m scripts.import_markdown --vault ./vault --apply
(.venv) python3 -m scripts.validate_edges --prefix mindnet --details
Abhängigkeiten: qdrant-client, python-dotenv
"""
import argparse
import os
import glob
@ -21,7 +45,7 @@ from typing import List, Dict
from dotenv import load_dotenv
# Core-Module
# Core-Module (gemäß Projektstruktur)
from app.core.parser import (
read_markdown,
normalize_frontmatter,
@ -39,14 +63,10 @@ from app.core.qdrant_points import (
points_for_edges,
upsert_batch,
)
# Neue direkte Edge-Ableitung beim Import
# Nur Wikilink-Ableitung (keine Kollision mit edges.py):
from app.core.derive_edges import build_note_index, derive_wikilink_edges
# -----------------------------------------------------------------------------
# Utilities
# -----------------------------------------------------------------------------
def iter_md(root: str, exclude=("/.obsidian/", "/_backup_frontmatter/", "/_imported/")) -> List[str]:
files = [p for p in glob.glob(os.path.join(root, "**", "*.md"), recursive=True)]
out: List[str] = []
@ -71,10 +91,6 @@ def minimal_note_index_payload(abs_path: str, vault_root: str) -> Dict:
}
# -----------------------------------------------------------------------------
# Main
# -----------------------------------------------------------------------------
def main():
load_dotenv()
@ -85,7 +101,7 @@ def main():
ap.add_argument("--embed-note", action="store_true", help="Auch Note-Volltext einbetten (optional)")
args = ap.parse_args()
# Qdrant
# Qdrant-Konfiguration (+ Collections sicherstellen)
cfg = QdrantConfig(
url=os.getenv("QDRANT_URL", "http://127.0.0.1:6333"),
api_key=os.getenv("QDRANT_API_KEY") or None,
@ -101,28 +117,23 @@ def main():
print("Keine Markdown-Dateien gefunden.", file=sys.stderr)
sys.exit(2)
# ------------------------------------------------------------------
# (1) Vorab-Lauf: Note-Index für robuste Zielauflösung
# ------------------------------------------------------------------
# (1) Vorab-Lauf: Note-Index (robuste Auflösung via ID/Titel-Slug/File-Slug)
index_payloads: List[Dict] = []
for path in files:
try:
pl = minimal_note_index_payload(path, root)
if not pl.get("note_id"):
# ohne stabile ID lassen wir die Note komplett aus
continue
if args.note_id and pl["note_id"] != args.note_id:
continue
index_payloads.append(pl)
except Exception:
# bewusst still: Einzeldefekte sollen den Gesamtimport nicht stoppen
# Einzeldefekte nicht fatal
continue
note_index = build_note_index(index_payloads) # (by_id, by_slug, by_file_slug)
note_index = build_note_index(index_payloads)
# ------------------------------------------------------------------
# (2) Hauptlauf: pro Note Chunks/Embeddings + Edges aus echten Chunks
# ------------------------------------------------------------------
# (2) Hauptlauf: Import + direkte Wikilink-Edge-Ableitung
total_notes = 0
for path in files:
parsed = read_markdown(path)
@ -154,31 +165,41 @@ def main():
note_vec = embed_one(parsed.body)
# Edges direkt aus realen Chunks + Note-Index
# Für derive_wikilink_edges darf der Note-Payload einen Volltext enthalten
note_pl_for_edges = dict(note_pl)
note_pl_for_edges.setdefault("fulltext", parsed.body)
edges = derive_wikilink_edges(note_pl_for_edges, chunk_pls, note_index)
# ACHTUNG: derive_wikilink_edges erwartet Chunks mit Text → wir bauen eine
# minimalistische Liste {"chunk_id","text"} in gleicher Reihenfolge.
note_pl_for_edges = {
"note_id": fm["id"],
"title": fm.get("title"),
"path": note_pl["path"],
"fulltext": parsed.body,
}
chunks_for_links = []
for i, pl in enumerate(chunk_pls):
cid = pl.get("chunk_id") or pl.get("id")
txt = chunks[i].text if i < len(chunks) else ""
chunks_for_links.append({"chunk_id": cid, "text": txt})
# Dry-Run-Log
summary = {
edges = derive_wikilink_edges(note_pl_for_edges, chunks_for_links, note_index)
# Dry-Run-Summary
print(json.dumps({
"note_id": fm["id"],
"title": fm["title"],
"chunks": len(chunk_pls),
"edges": len(edges),
"path": note_pl["path"],
}
print(json.dumps(summary, ensure_ascii=False))
}, ensure_ascii=False))
if args.apply:
# Notes upsert (idempotent; UUIDv5)
notes_col, note_pts = points_for_note(cfg.prefix, note_pl, note_vec, cfg.dim)
upsert_batch(client, notes_col, note_pts)
# Chunks upsert (idempotent; chunk_id stabilisiert notfalls qdrant_points)
# Chunks upsert (idempotent)
chunks_col, chunk_pts = points_for_chunks(cfg.prefix, chunk_pls, vectors)
upsert_batch(client, chunks_col, chunk_pts)
# Edges upsert (idempotent; edge_id deterministisch aus kind/src/tgt/seq)
# Edges upsert (idempotent; deterministisches edge_id)
edges_col, edge_pts = points_for_edges(cfg.prefix, edges)
upsert_batch(client, edges_col, edge_pts)