From 697214635bfb44e8d29f1ca421836074abdd9d49 Mon Sep 17 00:00:00 2001 From: Cyber MacGeddon Date: Mon, 12 Aug 2024 16:42:21 +0100 Subject: [PATCH] Change trustgraph.trustgraph to trustgraph.direct.cassandra --- trustgraph/{trustgraph.py => direct/cassandra.py} | 0 trustgraph/dump/graph_embeddings/parquet/processor.py | 2 -- trustgraph/dump/triples/parquet/processor.py | 2 -- trustgraph/graph_rag.py | 7 +++---- trustgraph/storage/triples/cassandra/write.py | 2 +- 5 files changed, 4 insertions(+), 9 deletions(-) rename trustgraph/{trustgraph.py => direct/cassandra.py} (100%) diff --git a/trustgraph/trustgraph.py b/trustgraph/direct/cassandra.py similarity index 100% rename from trustgraph/trustgraph.py rename to trustgraph/direct/cassandra.py diff --git a/trustgraph/dump/graph_embeddings/parquet/processor.py b/trustgraph/dump/graph_embeddings/parquet/processor.py index 73057362..795f3351 100755 --- a/trustgraph/dump/graph_embeddings/parquet/processor.py +++ b/trustgraph/dump/graph_embeddings/parquet/processor.py @@ -9,10 +9,8 @@ import os import argparse import time -from .... trustgraph import TrustGraph from .... schema import GraphEmbeddings from .... schema import graph_embeddings_store_queue -from .... log_level import LogLevel from .... base import Consumer from . writer import ParquetWriter diff --git a/trustgraph/dump/triples/parquet/processor.py b/trustgraph/dump/triples/parquet/processor.py index 42bebbe3..553a62b8 100755 --- a/trustgraph/dump/triples/parquet/processor.py +++ b/trustgraph/dump/triples/parquet/processor.py @@ -9,10 +9,8 @@ import os import argparse import time -from .... trustgraph import TrustGraph from .... schema import Triple from .... schema import triples_store_queue -from .... log_level import LogLevel from .... base import Consumer from . writer import ParquetWriter diff --git a/trustgraph/graph_rag.py b/trustgraph/graph_rag.py index 4aea074c..c4ebe158 100644 --- a/trustgraph/graph_rag.py +++ b/trustgraph/graph_rag.py @@ -1,9 +1,8 @@ -from trustgraph.trustgraph import TrustGraph +from . direct.cassandra import TrustGraph from . direct.milvus import TripleVectors -from trustgraph.trustgraph import TrustGraph -from trustgraph.llm_client import LlmClient -from trustgraph.embeddings_client import EmbeddingsClient +from . llm_client import LlmClient +from . embeddings_client import EmbeddingsClient from . schema import text_completion_request_queue from . schema import text_completion_response_queue from . schema import embeddings_request_queue diff --git a/trustgraph/storage/triples/cassandra/write.py b/trustgraph/storage/triples/cassandra/write.py index 7cff6304..5ba657aa 100755 --- a/trustgraph/storage/triples/cassandra/write.py +++ b/trustgraph/storage/triples/cassandra/write.py @@ -9,7 +9,7 @@ import os import argparse import time -from .... trustgraph import TrustGraph +from .... direct.cassandra import TrustGraph from .... schema import Triple from .... schema import triples_store_queue from .... log_level import LogLevel