apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-05-20 21:05:13 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Fix weird naming

Browse source
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
Download patch file Download diff file Expand all files Collapse all files

4
tests/test-milvus
View file

@ -2,9 +2,9 @@
from langchain_huggingface import HuggingFaceEmbeddings 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") embeddings = HuggingFaceEmbeddings(model_name="all-MiniLM-L6-v2")

4
trustgraph/graph_rag.py
View file

@ -1,6 +1,6 @@
from trustgraph.trustgraph import TrustGraph from trustgraph.trustgraph import TrustGraph
from trustgraph.edge_map import VectorStore from trustgraph.triple_vectors import TripleVectors
from trustgraph.trustgraph import TrustGraph from trustgraph.trustgraph import TrustGraph
from trustgraph.llm_client import LlmClient from trustgraph.llm_client import LlmClient
from trustgraph.embeddings_client import EmbeddingsClient from trustgraph.embeddings_client import EmbeddingsClient
@ -30,7 +30,7 @@ class GraphRag:
self.embeddings = EmbeddingsClient(pulsar_host=pulsar_host) self.embeddings = EmbeddingsClient(pulsar_host=pulsar_host)
self.vecstore = VectorStore(vector_store) self.vecstore = TripleVectors(vector_store)
self.entity_limit=50 self.entity_limit=50
self.query_limit=30 self.query_limit=30

2
trustgraph/edge_map.py → trustgraph/triple_vectors.py
View file

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

4
trustgraph/vector/milvus_write/write.py
View file

@ -14,7 +14,7 @@ import time
from ... schema import VectorsAssociation from ... schema import VectorsAssociation
from ... log_level import LogLevel 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_pulsar_host = os.getenv("PULSAR_HOST", 'pulsar://pulsar:6650')
default_input_queue = 'vectors-load' default_input_queue = 'vectors-load'
@ -44,7 +44,7 @@ class Processor:
schema=JsonSchema(VectorsAssociation), schema=JsonSchema(VectorsAssociation),
) )
self.vecstore = VectorStore(store_uri) self.vecstore = TripleVectors(store_uri)
def run(self): def run(self):