Spaces:
Runtime error
Runtime error
File size: 626 Bytes
83f7ed7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
import requests
import json
# 提供api获取ollama 模型列表
def get_llm():
respone = requests.get(url="http://localhost:11434/api/tags")
result = json.loads(respone.content)
llms = []
for llm in result["models"]:
if "code" not in llm["name"] and "embed" not in llm["name"]:
llms.append(llm["name"])
return llms
def get_embeding_model():
respone = requests.get(url="http://localhost:11434/api/tags")
result = json.loads(respone.content)
llms = []
for llm in result["models"]:
if "embed" in llm["name"]:
llms.append(llm["name"])
return llms
|