zhaoxi
/
Pretor
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat: 修复了一些bug

This commit is contained in:
朝夕 2026-04-28 19:47:19 +08:00
parent 73d28793ae
commit 0a7197295e
2 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
frontend/src/components/Agent/WorkerIndividualSettings.tsx
View File

@ -45,7 +45,7 @@ export function WorkerIndividualSettings() {
setProviders(Object.values(provRes.data.provider_list || {})); setProviders(Object.values(provRes.data.provider_list || {}));
setWorkers(workRes.data.workers || []); setWorkers(workRes.data.workers || []);
const allTools = toolsRes.data.tools ? Object.values(toolsRes.data.tools).flatMap(tGroup => Object.keys(tGroup as any)) : []; const allTools = toolsRes.data.tools || [];
setAvailableTools(allTools); setAvailableTools(allTools);
setAvailableSkills(Object.keys(skillsRes.data.skills || {})); setAvailableSkills(Object.keys(skillsRes.data.skills || {}));

7
pretor/worker_individual/worker_cluster.py
View File

@ -38,12 +38,14 @@ class WorkerCluster:
self.max_capacity = max_capacity self.max_capacity = max_capacity
self._active_workers: OrderedDict[str, BaseIndividual] = OrderedDict() self._active_workers: OrderedDict[str, BaseIndividual] = OrderedDict()
self.status = "running" self.status = "running"
self.task_queue = Queue() self.task_queue = None
self.results_futures = {} self.results_futures = {}
self.runners = [] self.runners = []
self.num_runners = num_runners self.num_runners = num_runners
async def start(self): async def start(self):
if self.task_queue is None:
self.task_queue = Queue()
self.runners = [asyncio.create_task(self._runner(i)) for i in range(self.num_runners)] self.runners = [asyncio.create_task(self._runner(i)) for i in range(self.num_runners)]
logger.info(f"WorkerCluster 已启动 {self.num_runners} 个 runner 协程。") logger.info(f"WorkerCluster 已启动 {self.num_runners} 个 runner 协程。")
@ -77,6 +79,9 @@ class WorkerCluster:
async def _runner(self, runner_id: int): async def _runner(self, runner_id: int):
while True: while True:
try: try:
if self.task_queue is None:
await asyncio.sleep(0.1)
continue
task = await self.task_queue.get_async() task = await self.task_queue.get_async()
task_id = task.get("task_id") task_id = task.get("task_id")
agent_id = task.get("agent_id") agent_id = task.get("agent_id")