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

Updated all CLIs

Browse source
This commit is contained in:
Cyber MacGeddon 2025-05-03 10:16:53 +01:00
parent cd88efa332
commit d0cc173112
5 changed files with 13 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

9
trustgraph-cli/scripts/tg-save-doc-embeds
View file

@ -27,9 +27,7 @@ async def fetch_de(running, queue, user, collection, url):
async with aiohttp.ClientSession() as session:
de_url = f"{url}stream/document-embeddings"
async with session.ws_connect(de_url) as ws:
async with session.ws_connect(url) as ws:
while running.get():
@ -117,11 +115,14 @@ async def run(running, **args):
q = asyncio.Queue()
url = args["url"]
flow_id = args["flow_id"]
de_task = asyncio.create_task(
fetch_de(
running=running,
queue=q, user=args["user"], collection=args["collection"],
url=args["url"] + "api/v1/"
url = f"{url}api/v1/flow/{flow_id}/export/document-embeddings"
)
)