From 72814c2029fc61b155718f9b2eab82f72814f4a6 Mon Sep 17 00:00:00 2001
From: Cyber MacGeddon <cybermaggedon@gmail.com>
Date: Tue, 16 Jul 2024 17:30:30 +0100
Subject: [PATCH] Fix weird naming

---
 tests/test-milvus                             | 4 ++--
 trustgraph/graph_rag.py                       | 4 ++--
 trustgraph/{edge_map.py => triple_vectors.py} | 2 +-
 trustgraph/vector/milvus_write/write.py       | 4 ++--
 4 files changed, 7 insertions(+), 7 deletions(-)
 rename trustgraph/{edge_map.py => triple_vectors.py} (99%)

diff --git a/tests/test-milvus b/tests/test-milvus
index e95955dc..7eab7938 100755
--- a/tests/test-milvus
+++ b/tests/test-milvus
@@ -2,9 +2,9 @@
 
 from langchain_huggingface import HuggingFaceEmbeddings
 
-from edge_map import VectorStore
+from trustgraph.triple_vectors import TripleVectors
 
-client = VectorStore()
+client = TripleVectors()
 
 embeddings = HuggingFaceEmbeddings(model_name="all-MiniLM-L6-v2")
 
diff --git a/trustgraph/graph_rag.py b/trustgraph/graph_rag.py
index 4175698e..d43ece48 100644
--- a/trustgraph/graph_rag.py
+++ b/trustgraph/graph_rag.py
@@ -1,6 +1,6 @@
 
 from trustgraph.trustgraph import TrustGraph
-from trustgraph.edge_map import VectorStore
+from trustgraph.triple_vectors import TripleVectors
 from trustgraph.trustgraph import TrustGraph
 from trustgraph.llm_client import LlmClient
 from trustgraph.embeddings_client import EmbeddingsClient
@@ -30,7 +30,7 @@ class GraphRag:
 
         self.embeddings = EmbeddingsClient(pulsar_host=pulsar_host)
 
-        self.vecstore = VectorStore(vector_store)
+        self.vecstore = TripleVectors(vector_store)
 
         self.entity_limit=50
         self.query_limit=30
diff --git a/trustgraph/edge_map.py b/trustgraph/triple_vectors.py
similarity index 99%
rename from trustgraph/edge_map.py
rename to trustgraph/triple_vectors.py
index 55d9077f..f631c506 100644
--- a/trustgraph/edge_map.py
+++ b/trustgraph/triple_vectors.py
@@ -1,7 +1,7 @@
 
 from pymilvus import MilvusClient, CollectionSchema, FieldSchema, DataType
 
-class VectorStore:
+class TripleVectors:
 
     def __init__(self, uri="http://localhost:19530"):
 
diff --git a/trustgraph/vector/milvus_write/write.py b/trustgraph/vector/milvus_write/write.py
index b25fa329..579c576d 100755
--- a/trustgraph/vector/milvus_write/write.py
+++ b/trustgraph/vector/milvus_write/write.py
@@ -14,7 +14,7 @@ import time
 
 from ... schema import VectorsAssociation
 from ... log_level import LogLevel
-from ... edge_map import VectorStore
+from ... triple_vectors import TripleVectors
 
 default_pulsar_host = os.getenv("PULSAR_HOST", 'pulsar://pulsar:6650')
 default_input_queue = 'vectors-load'
@@ -44,7 +44,7 @@ class Processor:
             schema=JsonSchema(VectorsAssociation),
         )
 
-        self.vecstore = VectorStore(store_uri)
+        self.vecstore = TripleVectors(store_uri)
 
     def run(self):