Oliver/presenton
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: default ollama url

Browse source
This commit is contained in:
sauravniraula 2025-11-28 02:56:33 +05:45
parent fb84032da4
commit c9b60d5f51
No known key found for this signature in database
GPG key ID: 60FCC1B5A5E83326
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
servers/fastapi/utils/ollama.py
View file

@ -10,7 +10,7 @@ from utils.get_env import get_ollama_url_env
async def pull_ollama_model(model: str) -> AsyncGenerator[dict, None]:
async with aiohttp.ClientSession() as session:
async with session.post(
f"{get_ollama_url_env()}/api/pull",
f"{get_ollama_url_env() or 'http://localhost:11434'}/api/pull",
json={"model": model},
) as response:
if response.status != 200:
@ -34,7 +34,7 @@ async def pull_ollama_model(model: str) -> AsyncGenerator[dict, None]:
async def list_pulled_ollama_models() -> list[OllamaModelStatus]:
async with aiohttp.ClientSession() as session:
async with session.get(
f"{get_ollama_url_env()}/api/tags",
f"{get_ollama_url_env() or 'http://localhost:11434'}/api/tags",
) as response:
if response.status == 200:
pulled_models = await response.json()