修改sql推理模型
This commit is contained in:
parent
235bc9e99d
commit
8d923b663c
@ -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'])
|
||||||
|
Loading…
Reference in New Issue
Block a user