优化AI模型请求参数,增加参数强制返回json格式

main
rogersun 1 week ago
parent 270b44e225
commit f581855743
  1. 3
      ai/utils/basic_func.py
  2. 5
      ai/utils/show_prompt.py

@ -5,6 +5,7 @@ from ai.models import *
from io import StringIO from io import StringIO
from django_redis import get_redis_connection from django_redis import get_redis_connection
from ai.models import CinemaHall from ai.models import CinemaHall
from django.db.models import Q
# 清除redis锁 # 清除redis锁
@ -178,7 +179,7 @@ def get_runtime_val(key: str):
# 获取当前开启的模型 # 获取当前开启的模型
def get_current_model(): def get_current_model():
return AiModel.objects.filter(current_used=True).first() return AiModel.objects.filter(Q(current_used=True) & Q(delete_flag=False)).first()
# 通不过不同key获取数据 # 通不过不同key获取数据

@ -242,7 +242,10 @@ class ShowAI:
], ],
stream=False, stream=False,
reasoning_effort="high", reasoning_effort="high",
extra_body={"thinking": {"type": "enabled"}} extra_body={"thinking": {"type": "enabled"}},
response_format={
'type': 'json_object'
}
) )
res_json = response.model_dump_json() res_json = response.model_dump_json()
res_obj = json.loads(res_json) res_obj = json.loads(res_json)

Loading…
Cancel
Save