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

Update embeddings integration for new batch embeddings interfaces (#669)

Browse source 
* Fix vector extraction

* Fix embeddings integration
This commit is contained in:
cybermaggedon 2026-03-08 19:41:52 +00:00 • committed by GitHub
parent 0a2ce47a88
commit 919b760c05
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
12 changed files with 55 additions and 56 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -72,12 +72,13 @@ class Query:
if self.verbose:
logger.debug("Computing embeddings...")
qembeds = await self.rag.embeddings_client.embed(query)
qembeds = await self.rag.embeddings_client.embed([query])
if self.verbose:
logger.debug("Done.")
return qembeds
# Return the vector set for the first (only) text
return qembeds[0] if qembeds else []
async def get_entities(self, query):