-
Notifications
You must be signed in to change notification settings - Fork 10.3k
Expand file tree
/
Copy pathapp.py
More file actions
781 lines (675 loc) · 31.7 KB
/
Copy pathapp.py
File metadata and controls
781 lines (675 loc) · 31.7 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
# encoding:utf-8
import logging
import os
import signal
import sys
import time
from channel import channel_factory
from common import const
from common.log import logger
from common.ssl_certs import ensure_ca_bundle
from config import load_config, conf
from plugins import *
import threading
_channel_mgr = None
# Desktop mode: a lighter runtime for the packaged Electron client. Plugins are
# loaded in a background thread (so command plugins like cow_cli/godcmd work
# without slowing startup), while MCP warmup is still skipped to keep it fast.
DESKTOP_MODE = os.environ.get("COW_DESKTOP") == "1"
def get_channel_manager():
return _channel_mgr
def _parse_channel_type(raw) -> list:
"""
Parse channel_type config value into a list of channel names.
Supports:
- single string: "feishu"
- comma-separated string: "feishu, dingtalk"
- list: ["feishu", "dingtalk"]
"""
if isinstance(raw, list):
return [ch.strip() for ch in raw if ch.strip()]
if isinstance(raw, str):
return [ch.strip() for ch in raw.split(",") if ch.strip()]
return []
def _has_web_entry(channel_names: list) -> bool:
"""True if the web console is already in the startup list (string or instance)."""
from channel.channel_instances import ChannelInstance
for entry in channel_names:
if isinstance(entry, ChannelInstance):
if entry.channel_type == "web":
return True
elif entry == "web":
return True
return False
def _resolve_startup_channels(raw_channel):
"""Startup channel list = config.json's channels plus team.json's instances,
de-duplicated by channel *type* for multi-instance-ready types.
config.json stays the source of truth for anything it configures: its
``channel_type`` list (dingtalk, wecom, ...) always starts, exactly as a
legacy install expects. team.json's ``channel_instances`` carries the
explicit multi-instance bots (feishu today), each with its own credentials
and Agent binding.
The subtlety is feishu: config.json's flat ``feishu`` entry and a
``channel_instances`` feishu record are the *same kind of connection*. Once
team.json manages feishu (at least one feishu instance exists), feishu's
single source of truth is ``channel_instances`` — the flat config entry is
dropped so the same bot is not started twice on one websocket. This holds no
matter which Agent the instances are bound to: rebinding every feishu
instance away from the default Agent must NOT resurrect config.json's feishu
as a stray default-bound bot. Non-multi-instance types are never managed by
instances, so config keeps starting them untouched.
A legacy single-Agent install has no team.json / no channel_instances and
this returns exactly ``_parse_channel_type(raw_channel)`` as before.
"""
from channel.channel_instances import MULTI_INSTANCE_READY, _normalize_type
names = _parse_channel_type(raw_channel)
instances = []
try:
from agent import team
from channel.channel_instances import resolve_channel_instances
settings = team.resolve(conf())
raw_instances = settings.get("channel_instances")
if isinstance(raw_instances, list) and raw_instances:
instances = resolve_channel_instances(settings)
except Exception as e:
logger.warning(
f"[App] Failed to resolve channel_instances, using config.json "
f"channel_type only: {e}"
)
instances = []
# Types now owned by channel_instances (feishu, ...). Drop config.json's
# flat entry for these so the instance records are the only source.
managed_types = {
inst.channel_type
for inst in instances
if inst.channel_type in MULTI_INSTANCE_READY
}
entries = []
for name in names:
if _normalize_type(name) in managed_types:
logger.info(
f"[App] channel_type '{name}' is managed by channel_instances; "
f"skipping the flat config.json entry to avoid a duplicate bot"
)
continue
entries.append(name)
if instances:
logger.info(
f"[App] Starting channel_instances: "
f"{[(i.instance_id, i.channel_type, i.agent_id) for i in instances]}"
)
entries.extend(instances)
if not entries:
entries = ["web"]
return entries
class ChannelManager:
"""
Manage the lifecycle of multiple channels running concurrently.
Each channel.startup() runs in its own daemon thread.
The web channel is started as default console unless explicitly disabled.
"""
def __init__(self):
self._channels = {} # channel_name -> channel instance
self._threads = {} # channel_name -> thread
self._primary_channel = None
self._lock = threading.Lock()
self.cloud_mode = False # set to True when cloud client is active
@property
def channel(self):
"""Return the primary (first non-web) channel for backward compatibility."""
return self._primary_channel
def get_channel(self, channel_name: str):
return self._channels.get(channel_name)
@staticmethod
def _normalize_entry(entry):
"""Accept both a legacy channel-type string and a ChannelInstance.
Returns (name, channel_type, factory_kwargs). For a plain string this
reproduces the old behavior exactly: name == channel_type and no
per-instance overrides. For a ChannelInstance, the registry key is the
instance_id, and the factory receives credentials + binding so several
instances of one type can coexist.
"""
from channel.channel_instances import ChannelInstance
if isinstance(entry, ChannelInstance):
return (
entry.instance_id,
entry.channel_type,
{
"instance_id": entry.instance_id,
"bound_agent_id": entry.agent_id,
"credentials": entry.credentials or None,
"members": entry.members or None,
},
)
return (entry, entry, {})
def start(self, channel_names: list, first_start: bool = False):
"""
Create and start one or more channels in sub-threads.
If first_start is True, plugins and linkai client will also be initialized.
Each entry may be a legacy channel-type string or a ChannelInstance.
"""
entries = [self._normalize_entry(e) for e in channel_names]
# A concurrent path may have started this channel already (saving its
# config restarts it, connecting it starts it). Overwriting the registry
# entry below would orphan that instance: nothing holds it any more, yet
# its connection stays up and keeps consuming events, so every inbound
# message gets handled twice.
for name, _ctype, _kw in entries:
if self._channels.get(name) is not None:
logger.warning(f"[ChannelManager] Channel '{name}' is already running, stopping it first")
self.stop(name)
with self._lock:
channels = []
for name, channel_type, factory_kwargs in entries:
# One misconfigured channel (e.g. wechatcom_app without its
# corp_id/token/aes_key) must not take the whole process down:
# instantiating it can raise while parsing config. The web
# console in particular has to come up so the desktop shell can
# surface the error and let the user fix the config. Skip the
# broken channel and keep the rest.
try:
ch = channel_factory.create_channel(channel_type, **factory_kwargs)
except Exception as e:
logger.error(f"[ChannelManager] Failed to create channel '{name}', skipping it: {e}")
logger.exception(e)
continue
ch.cloud_mode = self.cloud_mode
self._channels[name] = ch
channels.append((name, ch))
if self._primary_channel is None and name != "web":
self._primary_channel = ch
if self._primary_channel is None and channels:
self._primary_channel = channels[0][1]
if first_start:
if DESKTOP_MODE:
# Load plugins in the background so command plugins
# (cow_cli / godcmd, e.g. /status, #help) work in the
# desktop client, without blocking web-service readiness.
threading.Thread(
target=PluginManager().load_plugins, daemon=True
).start()
else:
PluginManager().load_plugins()
# Cloud client is optional. It is only started when
# use_linkai=True AND cloud_deployment_id is set.
# By default neither is configured, so the app runs
# entirely locally without any remote connection.
if conf().get("use_linkai") and (
os.environ.get("CLOUD_DEPLOYMENT_ID") or conf().get("cloud_deployment_id")
):
try:
from common import cloud_client
threading.Thread(
target=cloud_client.start,
args=(self._primary_channel, self),
daemon=True,
).start()
except Exception:
pass
# Start web console first so its logs print cleanly,
# then start remaining channels after a brief pause.
web_entry = None
other_entries = []
for entry in channels:
if entry[0] == "web":
web_entry = entry
else:
other_entries.append(entry)
ordered = ([web_entry] if web_entry else []) + other_entries
for i, (name, ch) in enumerate(ordered):
if i > 0 and name != "web":
time.sleep(0.1)
t = threading.Thread(target=self._run_channel, args=(name, ch), daemon=True)
self._threads[name] = t
t.start()
logger.debug(f"[ChannelManager] Channel '{name}' started in sub-thread")
def _run_channel(self, name: str, channel):
try:
channel.startup()
except Exception as e:
logger.error(f"[ChannelManager] Channel '{name}' startup error: {e}")
logger.exception(e)
# The desktop client IS the web channel: without it the Electron
# shell polls a health endpoint that will never answer and, 90s
# later, blames a generic "initialization failed". Exiting non-zero
# lets the shell surface the real error immediately. Server
# deployments keep the old behavior - other channels may still be
# serving, so one broken channel must not take the process down.
if DESKTOP_MODE and name == "web":
logging.shutdown()
os._exit(1)
def stop(self, channel_name: str = None):
"""
Stop channel(s). If channel_name is given, stop only that channel;
otherwise stop all channels.
"""
# Pop under lock, then stop outside lock to avoid deadlock
with self._lock:
names = [channel_name] if channel_name else list(self._channels.keys())
to_stop = []
for name in names:
ch = self._channels.pop(name, None)
th = self._threads.pop(name, None)
to_stop.append((name, ch, th))
if channel_name and self._primary_channel is self._channels.get(channel_name):
self._primary_channel = None
for name, ch, th in to_stop:
if ch is None:
logger.warning(f"[ChannelManager] Channel '{name}' not found in managed channels")
if th and th.is_alive():
self._interrupt_thread(th, name)
continue
logger.info(f"[ChannelManager] Stopping channel '{name}'...")
graceful = False
if hasattr(ch, 'stop'):
try:
ch.stop()
graceful = True
except Exception as e:
logger.warning(f"[ChannelManager] Error during channel '{name}' stop: {e}")
if th and th.is_alive():
th.join(timeout=5)
if th.is_alive():
if graceful:
logger.info(f"[ChannelManager] Channel '{name}' thread still alive after stop(), "
"leaving daemon thread to finish on its own")
else:
logger.warning(f"[ChannelManager] Channel '{name}' thread did not exit in 5s, forcing interrupt")
self._interrupt_thread(th, name)
@staticmethod
def _interrupt_thread(th: threading.Thread, name: str):
"""Raise SystemExit in target thread to break blocking loops like start_forever."""
import ctypes
try:
tid = th.ident
if tid is None:
return
res = ctypes.pythonapi.PyThreadState_SetAsyncExc(
ctypes.c_ulong(tid), ctypes.py_object(SystemExit)
)
if res == 1:
logger.info(f"[ChannelManager] Interrupted thread for channel '{name}'")
elif res > 1:
ctypes.pythonapi.PyThreadState_SetAsyncExc(ctypes.c_ulong(tid), None)
logger.warning(f"[ChannelManager] Failed to interrupt thread for channel '{name}'")
except Exception as e:
logger.warning(f"[ChannelManager] Thread interrupt error for '{name}': {e}")
def restart(self, new_channel):
"""
Restart a single channel.
Can be called from any thread (e.g. remote config callback).
Accepts a channel-type string or a ChannelInstance. When a bare string
names a known explicit instance, its record (binding + credentials) is
looked up so the restart keeps its identity instead of falling back to
the legacy global-config path.
"""
from channel.channel_instances import ChannelInstance
entry = new_channel
if not isinstance(entry, ChannelInstance):
entry = self._resolve_instance_entry(new_channel) or new_channel
name = entry.instance_id if isinstance(entry, ChannelInstance) else entry
logger.info(f"[ChannelManager] Restarting channel '{name}'...")
self.stop(name)
_clear_singleton_cache(name)
time.sleep(1)
self.start([entry], first_start=False)
logger.info(f"[ChannelManager] Channel '{name}' restarted successfully")
@staticmethod
def _resolve_instance_entry(name: str):
"""Return the ChannelInstance stored for *name*, or None.
Lets a restart/add triggered with a bare id recover the instance's
binding and credentials from the roster file. Absent (legacy installs),
returns None and the caller keeps the plain-string behavior.
"""
try:
from config import conf
from channel.channel_instances import get_instance
return get_instance(conf(), name)
except Exception:
return None
def add_channel(self, channel):
"""
Dynamically add and start a new channel.
If the channel is already running, restart it instead.
``channel`` may be a legacy channel-type string (single-instance,
credentials read from global config) or a ChannelInstance carrying its
own id, binding and credentials (one of several instances of a type).
"""
from channel.channel_instances import ChannelInstance
channel_name = (
channel.instance_id if isinstance(channel, ChannelInstance) else channel
)
with self._lock:
if channel_name in self._channels:
logger.info(f"[ChannelManager] Channel '{channel_name}' already exists, restarting")
if self._channels.get(channel_name):
self.restart(channel_name)
return
logger.info(f"[ChannelManager] Adding channel '{channel_name}'...")
_clear_singleton_cache(channel_name)
self.start([channel], first_start=False)
logger.info(f"[ChannelManager] Channel '{channel_name}' added successfully")
def remove_channel(self, channel_name: str):
"""
Dynamically stop and remove a running channel.
"""
with self._lock:
if channel_name not in self._channels:
logger.warning(f"[ChannelManager] Channel '{channel_name}' not found, nothing to remove")
return
logger.info(f"[ChannelManager] Removing channel '{channel_name}'...")
self.stop(channel_name)
logger.info(f"[ChannelManager] Channel '{channel_name}' removed successfully")
def _clear_singleton_cache(channel_name: str):
"""
Clear the singleton cache for the channel class so that
a new instance can be created with updated config.
"""
cls_map = {
"web": "channel.web.web_channel.WebChannel",
"wechatmp": "channel.wechatmp.wechatmp_channel.WechatMPChannel",
"wechatmp_service": "channel.wechatmp.wechatmp_channel.WechatMPChannel",
"wechatcom_app": "channel.wechatcom.wechatcomapp_channel.WechatComAppChannel",
const.WECHAT_KF: "channel.wechat_kf.wechat_kf_channel.WechatKfChannel",
const.FEISHU: "channel.feishu.feishu_channel.FeiShuChanel",
const.DINGTALK: "channel.dingtalk.dingtalk_channel.DingTalkChanel",
const.WECOM_BOT: "channel.wecom_bot.wecom_bot_channel.WecomBotChannel",
const.QQ: "channel.qq.qq_channel.QQChannel",
const.TELEGRAM: "channel.telegram.telegram_channel.TelegramChannel",
const.SLACK: "channel.slack.slack_channel.SlackChannel",
const.DISCORD: "channel.discord.discord_channel.DiscordChannel",
const.WEIXIN: "channel.weixin.weixin_channel.WeixinChannel",
"wx": "channel.weixin.weixin_channel.WeixinChannel",
}
module_path = cls_map.get(channel_name)
if not module_path:
return
try:
parts = module_path.rsplit(".", 1)
module_name, class_name = parts[0], parts[1]
import importlib
module = importlib.import_module(module_name)
wrapper = getattr(module, class_name, None)
if wrapper and hasattr(wrapper, '__closure__') and wrapper.__closure__:
for cell in wrapper.__closure__:
try:
cell_contents = cell.cell_contents
if isinstance(cell_contents, dict):
cell_contents.clear()
logger.debug(f"[ChannelManager] Cleared singleton cache for {class_name}")
break
except ValueError:
pass
except Exception as e:
logger.warning(f"[ChannelManager] Failed to clear singleton cache: {e}")
def sigterm_handler_wrap(_signo):
old_handler = signal.getsignal(_signo)
def func(_signo, _stack_frame):
logger.info("signal {} received, exiting...".format(_signo))
conf().save_user_datas()
if callable(old_handler): # check old_handler
return old_handler(_signo, _stack_frame)
sys.exit(0)
signal.signal(_signo, func)
def _warmup_mcp_tools():
"""
Kick off MCP server loading at process startup so subprocesses
(npx / uvx etc.) finish initializing before the first user message
arrives. Returns immediately — the actual work happens on a daemon
thread inside ToolManager. Safe to call when MCP is not configured.
Warms every enabled Agent: this runs before any routing has happened, so
without the loop only the default Agent's servers would be ready and the
rest would boot on their first message instead.
"""
try:
from agent.registry import get_agent_registry
from agent.tools import ToolManager
from common.runtime_identity import identity_scope
profiles = get_agent_registry().list(include_disabled=False)
except Exception as e:
logger.warning(f"[App] MCP warmup failed (non-fatal): {e}")
return
for profile in profiles:
# Per Agent, so one broken mcp.json does not stop the others warming.
try:
with identity_scope(agent_id=profile.id):
ToolManager()._load_mcp_tools()
except Exception as e:
logger.warning(f"[App] MCP warmup failed for '{profile.id}' (non-fatal): {e}")
def _preload_heavy_imports():
"""Resolve the scheduler's import graph on the main thread.
Python locks imports per module, so two threads walking overlapping graphs
in opposite order deadlock outright: the scheduler warmup pulls
agent.tools -> requests -> urllib3 while channel creation pulls
web_channel -> web -> http.client -> email, and the graphs meet. Desktop
mode warms up on a background thread, so its modules must already be in
sys.modules before that thread exists - afterwards it only builds objects.
"""
try:
from bridge.bridge import Bridge # noqa: F401
except Exception as e:
logger.warning(f"[App] Import preload failed (non-fatal): {e}")
WEB_STARTUP_TIMEOUT = 25
def _start_web_watchdog(timeout: int = WEB_STARTUP_TIMEOUT):
"""Exit if the web console hasn't bound within ``timeout`` seconds.
A crash in channel startup already exits, but a *hang* used to leave the
process alive forever: the Electron shell waited out its own timeout,
blamed a generic "initialization failed", and the wedged backend stayed
resident - one more of them per launch attempt. Dumping every thread's
stack turns the next such hang into a diagnosable log instead of a guess.
"""
# Resolved here, on the main thread: a background thread must not be the
# one to import these (see _preload_heavy_imports).
import faulthandler
from channel.web.web_channel import SERVING
def _watch():
if SERVING.wait(timeout):
return
logger.error(
f"[App] Web console did not start within {timeout}s, exiting. "
"Thread stacks follow:"
)
try:
faulthandler.dump_traceback()
except Exception:
pass
logging.shutdown()
os._exit(1)
threading.Thread(target=_watch, daemon=True).start()
def _warmup_scheduler():
"""Eager-init AgentBridge so the scheduler thread starts at process
boot rather than waiting for the first user message."""
try:
from bridge.bridge import Bridge
Bridge().get_agent_bridge()
except Exception as e:
logger.warning(f"[App] Scheduler warmup failed: {e}")
def _migrate_team_roster():
"""Move a roster still stored in config.json into the team file.
Here rather than on the next edit from the console: the gateway is the one
place this runs single-threaded and before anything has read the registry,
and until it has moved the roster is still open to being clobbered by any
of the callers that rewrite config.json wholesale.
"""
try:
import os
from agent import team
from config import conf, get_data_root
team.migrate(conf(), os.path.join(get_data_root(), "config.json"))
except Exception as e:
# Never a reason not to start: read() still falls back to config.json.
logger.warning(f"[App] Could not move the roster into its own file: {e}")
def _warn_if_legacy_workspace_data_exists():
"""
Warn if the hardcoded ~/cow default holds data that agent_workspace
doesn't - e.g. after changing agent_workspace without moving the old
directory's contents over. The new workspace would otherwise look
empty even though old data still exists, with no indication why.
"""
try:
from common.state_dir import state_root_str
from common.utils import expand_path
workspace_root = state_root_str()
legacy_root = expand_path("~/cow")
# samefile checks filesystem identity, so case-insensitive filesystems
# (default on Windows and macOS) are handled correctly - normcase
# alone isn't enough, since it only folds case on Windows. Falls back
# when either path doesn't exist yet (samefile requires both to).
try:
same = os.path.samefile(legacy_root, workspace_root)
except OSError:
same = os.path.normcase(os.path.realpath(legacy_root)) == os.path.normcase(os.path.realpath(workspace_root))
if same:
return
# Any visible entry counts - covers session/skills/memory alike. Hidden
# entries are ignored so OS noise (.DS_Store) can't warn on every boot.
leftovers = os.listdir(legacy_root) if os.path.isdir(legacy_root) else []
if any(not name.startswith(".") for name in leftovers):
logger.warning(
f"[App] Found existing data at the default workspace ({legacy_root}) "
f"that doesn't match your configured agent_workspace ({workspace_root}). "
f"It is not migrated automatically - if it has session history, memory, "
f"or skills you want to keep, move it into {workspace_root} manually."
)
except Exception as e:
logger.warning(f"[App] Legacy workspace check failed: {e}")
def _sync_builtin_skills():
"""Sync builtin skills from project skills/ into every enabled Agent's
workspace, so a newly configured Agent is not born without them."""
import shutil
try:
from agent.registry import get_agent_registry
from common.runtime_identity import RuntimeIdentity
from common.state_dir import skills_dir
project_root = os.path.dirname(os.path.abspath(__file__))
builtin_dir = os.path.join(project_root, "skills")
if not os.path.isdir(builtin_dir):
return
for profile in get_agent_registry().list(include_disabled=False):
custom_dir = str(
skills_dir(RuntimeIdentity(agent_id=profile.id), ensure=True)
)
synced = 0
for name in os.listdir(builtin_dir):
src = os.path.join(builtin_dir, name)
if not os.path.isdir(src) or not os.path.isfile(os.path.join(src, "SKILL.md")):
continue
dst = os.path.join(custom_dir, name)
try:
if os.path.isdir(dst):
shutil.rmtree(dst)
shutil.copytree(src, dst)
synced += 1
except Exception as e:
logger.warning(f"[App] Failed to sync builtin skill '{name}': {e}")
if synced:
logger.info(
f"[App] Synced {synced} builtin skill(s) to workspace of "
f"agent '{profile.id}'"
)
except Exception as e:
logger.warning(f"[App] Builtin skills sync failed: {e}")
def _scaffold_subagent_assets():
"""Seed every enabled Agent's subagents/ directory with the guide and the
example type, so there is something to copy from.
Only when the feature is on: an install that never enables sub agents
should not grow a directory for them. Files are written once rather than
synced like skills, so a user who edits or deletes one keeps that choice.
"""
import shutil
try:
from agent.registry import get_agent_registry
from agent.subagent import SubagentSettings
from common.runtime_identity import RuntimeIdentity
from common.state_dir import subagents_dir
if not SubagentSettings.from_config().enabled:
return
asset_dir = os.path.join(
os.path.dirname(os.path.abspath(__file__)), "agent", "subagent", "assets"
)
if not os.path.isdir(asset_dir):
return
for profile in get_agent_registry().list(include_disabled=False):
target_dir = subagents_dir(RuntimeIdentity(agent_id=profile.id), ensure=True)
written = 0
for name in sorted(os.listdir(asset_dir)):
src = os.path.join(asset_dir, name)
target = target_dir / name
if not os.path.isfile(src) or target.exists():
continue
try:
shutil.copyfile(src, target)
written += 1
except Exception as e:
logger.warning(f"[App] Failed to write sub agent asset '{name}': {e}")
if written:
logger.info(
f"[App] Seeded {written} sub agent file(s) in workspace of "
f"agent '{profile.id}'"
)
except Exception as e:
logger.warning(f"[App] Sub agent scaffold failed: {e}")
def run():
global _channel_mgr
try:
# Before any TLS connection: a packaged build has no OpenSSL CA store.
bundle = ensure_ca_bundle()
if bundle:
logger.debug(f"[App] using certifi CA bundle: {bundle}")
# load config
load_config()
_migrate_team_roster()
_warn_if_legacy_workspace_data_exists()
# ctrl + c
sigterm_handler_wrap(signal.SIGINT)
# kill signal
sigterm_handler_wrap(signal.SIGTERM)
# Parse channel_type into a list
raw_channel = conf().get("channel_type", "web")
if "--cmd" in sys.argv:
channel_names = ["terminal"]
else:
# Multi-instance opt-in: when team.json defines channel_instances,
# start those (each with its own credentials + Agent binding).
# Otherwise fall back to the legacy channel_type list untouched.
channel_names = _resolve_startup_channels(raw_channel)
# Auto-start web console unless explicitly disabled. The web entry stays
# a legacy string; only IM channels participate in multi-instance.
web_console_enabled = conf().get("web_console", True)
if web_console_enabled and not _has_web_entry(channel_names):
channel_names.append("web")
# Sync builtin skills to workspace before channels start
_sync_builtin_skills()
_scaffold_subagent_assets()
# Kick off MCP server loading in the background so first-message
# latency isn't dominated by npx package downloads. Skipped in desktop
# mode (MCP relies on external npx/uvx runtimes that aren't bundled).
if not DESKTOP_MODE:
_warmup_mcp_tools()
if DESKTOP_MODE:
# Defer the (heavy) AgentBridge/scheduler warmup to a background
# thread so the web API becomes available within a couple seconds.
# The scheduler still starts; it just doesn't block UI readiness.
_preload_heavy_imports()
_start_web_watchdog()
threading.Thread(target=_warmup_scheduler, daemon=True).start()
else:
_warmup_scheduler()
logger.info(f"[App] Starting channels: {channel_names}")
_channel_mgr = ChannelManager()
_channel_mgr.start(channel_names, first_start=True)
while True:
time.sleep(1)
except KeyboardInterrupt:
pass
except Exception as e:
logger.error("App startup failed!")
logger.exception(e)
# Desktop shell reads exit code 0 as a clean shutdown and would spin on
# "connecting" until its timeout. Exit non-zero so it surfaces the real
# error and offers a retry right away.
if DESKTOP_MODE:
logging.shutdown()
os._exit(1)
if __name__ == "__main__":
run()