Spaces:
Paused
Paused
from fastapi import APIRouter, Request | |
from core import service_config, llm_models | |
from llm_model import LLMModel | |
from log import log | |
import os, shutil, threading | |
router = APIRouter() | |
async def reload_config(request: Request): | |
body = await request.json() | |
project_name = body.get("project_name") | |
new_config_data = body.get("service_config") | |
if not project_name or not new_config_data: | |
return {"error": "project_name ve service_config gereklidir."} | |
def background_reload(): | |
try: | |
current_project = service_config.projects.get(project_name) | |
incoming_project = new_config_data.get("projects", {}).get(project_name) | |
if not incoming_project: | |
log(f"❌ '{project_name}' yeni config içinde bulunamadı, işlem durduruldu.") | |
return | |
project_path = f"/data/projects/{project_name}" | |
temp_path = f"/data/projects/{project_name}_temp" | |
if os.path.exists(temp_path): | |
shutil.rmtree(temp_path) | |
os.makedirs(temp_path, exist_ok=True) | |
llm_config = incoming_project["llm"] | |
temp_instance = LLMModel() | |
if current_project is None: | |
log(f"🆕 Yeni proje '{project_name}' tespit edildi, yükleme başlatılıyor...") | |
temp_instance.setup(service_config, llm_config, temp_path) | |
if os.path.exists(project_path): | |
shutil.rmtree(project_path) | |
shutil.copytree(temp_path, project_path) | |
llm_models[project_name] = temp_instance | |
service_config.projects[project_name] = incoming_project | |
log(f"✅ Yeni proje '{project_name}' başarıyla yüklendi ve belleğe alındı.") | |
return | |
if current_project == incoming_project: | |
log(f"ℹ️ '{project_name}' için değişiklik bulunamadı, işlem atlandı.") | |
return | |
log(f"🔄 '{project_name}' güncellemesi tespit edildi, güncelleme başlatılıyor...") | |
if current_project["llm"]["model_base"] != llm_config["model_base"]: | |
temp_instance.setup(service_config, llm_config, temp_path) | |
else: | |
temp_instance.model = llm_models[project_name].model | |
temp_instance.tokenizer = llm_models[project_name].tokenizer | |
if os.path.exists(project_path): | |
shutil.rmtree(project_path) | |
shutil.copytree(temp_path, project_path) | |
llm_models[project_name] = temp_instance | |
service_config.projects[project_name] = incoming_project | |
log(f"✅ '{project_name}' güncellemesi tamamlandı ve belleğe alındı.") | |
except Exception as e: | |
log(f"❌ reload_config background hatası: {e}") | |
threading.Thread(target=background_reload, daemon=True).start() | |
return { | |
"status": "accepted", | |
"message": f"'{project_name}' için güncelleme arka planda başlatıldı. İşlem loglardan takip edilebilir." | |
} | |