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

Fix doc embeddings invocation (#672)

Browse source 
* Fix doc embeddings invocation

* Tidy query embeddings invocation
This commit is contained in:
cybermaggedon 2026-03-09 11:07:32 +00:00 • committed by GitHub
parent f2ae0e8623
commit b2ef7bbb8c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 27 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