ciyidogan commited on
Commit
0831fe7
·
verified ·
1 Parent(s): 94f65ce

Update spark_startup.py

Browse files
Files changed (1) hide show
  1. 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, time, requests
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
- "project_name": p.name,
32
- "version_id": v.id,
33
- "general_prompt": v.general_prompt,
34
  "cloud_token": cfg.global_config.get_plain_token(),
35
- "llm": v.llm.model_dump(),
 
 
 
 
 
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.raise_for_status()
 
 
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}")