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

Fix doc embeddings invocation

Browse source
This commit is contained in:
Cyber MacGeddon 2026-03-09 11:01:15 +00:00
parent d211dcb58d
commit e851ceecbe
4 changed files with 22 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

10
trustgraph-cli/trustgraph/cli/invoke_document_embeddings.py
View file

@ -26,12 +26,14 @@ def query(url, flow_id, query_text, user, collection, limit, token=None):
limit=limit
)
chunk_ids = result.get("chunk_ids", [])
if not chunk_ids:
chunks = result.get("chunks", [])
if not chunks:
print("No matching chunks found.")
else:
for i, chunk_id in enumerate(chunk_ids, 1):
print(f"{i}. {chunk_id}")
for i, chunk in enumerate(chunks, 1):
chunk_id = chunk.get("chunk_id", "")
score = chunk.get("score", 0.0)
print(f"{i}. {chunk_id} (score: {score:.4f})")
finally:
# Clean up socket connection

14
trustgraph-cli/trustgraph/cli/invoke_graph_embeddings.py
View file

@ -27,8 +27,18 @@ def query(url, flow_id, query_text, user, collection, limit, token=None):
)
entities = result.get("entities", [])
for entity in entities:
print(entity)
if not entities:
print("No matching entities found.")
else:
for i, match in enumerate(entities, 1):
entity = match.get("entity", {})
score = match.get("score", 0.0)
# Format entity based on type
if entity.get("is_uri"):
entity_str = entity.get("value", "")
else:
entity_str = f'"{entity.get("value", "")}"'
print(f"{i}. {entity_str} (score: {score:.4f})")
finally:
# Clean up socket connection