zy123/zbparse
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

1.17 小bug

Browse Source
This commit is contained in:
zy123 2025-01-17 16:40:07 +08:00
parent 94b945e6b1
commit 20e297c3ad
2 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
flask_app/general/多线程提问.py
View File

@ -157,7 +157,7 @@ def llm_call(question, knowledge_name,file_id, result_queue, ans_index, llm_type
ans = send_message(assistant, message=question) ans = send_message(assistant, message=question)
result_queue.put((ans_index, (question, ans))) # 在队列中添加索引 (question, ans) result_queue.put((ans_index, (question, ans))) # 在队列中添加索引 (question, ans)
elif llm_type==2: elif llm_type==2:
print(f"qianwen_long! question:{question}") # print(f"qianwen_long! question:{question}")
# qianwen_res,usage = qianwen_long(file_id,question) #有bug # qianwen_res,usage = qianwen_long(file_id,question) #有bug
qianwen_res = qianwen_long(file_id, question,2,1,need_extra) qianwen_res = qianwen_long(file_id, question,2,1,need_extra)
if not qianwen_res: if not qianwen_res:
@ -172,6 +172,12 @@ def llm_call(question, knowledge_name,file_id, result_queue, ans_index, llm_type
result_queue.put((ans_index, None)) # 如果为空字符串，直接返回 None result_queue.put((ans_index, None)) # 如果为空字符串，直接返回 None
else: else:
result_queue.put((ans_index, (question, qianwen_plus_res))) result_queue.put((ans_index, (question, qianwen_plus_res)))
elif llm_type==4:
qianwen_plus_res = qianwen_plus(question, need_extra)
if not qianwen_plus_res:
result_queue.put((ans_index, None)) # 如果为空字符串，直接返回 None
else:
result_queue.put((ans_index, (question, qianwen_plus_res)))
else : else :
assistant = pure_assistant() assistant = pure_assistant()
ans = send_message(assistant, message=question) ans = send_message(assistant, message=question)

3
flask_app/货物标/提取采购需求main.py
View File

@ -33,8 +33,7 @@ def fetch_procurement_reqs(procurement_path, invalid_path):
if proc_path == invalid_path: if proc_path == invalid_path:
# 读取 PDF 页码数 # 读取 PDF 页码数
page_count = get_pdf_page_count(procurement_path) page_count = get_pdf_page_count(procurement_path)
if page_count > 60: # 如果页码数大于60,不转markdown
if page_count > 80: # 如果页码数大于 580
tech_model_type= 2 #long tech_model_type= 2 #long
busi_model_type=3 #long-stream busi_model_type=3 #long-stream
processed_filepath = "" processed_filepath = ""