Skip to content

Commit

Permalink
fix SILICONFLOW rerank error (#2980)
Browse files Browse the repository at this point in the history
### What problem does this PR solve?

#2977

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
  • Loading branch information
KevinHuSh authored Oct 23, 2024
1 parent 91ea559 commit 89d5b24
Show file tree
Hide file tree
Showing 2 changed files with 24 additions and 2 deletions.
23 changes: 21 additions & 2 deletions api/apps/chunk_app.py
Original file line number Diff line number Diff line change
Expand Up @@ -320,9 +320,28 @@ def knowledge_graph():
for id in sres.ids[:2]:
ty = sres.field[id]["knowledge_graph_kwd"]
try:
obj[ty] = json.loads(sres.field[id]["content_with_weight"])
content_json = json.loads(sres.field[id]["content_with_weight"])
except Exception as e:
print(traceback.format_exc(), flush=True)
continue

if ty == 'mind_map':
node_dict = {}

def repeat_deal(content_json, node_dict):
if 'id' in content_json:
if content_json['id'] in node_dict:
node_name = content_json['id']
content_json['id'] += f"({node_dict[content_json['id']]})"
node_dict[node_name] += 1
else:
node_dict[content_json['id']] = 1
if 'children' in content_json and content_json['children']:
for item in content_json['children']:
repeat_deal(item, node_dict)

repeat_deal(content_json, node_dict)

obj[ty] = content_json

return get_json_result(data=obj)

3 changes: 3 additions & 0 deletions rag/llm/rerank_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -344,6 +344,9 @@ def similarity(self, query: str, texts: list):
self.base_url, json=payload, headers=self.headers
).json()
rank = np.zeros(len(texts), dtype=float)
if "results" not in response:
return rank, 0

for d in response["results"]:
rank[d["index"]] = d["relevance_score"]
return (
Expand Down

0 comments on commit 89d5b24

Please sign in to comment.