main
HuangHai 4 weeks ago
parent 23dd8a0b48
commit f6546da6a7

@ -34,6 +34,7 @@ logger.addHandler(handler)
model = KeyedVectors.load_word2vec_format(MS_MODEL_PATH, binary=False, limit=MS_MODEL_LIMIT)
logger.info(f"模型加载成功,词向量维度: {model.vector_size}")
# 将HTML文件转换为Word文件
def html_to_word_pandoc(html_file, output_file):
subprocess.run(['pandoc', html_file, '-o', output_file])
@ -114,7 +115,6 @@ async def generate_stream(client, milvus_pool, collection_manager, query, docume
logger.info(f"ID: {hit.id}")
logger.info(f"标签: {record['tags']}")
logger.info(f"用户问题: {record['user_input']}")
logger.info(f"时间: {record['timestamp']}")
logger.info(f"距离: {hit.distance}")
logger.info("-" * 40) # 分隔线
@ -148,8 +148,11 @@ async def generate_stream(client, milvus_pool, collection_manager, query, docume
"""
# 调用阿里云大模型
html_content = client.chat(prompt)
yield {"data": html_content}
if len(context) > 0:
html_content = client.chat(prompt)
yield {"data": html_content}
else:
yield {"data": "没有在知识库中找到相关的信息,无法回答此问题。"}
except Exception as e:
yield {"data": f"生成报告时出错: {str(e)}"}
finally:

Loading…
Cancel
Save