apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-05-21 05:15:12 +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

12
trustgraph-base/trustgraph/api/async_socket_client.py
View file

@ -283,8 +283,8 @@ class AsyncSocketFlowInstance:
async def graph_embeddings_query(self, text: str, user: str, collection: str, limit: int = 10, **kwargs):
"""Query graph embeddings for semantic search"""
# First convert text to embeddings vectors
emb_result = await self.embeddings(text=text)
vectors = emb_result.get("vectors", [])
emb_result = await self.embeddings(texts=[text])
vectors = emb_result.get("vectors", [[]])[0]
request = {
"vectors": vectors,
@ -296,9 +296,9 @@ class AsyncSocketFlowInstance:
return await self.client._send_request("graph-embeddings", self.flow_id, request)
async def embeddings(self, text: str, **kwargs):
async def embeddings(self, texts: list, **kwargs):
"""Generate text embeddings"""
request = {"text": text}
request = {"texts": texts}
request.update(kwargs)
return await self.client._send_request("embeddings", self.flow_id, request)
@ -353,8 +353,8 @@ class AsyncSocketFlowInstance:
):
"""Query row embeddings for semantic search on structured data"""
# First convert text to embeddings vectors
emb_result = await self.embeddings(text=text)
vectors = emb_result.get("vectors", [])
emb_result = await self.embeddings(texts=[text])
vectors = emb_result.get("vectors", [[]])[0]
request = {
"vectors": vectors,