ciyidogan commited on
Commit
df604d9
·
verified ·
1 Parent(s): 10618d8

Update spark_startup.py

Browse files
Files changed (1) hide show
  1. spark_startup.py +12 -1
spark_startup.py CHANGED
@@ -44,9 +44,20 @@ def notify_startup():
44
  "fine_tune_zip": v.llm.fine_tune_zip,
45
  }
46
 
 
 
 
 
 
 
 
 
 
 
 
47
  try:
48
  log(f"🚀 Notifying Spark /startup for project '{p.name}' …")
49
- r = requests.post(f"{spark_base}/startup", json=body, timeout=10)
50
  if r.status_code >= 400:
51
  log(f"❌ Spark responded {r.status_code}: {r.text}")
52
  r.raise_for_status()
 
44
  "fine_tune_zip": v.llm.fine_tune_zip,
45
  }
46
 
47
+ # Hugging Face token'ı ortam değişkeninden al
48
+ spark_token = os.getenv("SPARK_TOKEN")
49
+ if not spark_token:
50
+ log("❌ SPARK_TOKEN environment variable is missing.")
51
+ return
52
+
53
+ headers = {
54
+ "Authorization": f"Bearer {spark_token}",
55
+ "Content-Type": "application/json"
56
+ }
57
+
58
  try:
59
  log(f"🚀 Notifying Spark /startup for project '{p.name}' …")
60
+ r = requests.post(f"{spark_base}/startup", json=body, headers=headers, timeout=10)
61
  if r.status_code >= 400:
62
  log(f"❌ Spark responded {r.status_code}: {r.text}")
63
  r.raise_for_status()