mirror of
https://github.com/trustgraph-ai/trustgraph.git
synced 2026-05-18 03:45:12 +02:00
Fix weird naming
This commit is contained in:
parent
50a114fbbe
commit
72814c2029
4 changed files with 7 additions and 7 deletions
|
|
@ -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")
|
||||
|
||||
|
|
|
|||
|
|
@ -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
|
||||
|
|
|
|||
|
|
@ -1,7 +1,7 @@
|
|||
|
||||
from pymilvus import MilvusClient, CollectionSchema, FieldSchema, DataType
|
||||
|
||||
class VectorStore:
|
||||
class TripleVectors:
|
||||
|
||||
def __init__(self, uri="http://localhost:19530"):
|
||||
|
||||
|
|
@ -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):
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue