added llm and rag queries

kg_ocr/embeddings/__init__.py

@@ -1,3 +1,3 @@
-from .indexer import create_and_index, query_embedding
+from .indexer import create_and_index
 
-__all__ = ["create_and_index", "query_embedding"]
+__all__ = ["create_and_index"]

kg_ocr/embeddings/indexer.py

@@ -15,9 +15,4 @@ def create_and_index(
     return embeddings
 
 
-def query_embedding(
-    embeddings: Embeddings, query: str, limit: int = 100
-) -> list[str]:
-    """Search embeddings and return matching texts."""
-    results = embeddings.search(query, limit)
-    return [r["text"] for r in results]
+