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

Tidying clients

Browse source
This commit is contained in:
Cyber MacGeddon 2025-04-19 13:41:33 +01:00
parent 93bb2adb9e
commit dc0bcba269
2 changed files with 10 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

5
trustgraph-base/trustgraph/base/embedings_client.py
View file

@ -4,12 +4,17 @@ from .. schema import EmbeddingsReqeust, EmbeddingsResponse
class EmbeddingsClient(RequestResponse):
async def embed(self, text, timeout=30):
resp = await self.request(
EmbeddingsRequest(
text = text
),
timeout=timeout
)
if resp.error:
raise RuntimeError(resp.error.message)
return resp.vectors
class EmbeddingsClientSpec(RequestResponseSpec):