Spaces:
Building
Building
Update spark_startup.py
Browse files- spark_startup.py +14 -6
spark_startup.py
CHANGED
@@ -5,7 +5,7 @@ Projeler açılırken Spark’a /startup çağrısı yapar.
|
|
5 |
"""
|
6 |
|
7 |
from __future__ import annotations
|
8 |
-
import threading,
|
9 |
from typing import Dict, Any
|
10 |
from utils import log
|
11 |
from config_provider import ConfigProvider, ProjectConfig, VersionConfig
|
@@ -14,12 +14,14 @@ cfg = ConfigProvider.get()
|
|
14 |
|
15 |
|
16 |
def _select_live_version(p: ProjectConfig) -> VersionConfig | None:
|
|
|
17 |
published = [v for v in p.versions if v.published]
|
18 |
return max(published, key=lambda v: v.id) if published else None
|
19 |
|
20 |
|
21 |
def notify_startup():
|
22 |
spark_base = str(cfg.global_config.spark_endpoint).rstrip("/")
|
|
|
23 |
for p in cfg.projects:
|
24 |
if not p.enabled:
|
25 |
continue
|
@@ -28,17 +30,23 @@ def notify_startup():
|
|
28 |
continue
|
29 |
|
30 |
body: Dict[str, Any] = {
|
31 |
-
|
32 |
-
"
|
33 |
-
"general_prompt": v.general_prompt,
|
34 |
"cloud_token": cfg.global_config.get_plain_token(),
|
35 |
-
"
|
|
|
|
|
|
|
|
|
|
|
36 |
}
|
37 |
|
38 |
try:
|
39 |
log(f"🚀 Notifying Spark /startup for project '{p.name}' …")
|
40 |
r = requests.post(f"{spark_base}/startup", json=body, timeout=10)
|
41 |
-
r.
|
|
|
|
|
42 |
log(f"✅ Spark acknowledged startup ({r.status_code})")
|
43 |
except Exception as e:
|
44 |
log(f"⚠️ Spark startup failed: {e}")
|
|
|
5 |
"""
|
6 |
|
7 |
from __future__ import annotations
|
8 |
+
import threading, requests
|
9 |
from typing import Dict, Any
|
10 |
from utils import log
|
11 |
from config_provider import ConfigProvider, ProjectConfig, VersionConfig
|
|
|
14 |
|
15 |
|
16 |
def _select_live_version(p: ProjectConfig) -> VersionConfig | None:
|
17 |
+
"""Yayınlanmış en güncel versiyonu getir."""
|
18 |
published = [v for v in p.versions if v.published]
|
19 |
return max(published, key=lambda v: v.id) if published else None
|
20 |
|
21 |
|
22 |
def notify_startup():
|
23 |
spark_base = str(cfg.global_config.spark_endpoint).rstrip("/")
|
24 |
+
|
25 |
for p in cfg.projects:
|
26 |
if not p.enabled:
|
27 |
continue
|
|
|
30 |
continue
|
31 |
|
32 |
body: Dict[str, Any] = {
|
33 |
+
# ----- zorunlu alanlar -----
|
34 |
+
"work_mode": cfg.global_config.work_mode,
|
|
|
35 |
"cloud_token": cfg.global_config.get_plain_token(),
|
36 |
+
"project_name": p.name,
|
37 |
+
"project_version": v.id,
|
38 |
+
"repo_id": v.llm.repo_id,
|
39 |
+
"generation_config": v.llm.generation_config,
|
40 |
+
"use_fine_tune": v.llm.use_fine_tune,
|
41 |
+
"fine_tune_zip": v.llm.fine_tune_zip,
|
42 |
}
|
43 |
|
44 |
try:
|
45 |
log(f"🚀 Notifying Spark /startup for project '{p.name}' …")
|
46 |
r = requests.post(f"{spark_base}/startup", json=body, timeout=10)
|
47 |
+
if r.status_code >= 400:
|
48 |
+
log(f"❌ Spark responded {r.status_code}: {r.text}")
|
49 |
+
r.raise_for_status()
|
50 |
log(f"✅ Spark acknowledged startup ({r.status_code})")
|
51 |
except Exception as e:
|
52 |
log(f"⚠️ Spark startup failed: {e}")
|