调整json处理方案

main
rogersun 3 days ago
parent 6594d98953
commit bea8d2db1c
  1. 2
      ai/utils/show_process.py
  2. 2
      ai/utils/show_prompt.py

@ -41,7 +41,7 @@ def show_main_process(zz_code=None, day_delta=3):
print('message:', message) print('message:', message)
print('tokens:', tokens) print('tokens:', tokens)
# 获取排片数据 # 获取排片数据
result_obj = json.loads(result) result_obj = result
# 预测排片数据 # 预测排片数据
_show = result_obj['show'].replace('```', '').replace('csv', '').strip() _show = result_obj['show'].replace('```', '').replace('csv', '').strip()
# 预测销售数据 # 预测销售数据

@ -144,4 +144,4 @@ class ShowAI:
) )
# print(dict(response)) # print(dict(response))
# print(response.choices[0].message.content) # print(response.choices[0].message.content)
return response.choices[0].message.content, response.model_dump_json(), response.usage.model_dump_json() return response.choices[0].message.content.model_dump_json(), response.model_dump_json(), response.usage.model_dump_json()

Loading…
Cancel
Save