Fix weird naming

This commit is contained in:
Cyber MacGeddon 2024-07-16 17:30:30 +01:00
parent 50a114fbbe
commit 72814c2029
4 changed files with 7 additions and 7 deletions

View file

@ -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")

View file

@ -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

View file

@ -1,7 +1,7 @@
from pymilvus import MilvusClient, CollectionSchema, FieldSchema, DataType
class VectorStore:
class TripleVectors:
def __init__(self, uri="http://localhost:19530"):

View file

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