cf21ba4516
Phase 1 (Foundation): - Project restructure (presenton-main → backend/ + frontend/) - Database schema (8 new models, Alembic config, seed script) - Auth (Azure AD SSO + dev bypass, JWT sessions, AuthMiddleware) - RBAC (access_service, rbac_middleware, admin routers) - Audit logging (fire-and-forget, AuditMiddleware, admin router) - i18n (react-i18next with 5 namespace files) Phase 2 (Admin Panel & Client Management): - Admin panel shell (sidebar layout, role guard, 12 pages) - Redux admin slice with 18 async thunks - User management (role changes, deactivation) - Client management (CRUD, brand config, team management) - Brand config editor (colors, fonts, logos, voice rules) - Master deck upload & parser (PPTX → HTML → React pipeline) - Audit log viewer with filters and CSV/JSON export Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
60 lines
2.1 KiB
Python
60 lines
2.1 KiB
Python
import json
|
|
from typing import AsyncGenerator
|
|
import aiohttp
|
|
from fastapi import HTTPException
|
|
|
|
from models.ollama_model_status import OllamaModelStatus
|
|
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() or 'http://localhost:11434'}/api/pull",
|
|
json={"model": model},
|
|
) as response:
|
|
if response.status != 200:
|
|
raise HTTPException(
|
|
status_code=response.status,
|
|
detail=f"Failed to pull model: {await response.text()}",
|
|
)
|
|
|
|
async for line in response.content:
|
|
if not line.strip():
|
|
continue
|
|
|
|
try:
|
|
event = json.loads(line.decode("utf-8"))
|
|
except json.JSONDecodeError:
|
|
continue
|
|
|
|
yield event
|
|
|
|
|
|
async def list_pulled_ollama_models() -> list[OllamaModelStatus]:
|
|
async with aiohttp.ClientSession() as session:
|
|
async with session.get(
|
|
f"{get_ollama_url_env() or 'http://localhost:11434'}/api/tags",
|
|
) as response:
|
|
if response.status == 200:
|
|
pulled_models = await response.json()
|
|
return [
|
|
OllamaModelStatus(
|
|
name=m["model"],
|
|
size=m["size"],
|
|
status="pulled",
|
|
downloaded=m["size"],
|
|
done=True,
|
|
)
|
|
for m in pulled_models["models"]
|
|
]
|
|
elif response.status == 403:
|
|
raise HTTPException(
|
|
status_code=403,
|
|
detail="Forbidden: Please check your Ollama Configuration",
|
|
)
|
|
else:
|
|
raise HTTPException(
|
|
status_code=response.status,
|
|
detail=f"Failed to list Ollama models: {response.status}",
|
|
)
|