First commit

scripts/generate_rag_embeddings.py

@@ -0,0 +1,47 @@
+#!/usr/bin/env python3
+from __future__ import annotations
+
+import argparse
+import sys
+from pathlib import Path
+
+ROOT = Path(__file__).resolve().parents[1]
+FRAMEWORK_SRC = ROOT / "agent_framework" / "src"
+if str(FRAMEWORK_SRC) not in sys.path:
+    sys.path.insert(0, str(FRAMEWORK_SRC))
+
+from agent_framework.config.settings import get_settings
+from agent_framework.rag.ingest import ingest_documents_sync, parse_csv
+
+
+def main() -> int:
+    parser = argparse.ArgumentParser(
+        description="Generate RAG embeddings and store document chunks in the configured vector store."
+    )
+    parser.add_argument("--docs-dir", default=None, help="Directory containing Markdown/text/YAML/JSON documents.")
+    parser.add_argument("--namespace", default=None, help="RAG namespace used by the agent profile.")
+    parser.add_argument("--globs", default=None, help="Comma-separated file globs. Example: '*.md,*.txt'.")
+    parser.add_argument("--chunk-size", type=int, default=None, help="Maximum chunk size in characters.")
+    parser.add_argument("--chunk-overlap", type=int, default=None, help="Chunk overlap in characters.")
+    args = parser.parse_args()
+
+    settings = get_settings()
+    result = ingest_documents_sync(
+        settings,
+        docs_dir=args.docs_dir,
+        namespace=args.namespace,
+        file_globs=parse_csv(args.globs, []) or None,
+        chunk_size=args.chunk_size,
+        chunk_overlap=args.chunk_overlap,
+    )
+
+    print("RAG embedding generation completed")
+    print(f"  namespace:      {result.namespace}")
+    print(f"  files read:     {result.files_read}")
+    print(f"  chunks created: {result.chunks_created}")
+    print(f"  documents saved:{result.documents_saved}")
+    return 0
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())