added llm and rag queries

2026-03-24 20:15:22 +01:00
parent cbfcf1e315
commit 012549b4bc
12 changed files with 278 additions and 2981 deletions
--- a/kg_ocr/init.py
+++ b/kg_ocr/init.py
@@ -1,4 +1,5 @@
 from .ocr import get_screenshots, extract_text
-from .embeddings import create_and_index, query_embedding
+from .embeddings import create_and_index
+from .rag import retrieve, ask_wllm

-__all__ = ["get_screenshots", "extract_text", "create_and_index", "query_embedding"]
+__all__ = ["get_screenshots", "extract_text", "create_and_index", "retrieve", "ask_wllm"]