flare / project_controller.py
ciyidogan's picture
Update project_controller.py
9508bbf verified
raw
history blame
3.26 kB
from fastapi import APIRouter, Request, HTTPException
from service_config import ServiceConfig
import datetime
router = APIRouter()
service_config = ServiceConfig()
service_config.load()
def get_utc_now():
return datetime.datetime.utcnow().isoformat()
@router.post("/add")
async def add_project(request: Request):
data = await request.json()
project_name = data.get("project_name")
if not project_name:
raise HTTPException(status_code=400, detail="project_name cannot be empty")
if project_name in service_config.projects:
raise HTTPException(status_code=400, detail="Project already exists")
service_config.projects[project_name] = {
"enabled": False,
"versions": [],
"last_updated": get_utc_now()
}
with open(service_config.config_path, "w", encoding="utf-8") as f:
import json
json.dump(service_config, f, indent=2)
return {"message": f"Project {project_name} added"}
@router.post("/update")
async def update_project(request: Request):
data = await request.json()
project_name = data.get("project_name")
client_last_updated = data.get("client_last_updated")
project = service_config.projects.get(project_name)
if not project:
raise HTTPException(status_code=404, detail="Project not found")
if project["last_updated"] != client_last_updated:
raise HTTPException(status_code=409, detail="This record has been updated by another user. Please reload.")
# Buraya update işlemleri eklenir (örneğin versiyon ekleme vb.)
# Örnek: project['enabled'] = data.get('enabled', project['enabled'])
project["last_updated"] = get_utc_now()
with open(service_config.config_path, "w", encoding="utf-8") as f:
import json
json.dump(service_config, f, indent=2)
return {"message": f"Project {project_name} updated"}
@router.post("/publish")
async def publish_project(request: Request):
data = await request.json()
project_name = data.get("project_name")
client_last_updated = data.get("client_last_updated")
project = service_config.projects.get(project_name)
if not project:
raise HTTPException(status_code=404, detail="Project not found")
if project["last_updated"] != client_last_updated:
raise HTTPException(status_code=409, detail="This record has been updated by another user. Please reload.")
latest_version = project["versions"][-1] if project["versions"] else None
if not latest_version:
raise HTTPException(status_code=400, detail="No version exists to publish")
llm = latest_version.get("llm", {})
if not llm.get("repo_id"):
raise HTTPException(status_code=400, detail="repo_id is required")
if llm.get("use_fine_tune") and not llm.get("fine_tune_zip"):
raise HTTPException(status_code=400, detail="fine_tune_zip is required when use_fine_tune is true")
latest_version["published"] = True
latest_version["last_updated"] = get_utc_now()
project["last_updated"] = get_utc_now()
with open(service_config.config_path, "w", encoding="utf-8") as f:
import json
json.dump(service_config, f, indent=2)
return {"message": f"Project {project_name} version published"}