Spaces:
Building
Building
Update llm_startup.py
Browse files- llm_startup.py +11 -11
llm_startup.py
CHANGED
@@ -8,7 +8,7 @@ from __future__ import annotations
|
|
8 |
import threading
|
9 |
import asyncio
|
10 |
from typing import Dict, Any
|
11 |
-
from
|
12 |
from config_provider import ConfigProvider, ProjectConfig, VersionConfig
|
13 |
from llm_factory import LLMFactory
|
14 |
|
@@ -28,27 +28,27 @@ async def notify_startup_async():
|
|
28 |
)
|
29 |
|
30 |
if not llm_provider_def or not llm_provider_def.requires_repo_info:
|
31 |
-
|
32 |
return
|
33 |
|
34 |
# Create LLM provider instance
|
35 |
try:
|
36 |
llm_provider = LLMFactory.create_provider()
|
37 |
except Exception as e:
|
38 |
-
|
39 |
return
|
40 |
|
41 |
# Notify for each enabled project
|
42 |
enabled_projects = [p for p in cfg.projects if p.enabled and not getattr(p, 'deleted', False)]
|
43 |
|
44 |
if not enabled_projects:
|
45 |
-
|
46 |
return
|
47 |
|
48 |
for project in enabled_projects:
|
49 |
version = _select_live_version(project)
|
50 |
if not version:
|
51 |
-
|
52 |
continue
|
53 |
|
54 |
# Build project config - version.id yerine version.no kullan
|
@@ -62,16 +62,16 @@ async def notify_startup_async():
|
|
62 |
}
|
63 |
|
64 |
try:
|
65 |
-
|
66 |
success = await llm_provider.startup(project_config)
|
67 |
|
68 |
if success:
|
69 |
-
|
70 |
else:
|
71 |
-
|
72 |
|
73 |
except Exception as e:
|
74 |
-
|
75 |
|
76 |
def notify_startup():
|
77 |
"""Synchronous wrapper for async startup notification"""
|
@@ -95,8 +95,8 @@ def run_in_thread():
|
|
95 |
)
|
96 |
|
97 |
if not llm_provider_def or not llm_provider_def.requires_repo_info:
|
98 |
-
|
99 |
return
|
100 |
|
101 |
-
|
102 |
threading.Thread(target=notify_startup, daemon=True).start()
|
|
|
8 |
import threading
|
9 |
import asyncio
|
10 |
from typing import Dict, Any
|
11 |
+
from logger import log_info, log_error, log_warning, log_debug
|
12 |
from config_provider import ConfigProvider, ProjectConfig, VersionConfig
|
13 |
from llm_factory import LLMFactory
|
14 |
|
|
|
28 |
)
|
29 |
|
30 |
if not llm_provider_def or not llm_provider_def.requires_repo_info:
|
31 |
+
log_info(f"βΉοΈ LLM provider '{cfg.global_config.llm_provider.name}' does not require startup notification")
|
32 |
return
|
33 |
|
34 |
# Create LLM provider instance
|
35 |
try:
|
36 |
llm_provider = LLMFactory.create_provider()
|
37 |
except Exception as e:
|
38 |
+
log_error("β Failed to create LLM provider for startup", e)
|
39 |
return
|
40 |
|
41 |
# Notify for each enabled project
|
42 |
enabled_projects = [p for p in cfg.projects if p.enabled and not getattr(p, 'deleted', False)]
|
43 |
|
44 |
if not enabled_projects:
|
45 |
+
log_info("βΉοΈ No enabled projects found for startup notification")
|
46 |
return
|
47 |
|
48 |
for project in enabled_projects:
|
49 |
version = _select_live_version(project)
|
50 |
if not version:
|
51 |
+
log_info(f"β οΈ No published version found for project '{project.name}', skipping startup")
|
52 |
continue
|
53 |
|
54 |
# Build project config - version.id yerine version.no kullan
|
|
|
62 |
}
|
63 |
|
64 |
try:
|
65 |
+
log_info(f"π Notifying LLM provider startup for project '{project.name}'...")
|
66 |
success = await llm_provider.startup(project_config)
|
67 |
|
68 |
if success:
|
69 |
+
log_info(f"β
LLM provider acknowledged startup for '{project.name}'")
|
70 |
else:
|
71 |
+
log_info(f"β οΈ LLM provider startup failed for '{project.name}'")
|
72 |
|
73 |
except Exception as e:
|
74 |
+
log_error(f"β Error during startup notification for '{project.name}'", e)
|
75 |
|
76 |
def notify_startup():
|
77 |
"""Synchronous wrapper for async startup notification"""
|
|
|
95 |
)
|
96 |
|
97 |
if not llm_provider_def or not llm_provider_def.requires_repo_info:
|
98 |
+
log_info(f"π€ {cfg.global_config.llm_provider.name} - Startup notification not required")
|
99 |
return
|
100 |
|
101 |
+
log_info("π Starting LLM provider startup notification thread...")
|
102 |
threading.Thread(target=notify_startup, daemon=True).start()
|