apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-07-23 04:01:02 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Hacking wth graph rag

Browse source
This commit is contained in:
Cyber MacGeddon 2025-04-21 18:52:52 +01:00
parent 5f9481dfed
commit 179285e186
5 changed files with 28 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

2
trustgraph-flow/trustgraph/query/graph_embeddings/qdrant/service.py
View file

@ -40,7 +40,7 @@ class Processor(GraphEmbeddingsQueryService):
else:
return Value(value=ent, is_uri=False)
async def handle(self, msg):
async def query_graph_embeddings(self, msg):
try:

16
trustgraph-flow/trustgraph/retrieval/graph_rag/graph_rag.py
View file

@ -25,7 +25,7 @@ class Query:
if self.verbose:
print("Compute embeddings...", flush=True)
qembeds = await self.rag.embeddings.request(query)
qembeds = await self.rag.embeddings_client.embed(query)
if self.verbose:
print("Done.", flush=True)
@ -39,11 +39,13 @@ class Query:
if self.verbose:
print("Get entities...", flush=True)
entities = await self.rag.ge_client.request(
user=self.user, collection=self.collection,
entities = await self.rag.graph_embeddings_client.query(
vectors=vectors, limit=self.entity_limit,
user=self.user, collection=self.collection,
)
print("ENT>", entities, flush=True)
entities = [
e.value
for e in entities
@ -83,10 +85,10 @@ class Query:
if len(subgraph) >= self.max_subgraph_size:
return
res = await self.rag.triples_client.request(
user=self.user, collection=self.collection,
res = await self.rag.triples_client.query(
s=ent, p=None, o=None,
limit=self.triple_limit
limit=self.triple_limit,
user=self.user, collection=self.collection,
)
for triple in res:
@ -208,7 +210,7 @@ class GraphRag:
print(kg)
print(query)
resp = await self.prompt.request_kg_prompt(query, kg)
resp = await self.prompt_client.request_kg_prompt(query, kg)
if self.verbose:
print("Done", flush=True)

4
trustgraph-flow/trustgraph/retrieval/graph_rag/rag.py
View file

@ -154,10 +154,6 @@ class Processor(FlowProcessor):
properties = {"id": id}
)
await self.send(r, properties={"id": id})
self.consumer.acknowledge(msg)
@staticmethod
def add_args(parser):