|
@@ -12,10 +12,13 @@ client = OpenAI(
|
|
|
)
|
|
|
|
|
|
|
|
|
+DEFAULT_USER_MSG = "解析文件中的表格内容:要求准确识别金额等小数的位数,日期用yyyy-MM-dd的格式返回,去掉金额单位、英文和多余的空格,结果用字典返回"
|
|
|
+
|
|
|
+
|
|
|
@app.get("/upload-filepath")
|
|
|
async def create_upload_file(filepath: str = None,
|
|
|
file_id: str = None,
|
|
|
- user_msg: str = '请准确提取文件中的表格内容,要求用字典返回并去掉金额单位、英文和空格'):
|
|
|
+ user_msg: str = DEFAULT_USER_MSG):
|
|
|
# 读取文件内容(可选)
|
|
|
# contents = await file.read()
|
|
|
|
|
@@ -30,6 +33,8 @@ async def create_upload_file(filepath: str = None,
|
|
|
# 初始化messages列表
|
|
|
completion = client.chat.completions.create(
|
|
|
model="qwen-long",
|
|
|
+ temperature=0.1,
|
|
|
+ presence_penalty=1,
|
|
|
messages=[
|
|
|
{'role': 'system', 'content': 'You are a helpful assistant.'},
|
|
|
{'role': 'system', 'content': f'fileid://{file_id}'},
|
|
@@ -43,7 +48,7 @@ async def create_upload_file(filepath: str = None,
|
|
|
@app.post("/upload-file")
|
|
|
async def create_upload_file(file: UploadFile = File(...),
|
|
|
file_id: str = None,
|
|
|
- user_msg: str = '请准确提取文件中的表格内容,要求用字典返回并去掉金额单位、英文和空格'):
|
|
|
+ user_msg: str = DEFAULT_USER_MSG):
|
|
|
if file_id is None:
|
|
|
# 读取文件内容(可选)
|
|
|
contents = await file.read()
|
|
@@ -58,6 +63,8 @@ async def create_upload_file(file: UploadFile = File(...),
|
|
|
# 初始化messages列表
|
|
|
completion = client.chat.completions.create(
|
|
|
model="qwen-long",
|
|
|
+ temperature=0.1,
|
|
|
+ presence_penalty=1,
|
|
|
messages=[
|
|
|
{'role': 'system', 'content': 'You are a helpful assistant.'},
|
|
|
{'role': 'system', 'content': f'fileid://{file_id}'},
|