add: 添加命令ai model list的逻辑
This commit is contained in:
@@ -13,7 +13,7 @@ system_prompt: ""
|
||||
platforms:
|
||||
local_ai:
|
||||
type: ollama
|
||||
url: http://localhost:11434
|
||||
url: http://192.168.32.162:11434
|
||||
api_key:
|
||||
model: llama3.2
|
||||
max_words: 1000
|
||||
|
||||
@@ -47,7 +47,7 @@ class Ollama(Platform):
|
||||
if response.status != 200:
|
||||
return []
|
||||
response_data = await response.json()
|
||||
return [f"{i + 1}: {model['name']}" for i, model in response_data['models']]
|
||||
return [f"{i + 1}: {model['name']}" for i, model in enumerate(response_data['models'])]
|
||||
|
||||
def get_type(self) -> str:
|
||||
return "local_ai"
|
||||
|
||||
Reference in New Issue
Block a user