yumoqing/llmage
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: bind hot_reload event via EventDispatcher

Browse Source
This commit is contained in:
yumoqing 2026-06-01 18:10:32 +08:00
parent bab415ba83
commit 87040915ee
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
llmage/init.py
View File

@ -20,6 +20,11 @@ from .utils import (
invalidate_uapi_cache,
)
def _on_hot_reload(data=None):
"""Event handler for hot_reload — wraps invalidate_uapi_cache to accept dispatcher's data arg."""
invalidate_uapi_cache()
from .llmclient import (
inference_generator,
inference
@ -69,6 +74,9 @@ def load_llmage():
env.backup_accounted_llmusage = backup_accounted_llmusage
env.get_failed_accounting_records = get_failed_accounting_records
env.get_llmage_stats = get_llmage_stats
# Bind hot_reload event — module-level function, ref safe (module keeps it alive)
if hasattr(env, 'event_dispatcher'):
env.event_dispatcher.bind('hot_reload', _on_hot_reload)
rf = RegisterFunction()
rf.register('jimeng_auth_headers', jimeng_auth_headers)