diff --git a/tests/test-milvus b/tests/test-milvus
index 7eab7938..6d134666 100755
--- a/tests/test-milvus
+++ b/tests/test-milvus
@@ -2,7 +2,7 @@
 
 from langchain_huggingface import HuggingFaceEmbeddings
 
-from trustgraph.triple_vectors import TripleVectors
+from trustgraph.direct.milvus import TripleVectors
 
 client = TripleVectors()
 
diff --git a/trustgraph/direct/__init__.py b/trustgraph/direct/__init__.py
new file mode 100644
index 00000000..e69de29b
diff --git a/trustgraph/triple_vectors.py b/trustgraph/direct/milvus.py
similarity index 100%
rename from trustgraph/triple_vectors.py
rename to trustgraph/direct/milvus.py
diff --git a/trustgraph/graph_rag.py b/trustgraph/graph_rag.py
index b37d35a1..4aea074c 100644
--- a/trustgraph/graph_rag.py
+++ b/trustgraph/graph_rag.py
@@ -1,6 +1,6 @@
 
 from trustgraph.trustgraph import TrustGraph
-from trustgraph.triple_vectors import TripleVectors
+from . direct.milvus import TripleVectors
 from trustgraph.trustgraph import TrustGraph
 from trustgraph.llm_client import LlmClient
 from trustgraph.embeddings_client import EmbeddingsClient
diff --git a/trustgraph/storage/graph_embeddings/milvus/write.py b/trustgraph/storage/graph_embeddings/milvus/write.py
index 1a6883a1..e5141ca8 100755
--- a/trustgraph/storage/graph_embeddings/milvus/write.py
+++ b/trustgraph/storage/graph_embeddings/milvus/write.py
@@ -6,7 +6,7 @@ Accepts entity/vector pairs and writes them to a Milvus store.
 from .... schema import GraphEmbeddings
 from .... schema import graph_embeddings_store_queue
 from .... log_level import LogLevel
-from .... triple_vectors import TripleVectors
+from .... direct.milvus import TripleVectors
 from .... base import Consumer
 
 module = ".".join(__name__.split(".")[1:-1])