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

trustgraph-base .chunks / .documents confusion in the API

Browse source
This commit is contained in:
Cyber MacGeddon 2025-09-02 16:35:21 +01:00
parent 210d600f78
commit 98c6de5fde
3 changed files with 6 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

2
trustgraph-base/trustgraph/base/document_embeddings_client.py
View file

@ -27,7 +27,7 @@ class DocumentEmbeddingsClient(RequestResponse):
if resp.error:
raise RuntimeError(resp.error.message)
return resp.documents
return resp.chunks
class DocumentEmbeddingsClientSpec(RequestResponseSpec):
def __init__(

2
trustgraph-base/trustgraph/clients/document_embeddings_client.py
View file

@ -47,5 +47,5 @@ class DocumentEmbeddingsClient(BaseClient):
return self.call(
user=user, collection=collection,
vectors=vectors, limit=limit, timeout=timeout
).documents
).chunks

8
trustgraph-base/trustgraph/messaging/translators/embeddings_query.py
View file

@ -36,10 +36,10 @@ class DocumentEmbeddingsResponseTranslator(MessageTranslator):
def from_pulsar(self, obj: DocumentEmbeddingsResponse) -> Dict[str, Any]:
result = {}
if obj.documents:
result["documents"] = [
doc.decode("utf-8") if isinstance(doc, bytes) else doc
for doc in obj.documents
if obj.chunks:
result["chunks"] = [
chunk.decode("utf-8") if isinstance(chunk, bytes) else chunk
for chunk in obj.chunks
]
return result