flare / project_controller.py
ciyidogan's picture
Update project_controller.py
f04c960 verified
raw
history blame
6.78 kB
from fastapi import APIRouter, Request, HTTPException
from service_config import ServiceConfig
import datetime
import json
router = APIRouter()
service_config = ServiceConfig()
service_config.load()
def get_utc_now():
return datetime.datetime.utcnow().isoformat()
@router.get("/list")
def list_projects():
return service_config.projects
@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:
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.")
# örnek update: sadece enable/disable flag güncelliyoruz
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:
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:
json.dump(service_config, f, indent=2)
return {"message": f"Project {project_name} version published"}
@router.post("/add_intent")
async def add_intent(request: Request):
data = await request.json()
project_name = data.get("project_name")
version_number = data.get("version_number")
intent = data.get("intent")
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.")
version = next((v for v in project["versions"] if v["version_number"] == version_number), None)
if not version:
raise HTTPException(status_code=404, detail="Version not found")
if any(i["name"] == intent["name"] for i in version.get("intents", [])):
raise HTTPException(status_code=400, detail="Intent already exists")
version.setdefault("intents", []).append(intent)
version["last_updated"] = get_utc_now()
project["last_updated"] = get_utc_now()
with open(service_config.config_path, "w", encoding="utf-8") as f:
json.dump(service_config, f, indent=2)
return {"message": f"Intent {intent['name']} added to project {project_name}, version {version_number}"}
@router.post("/delete_intent")
async def delete_intent(request: Request):
data = await request.json()
project_name = data.get("project_name")
version_number = data.get("version_number")
intent_name = data.get("intent_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.")
version = next((v for v in project["versions"] if v["version_number"] == version_number), None)
if not version:
raise HTTPException(status_code=404, detail="Version not found")
version["intents"] = [i for i in version.get("intents", []) if i["name"] != intent_name]
version["last_updated"] = get_utc_now()
project["last_updated"] = get_utc_now()
with open(service_config.config_path, "w", encoding="utf-8") as f:
json.dump(service_config, f, indent=2)
return {"message": f"Intent {intent_name} deleted from project {project_name}, version {version_number}"}
@router.post("/add_api")
async def add_api(request: Request):
data = await request.json()
api_name = data.get("api_name")
api_def = data.get("api_def")
if api_name in service_config.apis:
raise HTTPException(status_code=400, detail="API already exists")
service_config.apis[api_name] = api_def
with open(service_config.config_path, "w", encoding="utf-8") as f:
json.dump(service_config, f, indent=2)
return {"message": f"API {api_name} added"}
@router.post("/delete_api")
async def delete_api(request: Request):
data = await request.json()
api_name = data.get("api_name")
if api_name not in service_config.apis:
raise HTTPException(status_code=404, detail="API not found")
del service_config.apis[api_name]
with open(service_config.config_path, "w", encoding="utf-8") as f:
json.dump(service_config, f, indent=2)
return {"message": f"API {api_name} deleted"}