调整json处理方案

main
rogersun 2 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('tokens:', tokens)
# 获取排片数据
result_obj = json.loads(result)
result_obj = result
# 预测排片数据
_show = result_obj['show'].replace('```', '').replace('csv', '').strip()
# 预测销售数据

@ -144,4 +144,4 @@ class ShowAI:
)
# print(dict(response))
# 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