AdmMabe/ai-chat
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: update OLLAMA_API_URL to correct endpoint and enable follow_redirects in HTTP client

Browse Source
This commit is contained in:
Mabe
2026-06-21 17:36:12 +02:00
parent 6332615d06
commit 2a0379b653
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/app/main.py
+2 -2
View File
@@ -11,7 +11,7 @@ from app.auth import get_discovery, settings, validate_token
from app.schemas import AuthCallbackRequest, AuthConfigResponse, ChatRequest from app.schemas import AuthCallbackRequest, AuthConfigResponse, ChatRequest
OLLAMA_API_KEY = os.getenv("OLLAMA_API_KEY") OLLAMA_API_KEY = os.getenv("OLLAMA_API_KEY")
OLLAMA_API_URL = os.getenv("OLLAMA_API_URL", "https://api.ollama.com/v1/chat/completions") OLLAMA_API_URL = os.getenv("OLLAMA_API_URL", "https://ollama.com/v1/chat/completions")
if not OLLAMA_API_KEY: if not OLLAMA_API_KEY:
raise RuntimeError("Environment variable OLLAMA_API_KEY is required") raise RuntimeError("Environment variable OLLAMA_API_KEY is required")
@@ -44,7 +44,7 @@ async def event_stream(request: ChatRequest) -> AsyncGenerator[str, None]:
"Accept": "text/event-stream", "Accept": "text/event-stream",
} }
async with httpx.AsyncClient(timeout=None) as client: async with httpx.AsyncClient(timeout=None, follow_redirects=True) as client:
async with client.stream("POST", OLLAMA_API_URL, json=payload, headers=headers) as response: async with client.stream("POST", OLLAMA_API_URL, json=payload, headers=headers) as response:
if response.status_code >= 400: if response.status_code >= 400:
text = await response.aread() text = await response.aread()