Przeglądaj źródła

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 1 rok temu
rodzic
commit
923c3b8cac
No account linked to committer's email address
2 zmienionych plików z 3 dodań i 3 usunięć
  1. 1
    1
      api/apps/api_app.py
  2. 2
    2
      rag/llm/embedding_model.py

+ 1
- 1
api/apps/api_app.py Wyświetl plik

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

+ 2
- 2
rag/llm/embedding_model.py Wyświetl plik

token_count += resp["usage"]["total_tokens"] token_count += resp["usage"]["total_tokens"]
return np.array(res), token_count return np.array(res), token_count
except Exception as e: 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 return np.array([]), 0


def encode_queries(self, text): def encode_queries(self, text):
return np.array(resp["output"]["embeddings"][0] return np.array(resp["output"]["embeddings"][0]
["embedding"]), resp["usage"]["total_tokens"] ["embedding"]), resp["usage"]["total_tokens"]
except Exception as e: 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 return np.array([]), 0





Ładowanie…
Anuluj
Zapisz