Make the update script shorter. (#4854)

### What problem does this PR solve?


### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
Kevin Hu
2025-02-10 18:18:49 +08:00
committed by GitHub
parent bc68f18c48
commit 0d3ed37b48
3 changed files with 10 additions and 6 deletions

View File

@ -244,11 +244,16 @@ class KGSearch(Dealer):
else:
continue
rel["description"] = rela["description"]
desc = rel["description"]
try:
desc = json.loads(desc).get("description", "")
except Exception:
pass
relas.append({
"From Entity": f,
"To Entity": t,
"Score": "%.2f" % (rel["sim"] * rel["pagerank"]),
"Description": json.loads(rel["description"]).get("description", "")
"Description": desc
})
max_token -= num_tokens_from_string(str(relas[-1]))
if max_token <= 0: