KevinHuSh commited on
Commit
21cb28c
·
1 Parent(s): 40094ce

fix bug in api (#1088)

Browse files

### What problem does this PR solve?

#1075

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)

api/apps/api_app.py CHANGED
@@ -208,7 +208,7 @@ def completion():
208
  try:
209
  for ans in chat(dia, msg, True, **req):
210
  fillin_conv(ans)
211
- rename_field(rename_field)
212
  yield "data:" + json.dumps({"retcode": 0, "retmsg": "", "data": ans}, ensure_ascii=False) + "\n\n"
213
  API4ConversationService.append_message(conv.id, conv.to_dict())
214
  except Exception as e:
 
208
  try:
209
  for ans in chat(dia, msg, True, **req):
210
  fillin_conv(ans)
211
+ rename_field(ans)
212
  yield "data:" + json.dumps({"retcode": 0, "retmsg": "", "data": ans}, ensure_ascii=False) + "\n\n"
213
  API4ConversationService.append_message(conv.id, conv.to_dict())
214
  except Exception as e:
rag/llm/embedding_model.py CHANGED
@@ -140,7 +140,7 @@ class QWenEmbed(Base):
140
  token_count += resp["usage"]["total_tokens"]
141
  return np.array(res), token_count
142
  except Exception as e:
143
- raise Exception("Account abnormal. Please ensure it's on good standing.")
144
  return np.array([]), 0
145
 
146
  def encode_queries(self, text):
@@ -153,7 +153,7 @@ class QWenEmbed(Base):
153
  return np.array(resp["output"]["embeddings"][0]
154
  ["embedding"]), resp["usage"]["total_tokens"]
155
  except Exception as e:
156
- raise Exception("Account abnormal. Please ensure it's on good standing.")
157
  return np.array([]), 0
158
 
159
 
 
140
  token_count += resp["usage"]["total_tokens"]
141
  return np.array(res), token_count
142
  except Exception as e:
143
+ raise Exception("Account abnormal. Please ensure it's on good standing to use QWen's "+self.model_name)
144
  return np.array([]), 0
145
 
146
  def encode_queries(self, text):
 
153
  return np.array(resp["output"]["embeddings"][0]
154
  ["embedding"]), resp["usage"]["total_tokens"]
155
  except Exception as e:
156
+ raise Exception("Account abnormal. Please ensure it's on good standing to use QWen's "+self.model_name)
157
  return np.array([]), 0
158
 
159