fix QWenSeq2txt bug (#2245)
### What problem does this PR solve? #2243 ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
parent
9b4cceb3f7
commit
0fe19f3fbc
@ -43,7 +43,7 @@ def set_api_key():
|
||||
chat_passed, embd_passed, rerank_passed = False, False, False
|
||||
factory = req["llm_factory"]
|
||||
msg = ""
|
||||
for llm in LLMService.query(fid=factory):
|
||||
for llm in LLMService.query(fid=factory)[:3]:
|
||||
if not embd_passed and llm.model_type == LLMType.EMBEDDING.value:
|
||||
mdl = EmbeddingModel[factory](
|
||||
req["api_key"], llm.llm_name, base_url=req.get("base_url"))
|
||||
|
||||
@ -71,7 +71,7 @@ class QWenSeq2txt(Base):
|
||||
ans = ""
|
||||
if result.status_code == HTTPStatus.OK:
|
||||
for sentence in result.get_sentence():
|
||||
ans += str(sentence + '\n')
|
||||
ans += sentence.text.decode('utf-8') + '\n'
|
||||
return ans, num_tokens_from_string(ans)
|
||||
|
||||
return "**ERROR**: " + result.message, 0
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user