pengjunjie/ragflow
1
0
Fork 1
mirror of https://github.com/infiniflow/ragflow.git synced 2025-12-08 20:42:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

Test chat API and refine ppt chunker (#42)

Browse Source
This commit is contained in:
KevinHuSh
2024-01-23 19:45:36 +08:00
committed by GitHub
parent 34b2ab3b2f
commit e32ef75e99
10 changed files with 226 additions and 91 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
rag/llm/chat_model.py
View File

@ -37,7 +37,7 @@ class GptTurbo(Base):
model=self.model_name,
messages=history,
**gen_conf)
return res.choices[0].message.content.strip()
return res.choices[0].message.content.strip(), res.usage.completion_tokens
from dashscope import Generation
@ -56,5 +56,5 @@ class QWenChat(Base):
result_format='message'
)
if response.status_code == HTTPStatus.OK:
return response.output.choices[0]['message']['content']
return response.message
return response.output.choices[0]['message']['content'], response.usage.output_tokens
return response.message, 0