diff --git a/trustgraph/graph_rag.py b/trustgraph/graph_rag.py
index 135797da..b37d35a1 100644
--- a/trustgraph/graph_rag.py
+++ b/trustgraph/graph_rag.py
@@ -6,6 +6,8 @@ from trustgraph.llm_client import LlmClient
 from trustgraph.embeddings_client import EmbeddingsClient
 from . schema import text_completion_request_queue
 from . schema import text_completion_response_queue
+from . schema import embeddings_request_queue
+from . schema import embeddings_response_queue
 
 LABEL="http://www.w3.org/2000/01/rdf-schema#label"
 DEFINITION="http://www.w3.org/2004/02/skos/core#definition"