From 6f1f9b615c117589219ee1893ae970ef137fca8c Mon Sep 17 00:00:00 2001 From: Cyber MacGeddon Date: Mon, 12 Aug 2024 16:36:53 +0100 Subject: [PATCH] Rename triple_vectors to direct/milvus --- tests/test-milvus | 2 +- trustgraph/direct/__init__.py | 0 trustgraph/{triple_vectors.py => direct/milvus.py} | 0 trustgraph/graph_rag.py | 2 +- trustgraph/storage/graph_embeddings/milvus/write.py | 2 +- 5 files changed, 3 insertions(+), 3 deletions(-) create mode 100644 trustgraph/direct/__init__.py rename trustgraph/{triple_vectors.py => direct/milvus.py} (100%) 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])