Pārlūkot izejas kodu

fix bug in api (#1088)

### What problem does this PR solve?

#1075 

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
tags/v0.8.0
KevinHuSh pirms 1 gada
vecāks
revīzija
923c3b8cac
Revīzijas autora e-pasta adrese nav piesaistīta nevienam kontam
2 mainītis faili ar 3 papildinājumiem un 3 dzēšanām
  1. 1
    1
      api/apps/api_app.py
  2. 2
    2
      rag/llm/embedding_model.py

+ 1
- 1
api/apps/api_app.py Parādīt failu

@@ -208,7 +208,7 @@ def completion():
try:
for ans in chat(dia, msg, True, **req):
fillin_conv(ans)
rename_field(rename_field)
rename_field(ans)
yield "data:" + json.dumps({"retcode": 0, "retmsg": "", "data": ans}, ensure_ascii=False) + "\n\n"
API4ConversationService.append_message(conv.id, conv.to_dict())
except Exception as e:

+ 2
- 2
rag/llm/embedding_model.py Parādīt failu

@@ -140,7 +140,7 @@ class QWenEmbed(Base):
token_count += resp["usage"]["total_tokens"]
return np.array(res), token_count
except Exception as e:
raise Exception("Account abnormal. Please ensure it's on good standing.")
raise Exception("Account abnormal. Please ensure it's on good standing to use QWen's "+self.model_name)
return np.array([]), 0

def encode_queries(self, text):
@@ -153,7 +153,7 @@ class QWenEmbed(Base):
return np.array(resp["output"]["embeddings"][0]
["embedding"]), resp["usage"]["total_tokens"]
except Exception as e:
raise Exception("Account abnormal. Please ensure it's on good standing.")
raise Exception("Account abnormal. Please ensure it's on good standing to use QWen's "+self.model_name)
return np.array([]), 0



Notiek ielāde…
Atcelt
Saglabāt