ciyidogan commited on
Commit
eb65f56
Β·
verified Β·
1 Parent(s): db9dc05

Update llm_startup.py

Browse files
Files changed (1) hide show
  1. 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 utils import log
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
- log(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(f"❌ 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("ℹ️ 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(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,16 +62,16 @@ async def notify_startup_async():
62
  }
63
 
64
  try:
65
- log(f"πŸš€ Notifying LLM provider startup for project '{project.name}'...")
66
  success = await llm_provider.startup(project_config)
67
 
68
  if success:
69
- log(f"βœ… LLM provider acknowledged startup for '{project.name}'")
70
  else:
71
- log(f"⚠️ LLM provider startup failed for '{project.name}'")
72
 
73
  except Exception as e:
74
- log(f"❌ Error during startup notification for '{project.name}': {e}")
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
- log(f"πŸ€– {cfg.global_config.llm_provider.name} - Startup notification not required")
99
  return
100
 
101
- log("πŸš€ Starting LLM provider startup notification thread...")
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()