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):