Parcourir la source

Fix: wrong type hint (#7738)

### What problem does this PR solve?

Wrong hint type. #7729.

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
tags/v0.19.0
Yongteng Lei il y a 5 mois
Parent
révision
0c2b8182e4
Aucun compte lié à l'adresse e-mail de l'auteur
1 fichiers modifiés avec 1 ajouts et 2 suppressions
  1. 1
    2
      api/db/services/dialog_service.py

+ 1
- 2
api/db/services/dialog_service.py Voir le fichier

@@ -302,7 +302,7 @@ def chat(dialog, messages, stream=True, **kwargs):
if "max_tokens" in gen_conf:
gen_conf["max_tokens"] = min(gen_conf["max_tokens"], max_tokens - used_token_count)

def repair_bad_citation_formats(answer: str, kbinfos: dict, idx: dict):
def repair_bad_citation_formats(answer: str, kbinfos: dict, idx: set):
max_index = len(kbinfos["chunks"])

def safe_add(i):
@@ -623,4 +623,3 @@ def ask(question, kb_ids, tenant_id):
answer = ans
yield {"answer": answer, "reference": {}}
yield decorate_answer(answer)


Chargement…
Annuler
Enregistrer