apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-04-25 16:36: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

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

@ -27,8 +27,23 @@ 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 (wire format uses compact keys)
term_type = entity.get("t", "")
if term_type == "i": # IRI
entity_str = entity.get("i", "")
elif term_type == "l": # Literal
entity_str = f'"{entity.get("v", "")}"'
elif term_type == "b": # Blank node
entity_str = f'_:{entity.get("d", "")}'
else:
entity_str = str(entity)
print(f"{i}. {entity_str} (score: {score:.4f})")
finally:
# Clean up socket connection