apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-07-06 03:42:11 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fixup error handling change

Browse source
This commit is contained in:
Cyber MacGeddon 2024-08-22 16:55:17 +01:00
parent 8363090e8d
commit a4a777fad1
6 changed files with 9 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

2
trustgraph/clients/graph_embeddings_client.py
View file

@ -33,7 +33,7 @@ class GraphEmbeddingsClient(BaseClient):
if output_queue == None:
output_queue = graph_embeddings_response_queue
super(EmbeddingsClient, self).__init__(
super(GraphEmbeddingsClient, self).__init__(
log_level=log_level,
subscriber=subscriber,
input_queue=input_queue,

8
trustgraph/clients/graph_rag_client.py
View file

@ -16,7 +16,7 @@ WARN=_pulsar.LoggerLevel.Warn
INFO=_pulsar.LoggerLevel.Info
DEBUG=_pulsar.LoggerLevel.Debug
class GraphRagClient:
class GraphRagClient(BaseClient):
def __init__(
self,
@ -31,9 +31,9 @@ class GraphRagClient:
input_queue = graph_rag_request_queue
if output_queue == None:
output_queue = graph_rag_request_queue
output_queue = graph_rag_response_queue
super(EmbeddingsClient, self).__init__(
super(GraphRagClient, self).__init__(
log_level=log_level,
subscriber=subscriber,
input_queue=input_queue,
@ -47,5 +47,5 @@ class GraphRagClient:
return self.call(
query=query, timeout=timeout
).entities
).response

2
trustgraph/clients/triples_query_client.py
View file

@ -34,7 +34,7 @@ class TriplesQueryClient(BaseClient):
if output_queue == None:
output_queue = triples_response_queue
super(EmbeddingsClient, self).__init__(
super(TriplesQueryClient, self).__init__(
log_level=log_level,
subscriber=subscriber,
input_queue=input_queue,

2
trustgraph/query/graph_embeddings/milvus/service.py
View file

@ -78,7 +78,7 @@ class Processor(ConsumerProducer):
entities = ents2
print("Send response...", flush=True)
r = GraphEmbeddingsResponse(entities=entities)
r = GraphEmbeddingsResponse(entities=entities, error=None)
self.producer.send(r, properties={"id": id})
print("Done.", flush=True)

2
trustgraph/query/triples/cassandra/service.py
View file

@ -130,7 +130,7 @@ class Processor(ConsumerProducer):
]
print("Send response...", flush=True)
r = TriplesQueryResponse(triples=triples)
r = TriplesQueryResponse(triples=triples, error=None)
self.producer.send(r, properties={"id": id})
print("Done.", flush=True)

2
trustgraph/retrieval/graph_rag/rag.py
View file

@ -111,7 +111,7 @@ class Processor(ConsumerProducer):
response = self.rag.query(v.query)
print("Send response...", flush=True)
r = GraphRagResponse(response = response)
r = GraphRagResponse(response = response, error=None)
self.producer.send(r, properties={"id": id})
print("Done.", flush=True)