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-08 18:36:59 +02:00
parent aa7d0190c8
commit 0a116ea3f4

View File

@ -1,319 +1,294 @@
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Script: scripts/import_markdown.py
Version: 0.7.1 (2025-09-06)
Autor: mindnet / Architektur Datenimporte & Sync
import_markdown.py Markdown Qdrant (Notes, Chunks, Edges) mit Change-Detection & optionalem Purge
Kurzbeschreibung
---------------
Importiert Markdown-Notizen aus einem Obsidian-ähnlichen Vault in Qdrant:
- Validiert Frontmatter / Note-Payload (gegen note.schema.json).
- Chunking + Embeddings.
- Leitet Edges direkt beim Import aus [[Wikilinks]] ab:
- 'references' (NoteNote)
- 'references_at' (ChunkNote)
- 'backlink' (NoteNote) (symmetrisch zu 'references')
- Idempotente Upserts (deterministische IDs über qdrant_points).
Version: 3.0 (2025-09-08)
Neu in 0.7.1
------------
- Korrekte Änderungsdetektion via SHA-256 über die **komplette Datei** (Frontmatter+Body):
- Feld: payload.hash_fulltext
- Vergleicht neuen Hash gegen bestehenden Hash in Qdrant.
- Nur bei Änderung Verarbeitung/Upsert; sonst "skip".
- `--purge-before-upsert` wird **nur** ausgeführt, wenn sich die Note **wirklich geändert** hat.
- Robuste Qdrant-Scroll-Kompatibilität (2- oder 3-Tupel Rückgaben).
Änderungen ggü. bisherigen Varianten:
- Stabiler Qdrant-Delete über Filter-API (kompatibel mit aktuellen Clients).
- Option --purge-before-upsert: löscht NUR Chunks/Edges der geänderten Note(n) vor Upsert.
- Change-Detection ausschließlich über Note-Body-Hash (Frontmatter/Dateidatum ignoriert).
- Robuste Edge-Erzeugung:
- Chunk-Level: belongs_to, prev/next, references (+backlink)
- Fallback: Note-Level references, falls Chunk-Payloads keine references liefern
- Korrekte Qdrant-Scroll/Query-Nutzung (Rückgabesignaturen neuer Clients).
- Saubere Dry-Run/Apply-Ausgaben je Note als JSON.
Aufrufbeispiele
---------------
Dry-Run (nur prüfen, nichts schreiben):
python3 -m scripts.import_markdown --vault ./vault
Aufruf:
python3 -m scripts.import_markdown --vault ./vault
python3 -m scripts.import_markdown --vault ./vault --apply
python3 -m scripts.import_markdown --vault ./vault --apply --purge-before-upsert
python3 -m scripts.import_markdown --vault ./vault --note-id 20250821-architektur-ki-trainerassistent-761cfe --apply
Nur eine spezifische Note:
python3 -m scripts.import_markdown --vault ./vault --note-id 20250821-foo
Parameter:
--vault PATH Pflicht. Root des (Obsidian-)Vaults.
--apply Wirkt schreibend in Qdrant; ohne Flag nur Dry-Run.
--purge-before-upsert Vor dem Upsert (nur) für geänderte Notes: zugehörige Chunks & Edges löschen.
--note-id ID Optional: Nur eine Note mit genau dieser ID verarbeiten.
--embed-note Optional: zusätzlich Volltext-Vektor für die Note erzeugen.
--force-replace Ignoriert Change-Detection (erzwingt Upsert + optionalen Purge).
Apply (schreiben) mit Purge (nur geänderte Noten werden bereinigt + neu geschrieben):
python3 -m scripts.import_markdown --vault ./vault --apply --purge-before-upsert
Umgebungsvariablen (optional):
QDRANT_URL z.B. http://127.0.0.1:6333 (falls QDRANT_HOST/PORT nicht gesetzt)
QDRANT_API_KEY API Key (falls erforderlich)
QDRANT_HOST/QDRANT_PORT Alternative zu QDRANT_URL
COLLECTION_PREFIX Default "mindnet"
VECTOR_DIM Default 384
Parameter
---------
--vault PATH : Pflicht. Root-Verzeichnis des Vaults.
--apply : Wenn gesetzt, werden Upserts durchgeführt (sonst Dry-Run).
--purge-before-upsert : Vor Upsert alte Chunks/Edges der **geänderten** Note löschen.
--note-id ID : Optional, verarbeitet nur diese eine Note.
Umgebungsvariablen (.env)
-------------------------
QDRANT_URL, QDRANT_API_KEY, COLLECTION_PREFIX, VECTOR_DIM
Defaults: url=http://127.0.0.1:6333, prefix=mindnet, dim=384
Kompatibilität
--------------
- Bestehende Kernmodule werden weiterverwendet:
app.core.parser (read_markdown, normalize_frontmatter, validate_required_frontmatter)
app.core.validate_note (validate_note_payload)
app.core.chunker (assemble_chunks)
app.core.chunk_payload (make_chunk_payloads)
app.core.embed (embed_texts)
app.core.qdrant (QdrantConfig, ensure_collections, get_client, collection_names)
app.core.qdrant_points (points_for_note, points_for_chunks, points_for_edges, upsert_batch)
app.core.derive_edges (build_note_index, derive_wikilink_edges)
Hinweise
--------
- Bitte im aktivierten venv laufen lassen: source .venv/bin/activate
Voraussetzungen:
- app/core:
parser.read_markdown, parser.normalize_frontmatter, parser.validate_required_frontmatter
note_payload.make_note_payload (liefert u.a. hash_fulltext)
validate_note.validate_note_payload
chunker.assemble_chunks
chunk_payload.make_chunk_payloads (idealerweise inkl. 'references' je Chunk; wenn nicht Fallback)
embed.embed_texts, embed.embed_one
qdrant.QdrantConfig, qdrant.get_client, qdrant.ensure_collections
qdrant_points.points_for_note / points_for_chunks / points_for_edges / upsert_batch
edges.deriv_edges_for_note (nutzt Chunk-Payloads; wir ergänzen Fallback)
"""
from __future__ import annotations
import argparse
import glob
import hashlib
import json
import os
import sys
import json
import glob
from typing import List, Dict, Tuple, Optional
from dotenv import load_dotenv
from qdrant_client.http import models as rest
from qdrant_client import QdrantClient
# Kern-Bausteine (aus eurem Projekt)
from app.core.parser import (
read_markdown,
normalize_frontmatter,
validate_required_frontmatter,
)
from app.core.parser import read_markdown, normalize_frontmatter, validate_required_frontmatter
from app.core.note_payload import make_note_payload
from app.core.validate_note import validate_note_payload
from app.core.chunker import assemble_chunks
from app.core.chunk_payload import make_chunk_payloads
from app.core.embed import embed_texts
from app.core.qdrant import QdrantConfig, ensure_collections, get_client, collection_names
from app.core.embed import embed_texts, embed_one
from app.core.qdrant import QdrantConfig, get_client, ensure_collections
from app.core.qdrant_points import (
points_for_note,
points_for_chunks,
points_for_note,
points_for_edges,
upsert_batch,
)
from app.core.derive_edges import build_note_index, derive_wikilink_edges
from app.core.edges import deriv_edges_for_note
# -------------------------------------------------
# Hilfsfunktionen
# -------------------------------------------------
# -----------------------------
# Utils
# -----------------------------
def iter_md(root: str) -> List[str]:
patterns = ["**/*.md", "*.md"]
out: List[str] = []
for p in patterns:
out.extend(glob.glob(os.path.join(root, p), recursive=True))
# de-dupe + sort
return sorted(list(dict.fromkeys(out)))
files = [p for p in glob.glob(os.path.join(root, "**", "*.md"), recursive=True)]
out = []
for p in files:
pn = p.replace("\\", "/")
if any(ex in pn for ex in ["/.obsidian/", "/_backup_frontmatter/", "/_imported/"]):
continue
out.append(p)
return sorted(out)
def file_sha256(path: str) -> str:
"""SHA256 über die **Rohdatei** (Frontmatter + Body)."""
h = hashlib.sha256()
with open(path, "rb") as f:
for chunk in iter(lambda: f.read(8192), b""):
h.update(chunk)
return h.hexdigest()
def collections(prefix: str) -> Tuple[str, str, str]:
return f"{prefix}_notes", f"{prefix}_chunks", f"{prefix}_edges"
def qdrant_scroll(client: QdrantClient, **kwargs) -> Tuple[List, Optional[str]]:
"""
Wrapper, der sowohl (points, next_offset) als auch (points, next_page, _) Signaturen abdeckt.
"""
res = client.scroll(**kwargs)
if isinstance(res, tuple):
if len(res) == 2:
return res[0], res[1]
if len(res) >= 3:
return res[0], res[1]
return res, None
def fetch_existing_note_hash(client: QdrantClient, prefix: str, note_id: str) -> Optional[str]:
"""Liest hash_fulltext aus mindnet_notes.payload für eine Note."""
notes_col, _, _ = collection_names(prefix)
f = rest.Filter(must=[rest.FieldCondition(key="note_id", match=rest.MatchValue(value=note_id))])
pts, _ = qdrant_scroll(
client,
def fetch_existing_note_hash(client, prefix: str, note_id: str) -> Optional[str]:
notes_col, _, _ = collections(prefix)
f = rest.Filter(
must=[rest.FieldCondition(key="note_id", match=rest.MatchValue(value=note_id))]
)
# neuere Clients: scroll gibt (points, next_page_offset) zurück
points, _ = client.scroll(
collection_name=notes_col,
scroll_filter=f,
with_payload=True,
with_vectors=False,
limit=1
limit=1,
)
if not pts:
if not points:
return None
pl = getattr(pts[0], "payload", {}) or {}
pl = points[0].payload or {}
return pl.get("hash_fulltext")
def make_note_stub(abs_path: str, vault_root: str) -> Dict:
"""Minimaler Stub (id/title/path) für build_note_index."""
parsed = read_markdown(abs_path)
fm = normalize_frontmatter(parsed.frontmatter or {})
if "id" not in fm or not fm["id"]:
raise ValueError(f"Missing id in frontmatter: {abs_path}")
rel = os.path.relpath(abs_path, vault_root)
return {"note_id": fm["id"], "title": fm.get("title"), "path": rel}
def build_vault_index(vault_root: str) -> tuple[Dict, Dict, Dict]:
"""Index für robuste Wikilink-Auflösung über alle Noten im Vault."""
stubs = []
for p in iter_md(vault_root):
try:
stubs.append(make_note_stub(p, vault_root))
except Exception:
continue
return build_note_index(stubs)
def purge_for_note(client: QdrantClient, prefix: str, note_id: str, chunk_ids: List[str]) -> None:
def purge_note_artifacts(client, prefix: str, note_id: str) -> None:
"""
Selektives Purge der alten Daten für **diese** Note:
- löscht Chunks (payload.note_id == note_id)
- löscht Edges, deren source_id == note_id ODER in chunk_ids
Notes selbst werden nicht gelöscht (Upsert reicht).
Löscht NUR Chunks & Edges der angegebenen Note (vor Upsert).
- Chunks: payload.note_id == note_id
- Edges : payload.source_id startswith note_id#c ODER payload.target_id startswith note_id#c
plus note-scope edges, falls vorhanden (source_id == note_id oder target_id == note_id)
"""
notes_col, chunks_col, edges_col = collection_names(prefix)
_, chunks_col, edges_col = collections(prefix)
# Chunks löschen
f_chunks = rest.Filter(must=[rest.FieldCondition(key="note_id", match=rest.MatchValue(value=note_id))])
# Chunks der Note löschen
f_chunks = rest.Filter(
must=[rest.FieldCondition(key="note_id", match=rest.MatchValue(value=note_id))]
)
client.delete(collection_name=chunks_col, points_selector=f_chunks, wait=True)
# Edges löschen (OR über source_id Werte)
should_conds = [rest.FieldCondition(key="source_id", match=rest.MatchValue(value=note_id))]
for cid in chunk_ids:
should_conds.append(rest.FieldCondition(key="source_id", match=rest.MatchValue(value=cid)))
if should_conds:
f_edges = rest.Filter(should=should_conds)
client.delete(collection_name=edges_col, points_selector=f_edges, wait=True)
# Edges der Note löschen (Chunk- und Note-Scope)
should = [
rest.FieldCondition(key="source_id", match=rest.MatchText(text=f"{note_id}#")),
rest.FieldCondition(key="target_id", match=rest.MatchText(text=f"{note_id}#")),
rest.FieldCondition(key="source_id", match=rest.MatchValue(value=note_id)),
rest.FieldCondition(key="target_id", match=rest.MatchValue(value=note_id)),
]
f_edges = rest.Filter(should=should)
client.delete(collection_name=edges_col, points_selector=f_edges, wait=True)
# -------------------------------------------------
def has_chunk_level_refs(chunk_payloads: List[Dict]) -> bool:
return any(isinstance(ch.get("references"), list) and ch["references"] for ch in chunk_payloads)
def fallback_note_level_edges(note_meta: Dict, chunk_payloads: List[Dict]) -> List[Dict]:
"""
Falls chunk_payloads keine 'references' enthalten, erzeugen wir
- belongs_to + prev/next aus chunk_payloads
- references/backlink auf Note-Scope aus note_meta['references'] (falls vorhanden)
"""
edges: List[Dict] = []
# belongs_to + prev/next (Chunk-Scope)
for ch in chunk_payloads:
src = ch["id"]
edges.append({"src_id": src, "dst_id": note_meta["id"], "edge_type": "belongs_to", "scope": "chunk"})
nb = ch.get("neighbors") or {}
prev_id = nb.get("prev")
next_id = nb.get("next")
if prev_id:
edges.append({"src_id": src, "dst_id": prev_id, "edge_type": "next", "scope": "chunk"})
edges.append({"src_id": prev_id, "dst_id": src, "edge_type": "prev", "scope": "chunk"})
if next_id:
edges.append({"src_id": src, "dst_id": next_id, "edge_type": "next", "scope": "chunk"})
edges.append({"src_id": next_id, "dst_id": src, "edge_type": "prev", "scope": "chunk"})
# Note-Scope references/backlink
for tid in (note_meta.get("references") or []):
edges.append({"src_id": note_meta["id"], "dst_id": tid, "edge_type": "references", "scope": "note"})
edges.append({"src_id": tid, "dst_id": note_meta["id"], "edge_type": "backlink", "scope": "note"})
# Dedupe
uniq = {}
for e in edges:
k = (e["src_id"], e["edge_type"], e["dst_id"], e.get("scope", "note"))
uniq[k] = e
return list(uniq.values())
# -----------------------------
# Main
# -------------------------------------------------
# -----------------------------
def main():
load_dotenv()
ap = argparse.ArgumentParser()
ap.add_argument("--vault", required=True, help="Pfad zum Vault-Root")
ap.add_argument("--apply", action="store_true", help="Schreibt in Qdrant (sonst Dry-Run)")
ap.add_argument(
"--purge-before-upsert",
action="store_true",
help="Vor Upsert alte Chunks/Edges **nur für geänderte** Noten löschen.",
)
ap.add_argument("--note-id", help="Optional: nur diese Note verarbeiten")
ap.add_argument("--vault", required=True, help="Pfad zum Obsidian-Vault (Root-Ordner)")
ap.add_argument("--apply", action="store_true", help="Schreibt in Qdrant; ohne Flag nur Dry-Run")
ap.add_argument("--purge-before-upsert", action="store_true",
help="Vor Upsert Chunks & Edges der GEÄNDERTEN Note löschen")
ap.add_argument("--note-id", help="Nur eine bestimmte Note-ID verarbeiten")
ap.add_argument("--embed-note", action="store_true", help="Optional: Note-Volltext einbetten")
ap.add_argument("--force-replace", action="store_true",
help="Änderungserkennung ignorieren und immer upserten (+ optional Purge)")
args = ap.parse_args()
load_dotenv()
cfg = QdrantConfig(
url=os.getenv("QDRANT_URL", "http://127.0.0.1:6333"),
api_key=os.getenv("QDRANT_API_KEY", None),
prefix=os.getenv("COLLECTION_PREFIX", "mindnet"),
dim=int(os.getenv("VECTOR_DIM", "384")),
)
# Qdrant-Config
cfg = QdrantConfig.from_env()
client = get_client(cfg)
ensure_collections(client, cfg.prefix, cfg.dim)
notes_col, chunks_col, edges_col = collections(cfg.prefix)
vault_root = os.path.abspath(args.vault)
files = iter_md(vault_root)
root = os.path.abspath(args.vault)
files = iter_md(root)
if not files:
print("Keine Markdown-Dateien gefunden.", file=sys.stderr)
sys.exit(2)
# Index einmal bauen (für Linkauflösung bei geänderten Noten)
note_index = build_vault_index(vault_root)
processed = 0
for abs_path in files:
parsed = read_markdown(abs_path)
fm = normalize_frontmatter(parsed.frontmatter or {})
for path in files:
parsed = read_markdown(path)
fm = normalize_frontmatter(parsed.frontmatter)
# Pflichtfelder?
try:
validate_required_frontmatter(fm)
except Exception:
# unvollständige Note überspringen
# Überspringen, aber nicht abbrechen
continue
if args.note_id and fm.get("id") != args.note_id:
continue
processed += 1
# Änderungsdetektion (Datei-Hash vs. Qdrant)
new_hash = file_sha256(abs_path)
old_hash = fetch_existing_note_hash(client, cfg.prefix, fm["id"])
changed = (old_hash != new_hash)
decision = "skip"
if changed:
decision = "apply" if args.apply else "dry-run"
# Bei "skip" kein teures Chunking/Embedding/Edges nötig
if not changed:
print(json.dumps({
"note_id": fm["id"],
"title": fm.get("title"),
"changed": False,
"decision": "skip",
"path": os.path.relpath(abs_path, vault_root),
}, ensure_ascii=False))
continue
# --- Ab hier: Nur für geänderte Noten ---
# Note-Payload erzeugen
from app.core.note_payload import make_note_payload # lazy import
note_pl = make_note_payload(parsed, vault_root=vault_root)
# Hash im Payload mitschreiben (Schema erlaubt hash_fulltext)
note_pl["hash_fulltext"] = new_hash
# Note-Payload & Validierung
note_pl = make_note_payload(parsed, vault_root=root)
validate_note_payload(note_pl)
note_id = note_pl["note_id"]
# Chunking & Payloads
# Change-Detection (Body-Hash)
new_hash = note_pl.get("hash_fulltext")
old_hash = None
if not args.force_replace:
old_hash = fetch_existing_note_hash(client, cfg.prefix, note_id)
changed = args.force_replace or (old_hash != new_hash)
# Chunks + Embeddings
chunks = assemble_chunks(fm["id"], parsed.body, fm.get("type", "concept"))
chunk_pls = make_chunk_payloads(fm, note_pl["path"], chunks)
vecs = embed_texts([c.text for c in chunks])
# Embeddings
texts = [c.get("text") or c.get("content") or "" for c in chunk_pls]
vectors = embed_texts(texts)
# Optional: Note-Vektor
note_vec = embed_one(parsed.body) if args.embed_note else None
# Edges ableiten (Note-/Chunk-Level)
edges = derive_wikilink_edges(note_pl, chunk_pls, note_index)
# Edges
# 1) Versuch: volle Kanten aus Chunk-Payloads
edges = deriv_edges_for_note(fm, chunk_pls)
# 2) Fallback, falls keine Chunk-References vorhanden sind
if not edges or (not has_chunk_level_refs(chunk_pls) and (fm.get("references") or note_pl.get("references"))):
# Ergänze fm['references'] falls noch nicht im fm vorhanden
if "references" not in fm and "references" in note_pl:
fm["references"] = note_pl["references"]
edges = fallback_note_level_edges(fm, chunk_pls)
# Purge (nur wenn apply + Option gesetzt)
if args.apply and args.purge_before_upsert:
# Chunk-IDs bestimmen (für Edge-Purge by source_id)
chunk_ids = []
for i, ch in enumerate(chunk_pls, start=1):
cid = ch.get("chunk_id") or ch.get("id") or f"{fm['id']}#{i}"
ch["chunk_id"] = cid # sicherstellen
chunk_ids.append(cid)
purge_for_note(client, cfg.prefix, fm["id"], chunk_ids)
# Upserts (nur Apply)
if args.apply:
notes_col, note_pts = points_for_note(cfg.prefix, note_pl, note_vec=None, dim=cfg.dim)
upsert_batch(client, notes_col, note_pts)
chunks_col, chunk_pts = points_for_chunks(cfg.prefix, chunk_pls, vectors)
upsert_batch(client, chunks_col, chunk_pts)
edges_col, edge_pts = points_for_edges(cfg.prefix, edges)
upsert_batch(client, edges_col, edge_pts)
# Logging geänderte Note
print(json.dumps({
"note_id": fm["id"],
# Zusammenfassung für Log
summary = {
"note_id": note_id,
"title": fm.get("title"),
"chunks": len(chunk_pls),
"edges": len(edges),
"changed": True,
"decision": decision,
"changed": changed,
"decision": ("apply" if args.apply and changed else
"apply-skip-unchanged" if args.apply and not changed else
"dry-run"),
"path": note_pl["path"],
}, ensure_ascii=False))
}
print(json.dumps(summary, ensure_ascii=False))
# Apply?
if not args.apply:
continue
# Optionaler Purge NUR für geänderte Notes
if changed and args.purge_before_upsert:
purge_note_artifacts(client, cfg.prefix, note_id)
# Upserts
# Notes
notes_name, note_pts = points_for_note(cfg.prefix, note_pl, note_vec, cfg.dim)
upsert_batch(client, notes_name, note_pts)
# Chunks
chunks_name, chunk_pts = points_for_chunks(cfg.prefix, chunk_pls, vecs)
upsert_batch(client, chunks_name, chunk_pts)
# Edges
edges_name, edge_pts = points_for_edges(cfg.prefix, edges)
upsert_batch(client, edges_name, edge_pts)
print(f"Done. Processed notes: {processed}")