修改sql推理模型

This commit is contained in:
Xiaofeng 2025-05-21 16:40:02 +08:00
parent 235bc9e99d
commit 8d923b663c

View File

@ -766,7 +766,7 @@ class CreatorSQLSearchAPI(APIView):
# 2. 让大模型生成SQL # 2. 让大模型生成SQL
response = client.chat( response = client.chat(
model="deepseek-r1:70b", model="qwen2.5:32b",
messages=[{'role': 'user', 'content': prompt}], messages=[{'role': 'user', 'content': prompt}],
) )
sql = self._extract_sql(response['message']['content']) sql = self._extract_sql(response['message']['content'])