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

Fix kg-core downloading (#371)

Browse source
This commit is contained in:
cybermaggedon 2025-05-07 14:38:48 +01:00 • committed by GitHub
parent 4461d7b289
commit fdd9a9a9ae
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 16 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

24
trustgraph-cli/scripts/tg-get-kg-core
View file

@ -68,7 +68,7 @@ async def fetch(url, user, id, output):
"id": mid,
"service": "knowledge",
"request": {
"operation": "fetch-kg-core",
"operation": "get-kg-core",
"user": user,
"id": id,
}
@ -87,22 +87,24 @@ async def fetch(url, user, id, output):
obj = json.loads(msg)
if "error" in obj:
if "response" not in obj:
raise RuntimeError("No response?")
response = obj["response"]
if "error" in response:
raise RuntimeError(obj["error"])
if "response" not in obj:
continue
if "eos" in response:
if response["eos"]: break
if "eos" in obj["response"]:
if obj["response"]["eos"]: break
if "triples" in obj["response"]:
if "triples" in response:
t += 1
write_triple(f, obj["response"]["triples"])
write_triple(f, response["triples"])
if "graph-embeddings" in obj["response"]:
if "graph-embeddings" in response:
ge += 1
write_ge(f, obj["response"]["graph-embeddings"])
write_ge(f, response["graph-embeddings"])
print(f"Got: {t} triple, {ge} GE messages.")