@perstack/runtime 0.0.63 → 0.0.64

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
package/dist/src/index.js CHANGED
@@ -1,2454 +1,185 @@
1
- import { existsSync, mkdirSync, writeFileSync, readFileSync, readdirSync } from 'fs';
2
- import { readFile, mkdir, writeFile } from 'fs/promises';
3
- import path2 from 'path';
4
- import { jobSchema, runSettingSchema, checkpointSchema, knownModels, stopRunByDelegate, stopRunByInteractiveTool, resolveToolResults, attemptCompletion, callDelegate, callInteractiveTool, stopRunByExceededMaxSteps, continueToNextStep, retry, completeRun, callTools, startRun, resumeToolCalls, finishAllToolCalls, startGeneration, finishToolCall, runParamsSchema, createRuntimeEvent } from '@perstack/core';
5
- import { createAmazonBedrock } from '@ai-sdk/amazon-bedrock';
6
- import { createAnthropic } from '@ai-sdk/anthropic';
7
- import { createAzure } from '@ai-sdk/azure';
8
- import { createDeepSeek } from '@ai-sdk/deepseek';
9
- import { createGoogleGenerativeAI } from '@ai-sdk/google';
10
- import { createVertex } from '@ai-sdk/google-vertex';
11
- import { createOpenAI } from '@ai-sdk/openai';
12
- import { createOllama } from 'ollama-ai-provider-v2';
13
- import { createId } from '@paralleldrive/cuid2';
14
- import { setup, assign, createActor } from 'xstate';
15
- import { generateText, tool, jsonSchema } from 'ai';
16
- import { Client } from '@modelcontextprotocol/sdk/client/index.js';
17
- import { SSEClientTransport } from '@modelcontextprotocol/sdk/client/sse.js';
18
- import { StdioClientTransport } from '@modelcontextprotocol/sdk/client/stdio.js';
19
- import { McpError } from '@modelcontextprotocol/sdk/types.js';
20
- import { dedent } from 'ts-dedent';
21
- import { ApiV1Client } from '@perstack/api-client/v1';
22
-
23
- // package.json
24
- var package_default = {
25
- version: "0.0.63"};
26
- function getJobsDir() {
27
- return `${process.cwd()}/perstack/jobs`;
28
- }
29
- function getJobDir(jobId) {
30
- return `${getJobsDir()}/${jobId}`;
31
- }
32
- function storeJob(job) {
33
- const jobDir = getJobDir(job.id);
34
- if (!existsSync(jobDir)) {
35
- mkdirSync(jobDir, { recursive: true });
36
- }
37
- const jobPath = path2.resolve(jobDir, "job.json");
38
- writeFileSync(jobPath, JSON.stringify(job, null, 2));
39
- }
40
- function retrieveJob(jobId) {
41
- const jobDir = getJobDir(jobId);
42
- const jobPath = path2.resolve(jobDir, "job.json");
43
- if (!existsSync(jobPath)) {
44
- return void 0;
45
- }
46
- const content = readFileSync(jobPath, "utf-8");
47
- return jobSchema.parse(JSON.parse(content));
48
- }
49
- function getAllJobs() {
50
- const jobsDir = getJobsDir();
51
- if (!existsSync(jobsDir)) {
52
- return [];
53
- }
54
- const jobDirNames = readdirSync(jobsDir, { withFileTypes: true }).filter((dir) => dir.isDirectory()).map((dir) => dir.name);
55
- if (jobDirNames.length === 0) {
56
- return [];
57
- }
58
- const jobs = [];
59
- for (const jobDirName of jobDirNames) {
60
- const jobPath = path2.resolve(jobsDir, jobDirName, "job.json");
61
- if (!existsSync(jobPath)) {
62
- continue;
63
- }
64
- try {
65
- const content = readFileSync(jobPath, "utf-8");
66
- jobs.push(jobSchema.parse(JSON.parse(content)));
67
- } catch {
68
- }
69
- }
70
- return jobs.sort((a, b) => b.startedAt - a.startedAt);
71
- }
72
- function createInitialJob(jobId, expertKey, maxSteps) {
73
- return {
74
- id: jobId,
75
- status: "running",
76
- coordinatorExpertKey: expertKey,
77
- totalSteps: 0,
78
- maxSteps,
79
- usage: {
80
- inputTokens: 0,
81
- outputTokens: 0,
82
- reasoningTokens: 0,
83
- totalTokens: 0,
84
- cachedInputTokens: 0
85
- },
86
- startedAt: Date.now()
87
- };
88
- }
89
- async function createDefaultFileSystem() {
90
- const fs = await import('fs');
91
- const fsPromises = await import('fs/promises');
92
- return {
93
- existsSync: fs.existsSync,
94
- mkdir: async (p, options) => {
95
- await fsPromises.mkdir(p, options);
96
- },
97
- readFile: fsPromises.readFile,
98
- writeFile: fsPromises.writeFile
99
- };
100
- }
101
- function defaultGetRunDir(jobId, runId) {
102
- return `${process.cwd()}/perstack/jobs/${jobId}/runs/${runId}`;
103
- }
104
- async function storeRunSetting(setting, fs, getRunDir = defaultGetRunDir) {
105
- const fileSystem = fs ?? await createDefaultFileSystem();
106
- const runDir = getRunDir(setting.jobId, setting.runId);
107
- if (fileSystem.existsSync(runDir)) {
108
- const runSettingPath = path2.resolve(runDir, "run-setting.json");
109
- const runSetting = runSettingSchema.parse(
110
- JSON.parse(await fileSystem.readFile(runSettingPath, "utf-8"))
111
- );
112
- runSetting.updatedAt = Date.now();
113
- await fileSystem.writeFile(runSettingPath, JSON.stringify(runSetting), "utf-8");
114
- } else {
115
- await fileSystem.mkdir(runDir, { recursive: true });
116
- await fileSystem.writeFile(
117
- path2.resolve(runDir, "run-setting.json"),
118
- JSON.stringify(setting),
119
- "utf-8"
120
- );
121
- }
122
- }
123
- function getAllRuns() {
124
- const jobsDir = getJobsDir();
125
- if (!existsSync(jobsDir)) {
126
- return [];
127
- }
128
- const jobDirNames = readdirSync(jobsDir, { withFileTypes: true }).filter((dir) => dir.isDirectory()).map((dir) => dir.name);
129
- if (jobDirNames.length === 0) {
130
- return [];
131
- }
132
- const runs = [];
133
- for (const jobDirName of jobDirNames) {
134
- const runsDir = path2.resolve(jobsDir, jobDirName, "runs");
135
- if (!existsSync(runsDir)) {
136
- continue;
137
- }
138
- const runDirNames = readdirSync(runsDir, { withFileTypes: true }).filter((dir) => dir.isDirectory()).map((dir) => dir.name);
139
- for (const runDirName of runDirNames) {
140
- const runSettingPath = path2.resolve(runsDir, runDirName, "run-setting.json");
141
- if (!existsSync(runSettingPath)) {
142
- continue;
143
- }
144
- try {
145
- const content = readFileSync(runSettingPath, "utf-8");
146
- runs.push(runSettingSchema.parse(JSON.parse(content)));
147
- } catch {
148
- }
149
- }
150
- }
151
- return runs.sort((a, b) => b.updatedAt - a.updatedAt);
152
- }
153
-
154
- // src/default-store.ts
155
- function getCheckpointDir(jobId) {
156
- return `${getJobDir(jobId)}/checkpoints`;
157
- }
158
- function getCheckpointPath(jobId, checkpointId) {
159
- return `${getCheckpointDir(jobId)}/${checkpointId}.json`;
160
- }
161
- async function defaultRetrieveCheckpoint(jobId, checkpointId) {
162
- const checkpointPath = getCheckpointPath(jobId, checkpointId);
163
- if (!existsSync(checkpointPath)) {
164
- throw new Error(`checkpoint not found: ${checkpointId}`);
165
- }
166
- const checkpoint = await readFile(checkpointPath, "utf8");
167
- return checkpointSchema.parse(JSON.parse(checkpoint));
168
- }
169
- async function defaultStoreCheckpoint(checkpoint) {
170
- const { id, jobId } = checkpoint;
171
- const checkpointDir = getCheckpointDir(jobId);
172
- await mkdir(checkpointDir, { recursive: true });
173
- await writeFile(getCheckpointPath(jobId, id), JSON.stringify(checkpoint));
174
- }
175
- async function defaultStoreEvent(event) {
176
- const { timestamp, jobId, runId, stepNumber, type } = event;
177
- const runDir = defaultGetRunDir(jobId, runId);
178
- const eventPath = `${runDir}/event-${timestamp}-${stepNumber}-${type}.json`;
179
- await mkdir(runDir, { recursive: true });
180
- await writeFile(eventPath, JSON.stringify(event));
181
- }
182
- function getCheckpointsByJobId(jobId) {
183
- const checkpointDir = getCheckpointDir(jobId);
184
- if (!existsSync(checkpointDir)) {
185
- return [];
186
- }
187
- const files = readdirSync(checkpointDir).filter((file) => file.endsWith(".json"));
188
- const checkpoints = [];
189
- for (const file of files) {
190
- try {
191
- const content = readFileSync(path2.resolve(checkpointDir, file), "utf-8");
192
- checkpoints.push(checkpointSchema.parse(JSON.parse(content)));
193
- } catch {
194
- }
195
- }
196
- return checkpoints.sort((a, b) => a.stepNumber - b.stepNumber);
197
- }
198
- function getEventsByRun(jobId, runId) {
199
- const runDir = defaultGetRunDir(jobId, runId);
200
- if (!existsSync(runDir)) {
201
- return [];
202
- }
203
- return readdirSync(runDir).filter((file) => file.startsWith("event-")).map((file) => {
204
- const [_, timestamp, stepNumber, type] = file.split(".")[0].split("-");
205
- return { timestamp: Number(timestamp), stepNumber: Number(stepNumber), type };
206
- }).sort((a, b) => a.stepNumber - b.stepNumber);
207
- }
208
- function getEventContents(jobId, runId, maxStepNumber) {
209
- const runDir = defaultGetRunDir(jobId, runId);
210
- if (!existsSync(runDir)) {
211
- return [];
212
- }
213
- const eventFiles = readdirSync(runDir).filter((file) => file.startsWith("event-")).map((file) => {
214
- const [_, timestamp, step, type] = file.split(".")[0].split("-");
215
- return { file, timestamp: Number(timestamp), stepNumber: Number(step), type };
216
- }).filter((e) => maxStepNumber === void 0 || e.stepNumber <= maxStepNumber).sort((a, b) => a.timestamp - b.timestamp);
217
- const events = [];
218
- for (const { file } of eventFiles) {
219
- try {
220
- const content = readFileSync(path2.resolve(runDir, file), "utf-8");
221
- events.push(JSON.parse(content));
222
- } catch {
223
- }
224
- }
225
- return events;
226
- }
227
- function getModel(modelId, providerConfig) {
228
- switch (providerConfig.providerName) {
229
- case "anthropic": {
230
- const anthropic = createAnthropic({
231
- apiKey: providerConfig.apiKey,
232
- baseURL: providerConfig.baseUrl,
233
- headers: providerConfig.headers
234
- });
235
- return anthropic(modelId);
236
- }
237
- case "google": {
238
- const google = createGoogleGenerativeAI({
239
- apiKey: providerConfig.apiKey,
240
- baseURL: providerConfig.baseUrl,
241
- headers: providerConfig.headers
242
- });
243
- return google(modelId);
244
- }
245
- case "openai": {
246
- const openai = createOpenAI({
247
- apiKey: providerConfig.apiKey,
248
- baseURL: providerConfig.baseUrl,
249
- organization: providerConfig.organization,
250
- project: providerConfig.project,
251
- name: providerConfig.name,
252
- headers: providerConfig.headers
253
- });
254
- return openai(modelId);
255
- }
256
- case "ollama": {
257
- const ollama = createOllama({
258
- baseURL: providerConfig.baseUrl,
259
- headers: providerConfig.headers
260
- });
261
- return ollama(modelId);
262
- }
263
- case "azure-openai": {
264
- const azure = createAzure({
265
- apiKey: providerConfig.apiKey,
266
- resourceName: providerConfig.resourceName,
267
- apiVersion: providerConfig.apiVersion,
268
- baseURL: providerConfig.baseUrl,
269
- headers: providerConfig.headers,
270
- useDeploymentBasedUrls: providerConfig.useDeploymentBasedUrls
271
- });
272
- return azure(modelId);
273
- }
274
- case "amazon-bedrock": {
275
- const amazonBedrock = createAmazonBedrock({
276
- accessKeyId: providerConfig.accessKeyId,
277
- secretAccessKey: providerConfig.secretAccessKey,
278
- region: providerConfig.region,
279
- sessionToken: providerConfig.sessionToken
280
- });
281
- return amazonBedrock(modelId);
282
- }
283
- case "google-vertex": {
284
- const vertex = createVertex({
285
- project: providerConfig.project,
286
- location: providerConfig.location,
287
- baseURL: providerConfig.baseUrl,
288
- headers: providerConfig.headers
289
- });
290
- return vertex(modelId);
291
- }
292
- case "deepseek": {
293
- const deepseek = createDeepSeek({
294
- apiKey: providerConfig.apiKey,
295
- baseURL: providerConfig.baseUrl,
296
- headers: providerConfig.headers
297
- });
298
- return deepseek(modelId);
299
- }
300
- default: {
301
- const _exhaustive = providerConfig;
302
- throw new Error(`Unknown provider: ${_exhaustive.providerName}`);
303
- }
304
- }
305
- }
306
- function getContextWindow(providerName, modelId) {
307
- const modelConfig = knownModels.find((model) => model.provider === providerName)?.models.find((model) => model.name === modelId);
308
- return modelConfig?.contextWindow;
309
- }
310
- function calculateContextWindowUsage(usage, contextWindow) {
311
- return (usage.inputTokens + usage.cachedInputTokens + usage.outputTokens) / contextWindow;
312
- }
313
-
314
- // src/usage.ts
315
- function createEmptyUsage() {
316
- return {
317
- inputTokens: 0,
318
- outputTokens: 0,
319
- reasoningTokens: 0,
320
- totalTokens: 0,
321
- cachedInputTokens: 0
322
- };
323
- }
324
- function usageFromGenerateTextResult(result) {
325
- return {
326
- inputTokens: result.usage.inputTokens || 0,
327
- outputTokens: result.usage.outputTokens || 0,
328
- reasoningTokens: result.usage.reasoningTokens || 0,
329
- totalTokens: result.usage.totalTokens || 0,
330
- cachedInputTokens: result.usage.cachedInputTokens || 0
331
- };
332
- }
333
- function sumUsage(a, b) {
334
- return {
335
- inputTokens: a.inputTokens + b.inputTokens,
336
- outputTokens: a.outputTokens + b.outputTokens,
337
- reasoningTokens: a.reasoningTokens + b.reasoningTokens,
338
- totalTokens: a.totalTokens + b.totalTokens,
339
- cachedInputTokens: a.cachedInputTokens + b.cachedInputTokens
340
- };
341
- }
342
-
343
- // src/checkpoint-helpers.ts
344
- function createInitialCheckpoint(checkpointId, params) {
345
- return {
346
- id: checkpointId,
347
- jobId: params.jobId,
348
- runId: params.runId,
349
- expert: {
350
- key: params.expertKey,
351
- name: params.expert.name,
352
- version: params.expert.version
353
- },
354
- stepNumber: 1,
355
- status: "init",
356
- messages: [],
357
- usage: createEmptyUsage(),
358
- contextWindow: params.contextWindow,
359
- contextWindowUsage: params.contextWindow ? 0 : void 0
360
- };
361
- }
362
- function createNextStepCheckpoint(checkpointId, checkpoint) {
363
- return {
364
- ...checkpoint,
365
- id: checkpointId,
366
- stepNumber: checkpoint.stepNumber + 1
367
- };
368
- }
369
- function buildDelegationReturnState(currentSetting, resultCheckpoint, parentCheckpoint) {
370
- const { messages, delegatedBy } = resultCheckpoint;
371
- if (!delegatedBy) {
372
- throw new Error("delegatedBy is required for buildDelegationReturnState");
373
- }
374
- const delegateResultMessage = messages[messages.length - 1];
375
- if (!delegateResultMessage || delegateResultMessage.type !== "expertMessage") {
376
- throw new Error("Delegation error: delegation result message is incorrect");
377
- }
378
- const delegateText = delegateResultMessage.contents.find((content) => content.type === "textPart");
379
- if (!delegateText) {
380
- throw new Error("Delegation error: delegation result message does not contain a text");
381
- }
382
- const { expert, toolCallId, toolName } = delegatedBy;
383
- return {
384
- setting: {
385
- ...currentSetting,
386
- expertKey: expert.key,
387
- input: {
388
- interactiveToolCallResult: {
389
- toolCallId,
390
- toolName,
391
- skillName: `delegate/${resultCheckpoint.expert.key}`,
392
- text: delegateText.text
393
- }
394
- }
395
- },
396
- checkpoint: {
397
- ...parentCheckpoint,
398
- stepNumber: resultCheckpoint.stepNumber,
399
- usage: resultCheckpoint.usage,
400
- pendingToolCalls: parentCheckpoint.pendingToolCalls,
401
- partialToolResults: parentCheckpoint.partialToolResults
402
- }
403
- };
404
- }
405
- function buildDelegateToState(currentSetting, resultCheckpoint, currentExpert) {
406
- const { delegateTo } = resultCheckpoint;
407
- if (!delegateTo || delegateTo.length === 0) {
408
- throw new Error("delegateTo is required for buildDelegateToState");
409
- }
410
- const firstDelegation = delegateTo[0];
411
- const { expert, toolCallId, toolName, query } = firstDelegation;
412
- return {
413
- setting: {
414
- ...currentSetting,
415
- expertKey: expert.key,
416
- input: {
417
- text: query
418
- }
419
- },
420
- checkpoint: {
421
- ...resultCheckpoint,
422
- status: "init",
423
- messages: [],
424
- expert: {
425
- key: expert.key,
426
- name: expert.name,
427
- version: expert.version
428
- },
429
- delegatedBy: {
430
- expert: {
431
- key: currentExpert.key,
432
- name: currentExpert.name,
433
- version: currentExpert.version
434
- },
435
- toolCallId,
436
- toolName,
437
- checkpointId: resultCheckpoint.id
438
- },
439
- usage: resultCheckpoint.usage,
440
- pendingToolCalls: void 0,
441
- partialToolResults: void 0
442
- }
443
- };
444
- }
445
- var RunEventEmitter = class {
446
- listeners = [];
447
- subscribe(listener) {
448
- this.listeners.push(listener);
449
- }
450
- async emit(event) {
451
- const errors = [];
452
- for (const listener of this.listeners) {
453
- try {
454
- await listener({
455
- ...event,
456
- id: createId(),
457
- timestamp: Date.now()
458
- });
459
- } catch (error) {
460
- errors.push(error instanceof Error ? error : new Error(String(error)));
461
- }
462
- }
463
- if (errors.length > 0) {
464
- throw new AggregateError(errors, "One or more event listeners failed");
465
- }
466
- }
467
- };
468
-
469
- // src/skill-manager/base.ts
470
- var BaseSkillManager = class {
471
- _toolDefinitions = [];
472
- _initialized = false;
473
- _initializing;
474
- skill;
475
- interactiveSkill;
476
- expert;
477
- _jobId;
478
- _runId;
479
- _eventListener;
480
- constructor(jobId, runId, eventListener) {
481
- this._jobId = jobId;
482
- this._runId = runId;
483
- this._eventListener = eventListener;
484
- }
485
- async init() {
486
- if (this._initialized) {
487
- throw new Error(`Skill ${this.name} is already initialized`);
488
- }
489
- if (this._initializing) {
490
- throw new Error(`Skill ${this.name} is already initializing`);
491
- }
492
- const initPromise = this._performInit();
493
- this._initializing = initPromise;
494
- if (!this.lazyInit) {
495
- try {
496
- await initPromise;
497
- } catch (error) {
498
- this._initialized = false;
499
- this._initializing = void 0;
500
- throw error;
501
- }
502
- }
503
- }
504
- isInitialized() {
505
- return this._initialized;
506
- }
507
- async _performInit() {
508
- await this._doInit();
509
- this._initialized = true;
510
- this._initializing = void 0;
511
- }
512
- async getToolDefinitions() {
513
- if (!this.isInitialized() && !this.lazyInit) {
514
- throw new Error(`Skill ${this.name} is not initialized`);
515
- }
516
- if (!this.isInitialized() && this.lazyInit) {
517
- return [];
518
- }
519
- return this._filterTools(this._toolDefinitions);
520
- }
521
- _filterTools(tools) {
522
- return tools;
523
- }
524
- };
525
-
526
- // src/skill-manager/delegate.ts
527
- var DelegateSkillManager = class extends BaseSkillManager {
528
- name;
529
- type = "delegate";
530
- lazyInit = false;
531
- expert;
532
- constructor(expert, jobId, runId, eventListener) {
533
- super(jobId, runId, eventListener);
534
- this.name = expert.name;
535
- this.expert = expert;
536
- }
537
- async _doInit() {
538
- this._toolDefinitions = [
539
- {
540
- skillName: this.expert.name,
541
- name: this.expert.name.split("/").pop() ?? this.expert.name,
542
- description: this.expert.description,
543
- inputSchema: {
544
- type: "object",
545
- properties: { query: { type: "string" } },
546
- required: ["query"]
547
- },
548
- interactive: false
549
- }
550
- ];
551
- }
552
- async close() {
553
- }
554
- async callTool(_toolName, _input) {
555
- return [];
556
- }
557
- };
558
-
559
- // src/skill-manager/interactive.ts
560
- var InteractiveSkillManager = class extends BaseSkillManager {
561
- name;
562
- type = "interactive";
563
- lazyInit = false;
564
- interactiveSkill;
565
- constructor(interactiveSkill, jobId, runId, eventListener) {
566
- super(jobId, runId, eventListener);
567
- this.name = interactiveSkill.name;
568
- this.interactiveSkill = interactiveSkill;
569
- }
570
- async _doInit() {
571
- this._toolDefinitions = Object.values(this.interactiveSkill.tools).map((tool2) => ({
572
- skillName: this.interactiveSkill.name,
573
- name: tool2.name,
574
- description: tool2.description,
575
- inputSchema: JSON.parse(tool2.inputJsonSchema),
576
- interactive: true
577
- }));
578
- }
579
- async close() {
580
- }
581
- async callTool(_toolName, _input) {
582
- return [];
583
- }
584
- };
585
- var McpSkillManager = class extends BaseSkillManager {
586
- name;
587
- type = "mcp";
588
- lazyInit;
589
- skill;
590
- _mcpClient;
591
- _env;
592
- constructor(skill, env, jobId, runId, eventListener) {
593
- super(jobId, runId, eventListener);
594
- this.name = skill.name;
595
- this.skill = skill;
596
- this._env = env;
597
- this.lazyInit = skill.type === "mcpStdioSkill" && skill.lazyInit && skill.name !== "@perstack/base";
598
- }
599
- async _doInit() {
600
- this._mcpClient = new Client({
601
- name: `${this.skill.name}-mcp-client`,
602
- version: "1.0.0"
603
- });
604
- if (this.skill.type === "mcpStdioSkill") {
605
- await this._initStdio(this.skill);
606
- } else {
607
- await this._initSse(this.skill);
608
- }
609
- const { tools } = await this._mcpClient.listTools();
610
- this._toolDefinitions = tools.map((tool2) => ({
611
- skillName: this.skill.name,
612
- name: tool2.name,
613
- description: tool2.description,
614
- inputSchema: tool2.inputSchema,
615
- interactive: false
616
- }));
617
- }
618
- async _initStdio(skill) {
619
- if (!skill.command) {
620
- throw new Error(`Skill ${skill.name} has no command`);
621
- }
622
- const env = {};
623
- for (const envName of skill.requiredEnv) {
624
- if (!this._env[envName]) {
625
- throw new Error(`Skill ${skill.name} requires environment variable ${envName}`);
626
- }
627
- env[envName] = this._env[envName];
628
- }
629
- const startTime = Date.now();
630
- const { command, args } = this._getCommandArgs(skill);
631
- if (this._eventListener) {
632
- const event = createRuntimeEvent("skillStarting", this._jobId, this._runId, {
633
- skillName: skill.name,
634
- command,
635
- args
636
- });
637
- this._eventListener(event);
638
- }
639
- const transport = new StdioClientTransport({ command, args, env, stderr: "pipe" });
640
- if (transport.stderr) {
641
- transport.stderr.on("data", (chunk) => {
642
- if (this._eventListener) {
643
- const event = createRuntimeEvent("skillStderr", this._jobId, this._runId, {
644
- skillName: skill.name,
645
- message: chunk.toString().trim()
646
- });
647
- this._eventListener(event);
648
- }
649
- });
650
- }
651
- const connectStartTime = Date.now();
652
- await this._mcpClient.connect(transport);
653
- const connectTime = Date.now();
654
- if (this._eventListener) {
655
- const serverInfo = this._mcpClient.getServerVersion();
656
- const event = createRuntimeEvent("skillConnected", this._jobId, this._runId, {
657
- skillName: skill.name,
658
- serverInfo: serverInfo ? { name: serverInfo.name, version: serverInfo.version } : void 0,
659
- connectDurationMs: connectTime - connectStartTime,
660
- totalDurationMs: connectTime - startTime
661
- });
662
- this._eventListener(event);
663
- }
664
- }
665
- async _initSse(skill) {
666
- if (!skill.endpoint) {
667
- throw new Error(`Skill ${skill.name} has no endpoint`);
668
- }
669
- const transport = new SSEClientTransport(new URL(skill.endpoint));
670
- await this._mcpClient.connect(transport);
671
- }
672
- _getCommandArgs(skill) {
673
- const { name, command, packageName, args } = skill;
674
- if (!packageName && (!args || args.length === 0)) {
675
- throw new Error(`Skill ${name} has no packageName or args. Please provide one of them.`);
676
- }
677
- if (packageName && args && args.length > 0) {
678
- throw new Error(
679
- `Skill ${name} has both packageName and args. Please provide only one of them.`
680
- );
681
- }
682
- let newArgs = args && args.length > 0 ? args : [packageName];
683
- if (command === "npx" && !newArgs.includes("-y")) {
684
- newArgs = ["-y", ...newArgs];
685
- }
686
- return { command, args: newArgs };
687
- }
688
- async close() {
689
- if (this._mcpClient) {
690
- await this._mcpClient.close();
691
- if (this._eventListener && this.skill) {
692
- const event = createRuntimeEvent("skillDisconnected", this._jobId, this._runId, {
693
- skillName: this.skill.name
694
- });
695
- this._eventListener(event);
696
- }
697
- }
698
- }
699
- _filterTools(tools) {
700
- const omit = this.skill.omit ?? [];
701
- const pick = this.skill.pick ?? [];
702
- return tools.filter((tool2) => omit.length > 0 ? !omit.includes(tool2.name) : true).filter((tool2) => pick.length > 0 ? pick.includes(tool2.name) : true);
703
- }
704
- async callTool(toolName, input) {
705
- if (!this.isInitialized() || !this._mcpClient) {
706
- throw new Error(`${this.name} is not initialized`);
1
+ import { package_default, run } from '../chunk-C7AFEVYF.js';
2
+ export { getModel, run, runtimeStateMachine } from '../chunk-C7AFEVYF.js';
3
+ import { registerAdapter, BaseAdapter, checkpointSchema } from '@perstack/core';
4
+ import { spawn } from 'child_process';
5
+
6
+ var PerstackAdapter = class extends BaseAdapter {
7
+ name = "perstack";
8
+ version = "unknown";
9
+ useDirectExecution;
10
+ constructor(options) {
11
+ super();
12
+ this.useDirectExecution = options?.useDirectExecution ?? true;
13
+ }
14
+ async checkPrerequisites() {
15
+ if (this.useDirectExecution) {
16
+ return { ok: true };
707
17
  }
708
18
  try {
709
- const result = await this._mcpClient.callTool({
710
- name: toolName,
711
- arguments: input
712
- });
713
- return this._convertToolResult(result, toolName, input);
714
- } catch (error) {
715
- return this._handleToolError(error, toolName);
716
- }
717
- }
718
- _handleToolError(error, toolName) {
719
- if (error instanceof McpError) {
720
- return [
721
- {
722
- type: "textPart",
723
- text: `Error calling tool ${toolName}: ${error.message}`,
724
- id: createId()
725
- }
726
- ];
727
- }
728
- throw error;
729
- }
730
- _convertToolResult(result, toolName, input) {
731
- if (!result.content || result.content.length === 0) {
732
- return [
733
- {
734
- type: "textPart",
735
- text: `Tool ${toolName} returned nothing with arguments: ${JSON.stringify(input)}`,
736
- id: createId()
737
- }
738
- ];
739
- }
740
- return result.content.filter((part) => part.type !== "audio" && part.type !== "resource_link").map((part) => this._convertPart(part));
741
- }
742
- _convertPart(part) {
743
- switch (part.type) {
744
- case "text":
745
- if (!part.text || part.text === "") {
746
- return { type: "textPart", text: "Error: No content", id: createId() };
747
- }
748
- return { type: "textPart", text: part.text, id: createId() };
749
- case "image":
750
- if (!part.data || !part.mimeType) {
751
- throw new Error("Image part must have both data and mimeType");
752
- }
753
- return {
754
- type: "imageInlinePart",
755
- encodedData: part.data,
756
- mimeType: part.mimeType,
757
- id: createId()
758
- };
759
- case "resource":
760
- if (!part.resource) {
761
- throw new Error("Resource part must have resource content");
762
- }
763
- return this._convertResource(part.resource);
764
- }
765
- }
766
- _convertResource(resource) {
767
- if (!resource.mimeType) {
768
- throw new Error(`Resource ${JSON.stringify(resource)} has no mimeType`);
769
- }
770
- if (resource.text && typeof resource.text === "string") {
771
- return { type: "textPart", text: resource.text, id: createId() };
772
- }
773
- if (resource.blob && typeof resource.blob === "string") {
774
- return {
775
- type: "fileInlinePart",
776
- encodedData: resource.blob,
777
- mimeType: resource.mimeType,
778
- id: createId()
779
- };
780
- }
781
- throw new Error(`Unsupported resource type: ${JSON.stringify(resource)}`);
782
- }
783
- };
784
-
785
- // src/skill-manager/helpers.ts
786
- async function initSkillManagersWithCleanup(managers, allManagers) {
787
- const results = await Promise.allSettled(managers.map((m) => m.init()));
788
- const firstRejected = results.find((r) => r.status === "rejected");
789
- if (firstRejected) {
790
- await Promise.all(allManagers.map((m) => m.close().catch(() => {
791
- })));
792
- throw firstRejected.reason;
793
- }
794
- }
795
- async function getSkillManagers(expert, experts, setting, eventListener, options) {
796
- const { perstackBaseSkillCommand, env, jobId, runId } = setting;
797
- const { skills } = expert;
798
- if (!skills["@perstack/base"]) {
799
- throw new Error("Base skill is not defined");
800
- }
801
- const allManagers = [];
802
- const mcpSkills = Object.values(skills).filter((skill) => skill.type === "mcpStdioSkill" || skill.type === "mcpSseSkill").map((skill) => {
803
- if (perstackBaseSkillCommand && skill.type === "mcpStdioSkill") {
804
- const matchesBaseByPackage = skill.command === "npx" && skill.packageName === "@perstack/base";
805
- const matchesBaseByArgs = skill.command === "npx" && Array.isArray(skill.args) && skill.args.includes("@perstack/base");
806
- if (matchesBaseByPackage || matchesBaseByArgs) {
807
- const [overrideCommand, ...overrideArgs] = perstackBaseSkillCommand;
808
- if (!overrideCommand) {
809
- throw new Error("perstackBaseSkillCommand must have at least one element");
810
- }
19
+ const result = await this.execCommand(["perstack-runtime", "--version"]);
20
+ if (result.exitCode !== 0) {
811
21
  return {
812
- ...skill,
813
- command: overrideCommand,
814
- packageName: void 0,
815
- args: overrideArgs,
816
- lazyInit: false
22
+ ok: false,
23
+ error: {
24
+ type: "cli-not-found",
25
+ message: "perstack-runtime CLI is not available."
26
+ }
817
27
  };
818
28
  }
819
- }
820
- return skill;
821
- });
822
- const mcpSkillManagers = mcpSkills.map((skill) => {
823
- const manager = new McpSkillManager(skill, env, jobId, runId, eventListener);
824
- allManagers.push(manager);
825
- return manager;
826
- });
827
- await initSkillManagersWithCleanup(mcpSkillManagers, allManagers);
828
- if (!options?.isDelegatedRun) {
829
- const interactiveSkills = Object.values(skills).filter(
830
- (skill) => skill.type === "interactiveSkill"
831
- );
832
- const interactiveSkillManagers = interactiveSkills.map((interactiveSkill) => {
833
- const manager = new InteractiveSkillManager(interactiveSkill, jobId, runId, eventListener);
834
- allManagers.push(manager);
835
- return manager;
836
- });
837
- await initSkillManagersWithCleanup(interactiveSkillManagers, allManagers);
838
- }
839
- const delegateSkillManagers = [];
840
- for (const delegateExpertName of expert.delegates) {
841
- const delegate = experts[delegateExpertName];
842
- if (!delegate) {
843
- await Promise.all(allManagers.map((m) => m.close().catch(() => {
844
- })));
845
- throw new Error(`Delegate expert "${delegateExpertName}" not found in experts`);
846
- }
847
- const manager = new DelegateSkillManager(delegate, jobId, runId, eventListener);
848
- allManagers.push(manager);
849
- delegateSkillManagers.push(manager);
850
- }
851
- await initSkillManagersWithCleanup(delegateSkillManagers, allManagers);
852
- const skillManagers = {};
853
- for (const manager of allManagers) {
854
- skillManagers[manager.name] = manager;
855
- }
856
- return skillManagers;
857
- }
858
- async function closeSkillManagers(skillManagers) {
859
- await Promise.all(Object.values(skillManagers).map((m) => m.close().catch(() => {
860
- })));
861
- }
862
- async function getSkillManagerByToolName(skillManagers, toolName) {
863
- for (const skillManager of Object.values(skillManagers)) {
864
- const toolDefinitions = await skillManager.getToolDefinitions();
865
- for (const toolDefinition of toolDefinitions) {
866
- if (toolDefinition.name === toolName) {
867
- return skillManager;
868
- }
869
- }
870
- }
871
- throw new Error(`Tool ${toolName} not found`);
872
- }
873
- async function getToolSet(skillManagers) {
874
- const tools = {};
875
- for (const skillManager of Object.values(skillManagers)) {
876
- const toolDefinitions = await skillManager.getToolDefinitions();
877
- for (const toolDefinition of toolDefinitions) {
878
- tools[toolDefinition.name] = tool({
879
- description: toolDefinition.description,
880
- inputSchema: jsonSchema(toolDefinition.inputSchema)
881
- });
882
- }
883
- }
884
- return tools;
885
- }
886
-
887
- // src/states/calling-delegate.ts
888
- async function getToolType(toolName, skillManagers) {
889
- const skillManager = await getSkillManagerByToolName(skillManagers, toolName);
890
- return skillManager.type;
891
- }
892
- async function callingDelegateLogic({
893
- setting,
894
- checkpoint,
895
- step,
896
- skillManagers
897
- }) {
898
- if (!step.pendingToolCalls || step.pendingToolCalls.length === 0) {
899
- throw new Error("No pending tool calls found");
900
- }
901
- const toolCallTypes = await Promise.all(
902
- step.pendingToolCalls.map(async (tc) => ({
903
- toolCall: tc,
904
- type: await getToolType(tc.toolName, skillManagers)
905
- }))
906
- );
907
- const delegateToolCalls = toolCallTypes.filter((t) => t.type === "delegate").map((t) => t.toolCall);
908
- const nonDelegateToolCalls = toolCallTypes.filter((t) => t.type !== "delegate").map((t) => t.toolCall);
909
- if (delegateToolCalls.length === 0) {
910
- throw new Error("No delegate tool calls found");
911
- }
912
- const delegations = await Promise.all(
913
- delegateToolCalls.map(async (tc) => {
914
- const skillManager = await getSkillManagerByToolName(skillManagers, tc.toolName);
915
- if (!skillManager.expert) {
916
- throw new Error(`Delegation error: skill manager "${tc.toolName}" not found`);
917
- }
918
- if (!tc.args || !tc.args.query || typeof tc.args.query !== "string") {
919
- throw new Error(`Delegation error: query is undefined for ${tc.toolName}`);
920
- }
921
- return {
922
- expert: {
923
- key: skillManager.expert.key,
924
- name: skillManager.expert.name,
925
- version: skillManager.expert.version
926
- },
927
- toolCallId: tc.id,
928
- toolName: tc.toolName,
929
- query: tc.args.query
930
- };
931
- })
932
- );
933
- return stopRunByDelegate(setting, checkpoint, {
934
- checkpoint: {
935
- ...checkpoint,
936
- status: "stoppedByDelegate",
937
- delegateTo: delegations,
938
- pendingToolCalls: nonDelegateToolCalls.length > 0 ? nonDelegateToolCalls : void 0,
939
- partialToolResults: step.partialToolResults
940
- },
941
- step: {
942
- ...step,
943
- finishedAt: Date.now()
944
- }
945
- });
946
- }
947
- async function callingInteractiveToolLogic({
948
- setting,
949
- checkpoint,
950
- step
951
- }) {
952
- if (!step.pendingToolCalls || step.pendingToolCalls.length === 0) {
953
- throw new Error("No pending tool calls found");
954
- }
955
- const currentToolCall = step.pendingToolCalls[0];
956
- const remainingToolCalls = step.pendingToolCalls.slice(1);
957
- return stopRunByInteractiveTool(setting, checkpoint, {
958
- checkpoint: {
959
- ...checkpoint,
960
- status: "stoppedByInteractiveTool",
961
- pendingToolCalls: [currentToolCall, ...remainingToolCalls],
962
- partialToolResults: step.partialToolResults
963
- },
964
- step: {
965
- ...step,
966
- finishedAt: (/* @__PURE__ */ new Date()).getTime()
967
- }
968
- });
969
- }
970
- function hasRemainingTodos(toolResult) {
971
- const firstPart = toolResult.result[0];
972
- if (!firstPart || firstPart.type !== "textPart") {
973
- return false;
974
- }
975
- try {
976
- const parsed = JSON.parse(firstPart.text);
977
- return Array.isArray(parsed.remainingTodos) && parsed.remainingTodos.length > 0;
978
- } catch {
979
- return false;
980
- }
981
- }
982
- function isFileInfo(value) {
983
- return typeof value === "object" && value !== null && "path" in value && "mimeType" in value && "size" in value && typeof value.path === "string" && typeof value.mimeType === "string" && typeof value.size === "number";
984
- }
985
- async function processFileToolResult(toolResult, toolName) {
986
- const processedContents = [];
987
- for (const part of toolResult.result) {
988
- if (part.type !== "textPart") {
989
- processedContents.push(part);
990
- continue;
991
- }
992
- let fileInfo;
993
- try {
994
- const parsed = JSON.parse(part.text);
995
- if (isFileInfo(parsed)) {
996
- fileInfo = parsed;
997
- }
29
+ this.version = result.stdout.trim() || "unknown";
998
30
  } catch {
999
- processedContents.push(part);
1000
- continue;
1001
- }
1002
- if (!fileInfo) {
1003
- processedContents.push(part);
1004
- continue;
1005
- }
1006
- const { path: path4, mimeType } = fileInfo;
1007
- try {
1008
- const buffer = await readFile(path4);
1009
- if (toolName === "readImageFile") {
1010
- processedContents.push({
1011
- type: "imageInlinePart",
1012
- id: part.id,
1013
- encodedData: buffer.toString("base64"),
1014
- mimeType
1015
- });
1016
- } else {
1017
- processedContents.push({
1018
- type: "fileInlinePart",
1019
- id: part.id,
1020
- encodedData: buffer.toString("base64"),
1021
- mimeType
1022
- });
1023
- }
1024
- } catch (error) {
1025
- processedContents.push({
1026
- type: "textPart",
1027
- id: part.id,
1028
- text: `Failed to read file "${path4}": ${error instanceof Error ? error.message : String(error)}`
1029
- });
1030
- }
1031
- }
1032
- return { ...toolResult, result: processedContents };
1033
- }
1034
- async function executeMcpToolCall(toolCall, skillManagers) {
1035
- const skillManager = await getSkillManagerByToolName(skillManagers, toolCall.toolName);
1036
- if (skillManager.type !== "mcp") {
1037
- throw new Error(`Incorrect SkillType, required MCP, got ${skillManager.type}`);
1038
- }
1039
- const result = await skillManager.callTool(toolCall.toolName, toolCall.args);
1040
- const toolResult = {
1041
- id: toolCall.id,
1042
- skillName: toolCall.skillName,
1043
- toolName: toolCall.toolName,
1044
- result
1045
- };
1046
- if (toolCall.toolName === "readPdfFile" || toolCall.toolName === "readImageFile") {
1047
- return processFileToolResult(toolResult, toolCall.toolName);
1048
- }
1049
- return toolResult;
1050
- }
1051
- async function getToolType2(toolCall, skillManagers) {
1052
- const skillManager = await getSkillManagerByToolName(skillManagers, toolCall.toolName);
1053
- return skillManager.type;
1054
- }
1055
- async function callingToolLogic({
1056
- setting,
1057
- checkpoint,
1058
- step,
1059
- skillManagers
1060
- }) {
1061
- const pendingToolCalls = step.pendingToolCalls ?? step.toolCalls ?? [];
1062
- if (pendingToolCalls.length === 0) {
1063
- throw new Error("No tool calls found");
1064
- }
1065
- const toolResults = step.toolResults ? [...step.toolResults] : [];
1066
- const attemptCompletionTool = pendingToolCalls.find(
1067
- (tc) => tc.skillName === "@perstack/base" && tc.toolName === "attemptCompletion"
1068
- );
1069
- if (attemptCompletionTool) {
1070
- const toolResult = await executeMcpToolCall(attemptCompletionTool, skillManagers);
1071
- if (hasRemainingTodos(toolResult)) {
1072
- return resolveToolResults(setting, checkpoint, { toolResults: [toolResult] });
1073
- }
1074
- return attemptCompletion(setting, checkpoint, { toolResult });
1075
- }
1076
- const toolCallTypes = await Promise.all(
1077
- pendingToolCalls.map(async (tc) => ({
1078
- toolCall: tc,
1079
- type: await getToolType2(tc, skillManagers)
1080
- }))
1081
- );
1082
- const mcpToolCalls = toolCallTypes.filter((t) => t.type === "mcp").map((t) => t.toolCall);
1083
- const delegateToolCalls = toolCallTypes.filter((t) => t.type === "delegate").map((t) => t.toolCall);
1084
- const interactiveToolCalls = toolCallTypes.filter((t) => t.type === "interactive").map((t) => t.toolCall);
1085
- if (mcpToolCalls.length > 0) {
1086
- const mcpResults = await Promise.all(
1087
- mcpToolCalls.map((tc) => executeMcpToolCall(tc, skillManagers))
1088
- );
1089
- toolResults.push(...mcpResults);
1090
- }
1091
- if (delegateToolCalls.length > 0) {
1092
- step.partialToolResults = toolResults;
1093
- step.pendingToolCalls = [...delegateToolCalls, ...interactiveToolCalls];
1094
- return callDelegate(setting, checkpoint, {
1095
- newMessage: checkpoint.messages[checkpoint.messages.length - 1],
1096
- toolCalls: delegateToolCalls,
1097
- usage: step.usage
1098
- });
1099
- }
1100
- if (interactiveToolCalls.length > 0) {
1101
- const interactiveToolCall = interactiveToolCalls[0];
1102
- if (!interactiveToolCall) {
1103
- throw new Error("No interactive tool call found");
1104
- }
1105
- step.partialToolResults = toolResults;
1106
- step.pendingToolCalls = interactiveToolCalls;
1107
- return callInteractiveTool(setting, checkpoint, {
1108
- newMessage: checkpoint.messages[checkpoint.messages.length - 1],
1109
- toolCall: interactiveToolCall,
1110
- usage: step.usage
1111
- });
1112
- }
1113
- return resolveToolResults(setting, checkpoint, { toolResults });
1114
- }
1115
- async function finishingStepLogic({
1116
- setting,
1117
- checkpoint,
1118
- step
1119
- }) {
1120
- if (setting.maxSteps !== void 0 && checkpoint.stepNumber >= setting.maxSteps) {
1121
- return stopRunByExceededMaxSteps(setting, checkpoint, {
1122
- checkpoint: {
1123
- ...checkpoint,
1124
- status: "stoppedByExceededMaxSteps"
1125
- },
1126
- step: {
1127
- ...step,
1128
- finishedAt: Date.now()
1129
- }
1130
- });
1131
- }
1132
- return continueToNextStep(setting, checkpoint, {
1133
- checkpoint: {
1134
- ...checkpoint
1135
- },
1136
- step: {
1137
- ...step,
1138
- finishedAt: Date.now()
1139
- },
1140
- nextCheckpoint: {
1141
- ...checkpoint,
1142
- id: createId(),
1143
- stepNumber: checkpoint.stepNumber + 1
1144
- }
1145
- });
1146
- }
1147
- function createUserMessage(contents) {
1148
- return {
1149
- type: "userMessage",
1150
- contents: contents.map((part) => ({
1151
- ...part,
1152
- id: createId()
1153
- })),
1154
- id: createId()
1155
- };
1156
- }
1157
- function createExpertMessage(contents) {
1158
- return {
1159
- type: "expertMessage",
1160
- contents: contents.map((part) => ({
1161
- ...part,
1162
- id: createId()
1163
- })),
1164
- id: createId()
1165
- };
1166
- }
1167
- function createToolMessage(contents) {
1168
- return {
1169
- type: "toolMessage",
1170
- contents: contents.map((part) => ({
1171
- ...part,
1172
- contents: part.contents.map((part2) => ({
1173
- ...part2,
1174
- id: createId()
1175
- })),
1176
- id: createId()
1177
- })),
1178
- id: createId()
1179
- };
1180
- }
1181
- function messageToCoreMessage(message) {
1182
- switch (message.type) {
1183
- case "instructionMessage":
1184
- return {
1185
- role: "system",
1186
- content: instructionContentsToCoreContent(message.contents),
1187
- providerOptions: message.cache ? {
1188
- anthropic: { cacheControl: { type: "ephemeral" } }
1189
- } : void 0
1190
- };
1191
- case "userMessage":
1192
31
  return {
1193
- role: "user",
1194
- content: userContentsToCoreContent(message.contents),
1195
- providerOptions: message.cache ? {
1196
- anthropic: { cacheControl: { type: "ephemeral" } }
1197
- } : void 0
1198
- };
1199
- case "expertMessage":
1200
- return {
1201
- role: "assistant",
1202
- content: expertContentsToCoreContent(message.contents),
1203
- providerOptions: message.cache ? {
1204
- anthropic: { cacheControl: { type: "ephemeral" } }
1205
- } : void 0
1206
- };
1207
- case "toolMessage":
1208
- return {
1209
- role: "tool",
1210
- content: toolContentsToCoreContent(message.contents),
1211
- providerOptions: message.cache ? {
1212
- anthropic: { cacheControl: { type: "ephemeral" } }
1213
- } : void 0
1214
- };
1215
- }
1216
- }
1217
- function instructionContentsToCoreContent(contents) {
1218
- return contents.reduce((acc, part) => {
1219
- return dedent`
1220
- ${acc}
1221
- ${part.text}
1222
- `.trim();
1223
- }, "");
1224
- }
1225
- function userContentsToCoreContent(contents) {
1226
- return contents.map((part) => {
1227
- switch (part.type) {
1228
- case "textPart":
1229
- return textPartToCoreTextPart(part);
1230
- case "imageUrlPart":
1231
- return imageUrlPartToCoreImagePart(part);
1232
- case "imageInlinePart":
1233
- return imageInlinePartToCoreImagePart(part);
1234
- case "imageBinaryPart":
1235
- return imageBinaryPartToCoreImagePart(part);
1236
- case "fileUrlPart":
1237
- return fileUrlPartToCoreFilePart(part);
1238
- case "fileInlinePart":
1239
- return fileInlinePartToCoreFilePart(part);
1240
- case "fileBinaryPart":
1241
- return fileBinaryPartToCoreFilePart(part);
1242
- }
1243
- });
1244
- }
1245
- function expertContentsToCoreContent(contents) {
1246
- return contents.map((part) => {
1247
- switch (part.type) {
1248
- case "textPart":
1249
- return textPartToCoreTextPart(part);
1250
- case "toolCallPart":
1251
- return toolCallPartToCoreToolCallPart(part);
1252
- }
1253
- });
1254
- }
1255
- function toolContentsToCoreContent(contents) {
1256
- return contents.map((part) => {
1257
- switch (part.type) {
1258
- case "toolResultPart":
1259
- return toolResultPartToCoreToolResultPart(part);
1260
- }
1261
- });
1262
- }
1263
- function textPartToCoreTextPart(part) {
1264
- return {
1265
- type: "text",
1266
- text: part.text
1267
- };
1268
- }
1269
- function imageUrlPartToCoreImagePart(part) {
1270
- return {
1271
- type: "image",
1272
- image: part.url,
1273
- mediaType: part.mimeType
1274
- };
1275
- }
1276
- function imageInlinePartToCoreImagePart(part) {
1277
- return {
1278
- type: "image",
1279
- image: part.encodedData,
1280
- mediaType: part.mimeType
1281
- };
1282
- }
1283
- function imageBinaryPartToCoreImagePart(part) {
1284
- return {
1285
- type: "image",
1286
- image: part.data,
1287
- mediaType: part.mimeType
1288
- };
1289
- }
1290
- function fileUrlPartToCoreFilePart(part) {
1291
- return {
1292
- type: "file",
1293
- data: part.url,
1294
- mediaType: part.mimeType
1295
- };
1296
- }
1297
- function fileInlinePartToCoreFilePart(part) {
1298
- return {
1299
- type: "file",
1300
- data: part.encodedData,
1301
- mediaType: part.mimeType
1302
- };
1303
- }
1304
- function fileBinaryPartToCoreFilePart(part) {
1305
- return {
1306
- type: "file",
1307
- data: part.data,
1308
- mediaType: part.mimeType
1309
- };
1310
- }
1311
- function toolCallPartToCoreToolCallPart(part) {
1312
- return {
1313
- type: "tool-call",
1314
- toolCallId: part.toolCallId,
1315
- toolName: part.toolName,
1316
- input: part.args
1317
- };
1318
- }
1319
- function toolResultPartToCoreToolResultPart(part) {
1320
- const { contents } = part;
1321
- if (contents.length === 1 && contents[0].type === "textPart") {
1322
- return {
1323
- type: "tool-result",
1324
- toolCallId: part.toolCallId,
1325
- toolName: part.toolName,
1326
- output: { type: "text", value: contents[0].text }
1327
- };
1328
- }
1329
- const contentValue = contents.map((content) => {
1330
- if (content.type === "textPart") {
1331
- return { type: "text", text: content.text };
1332
- }
1333
- return { type: "media", data: content.encodedData, mediaType: content.mimeType };
1334
- });
1335
- return {
1336
- type: "tool-result",
1337
- toolCallId: part.toolCallId,
1338
- toolName: part.toolName,
1339
- output: { type: "content", value: contentValue }
1340
- };
1341
- }
1342
-
1343
- // src/states/generating-run-result.ts
1344
- async function generatingRunResultLogic({
1345
- setting,
1346
- checkpoint,
1347
- step
1348
- }) {
1349
- if (!step.toolCalls || !step.toolResults || step.toolResults.length === 0) {
1350
- throw new Error("No tool calls or tool results found");
1351
- }
1352
- const toolResultParts = step.toolResults.map((toolResult) => {
1353
- const toolCall = step.toolCalls?.find((tc) => tc.id === toolResult.id);
1354
- return {
1355
- type: "toolResultPart",
1356
- toolCallId: toolResult.id,
1357
- toolName: toolCall?.toolName ?? toolResult.toolName,
1358
- contents: toolResult.result.filter(
1359
- (part) => part.type === "textPart" || part.type === "imageInlinePart" || part.type === "fileInlinePart"
1360
- )
1361
- };
1362
- });
1363
- const toolMessage = createToolMessage(toolResultParts);
1364
- const model = getModel(setting.model, setting.providerConfig);
1365
- const { messages } = checkpoint;
1366
- let generationResult;
1367
- try {
1368
- generationResult = await generateText({
1369
- model,
1370
- messages: [...messages, toolMessage].map(messageToCoreMessage),
1371
- temperature: setting.temperature,
1372
- maxRetries: setting.maxRetries,
1373
- abortSignal: AbortSignal.timeout(setting.timeout)
1374
- });
1375
- } catch (error) {
1376
- if (error instanceof Error) {
1377
- const reason = JSON.stringify({ error: error.name, message: error.message });
1378
- return retry(setting, checkpoint, {
1379
- reason,
1380
- newMessages: [toolMessage, createUserMessage([{ type: "textPart", text: reason }])],
1381
- usage: createEmptyUsage()
1382
- });
1383
- }
1384
- throw error;
1385
- }
1386
- const usage = usageFromGenerateTextResult(generationResult);
1387
- const { text } = generationResult;
1388
- const newMessages = [toolMessage, createExpertMessage(text ? [{ type: "textPart", text }] : [])];
1389
- return completeRun(setting, checkpoint, {
1390
- checkpoint: {
1391
- ...checkpoint,
1392
- messages: [...messages, ...newMessages],
1393
- usage: sumUsage(checkpoint.usage, usage),
1394
- contextWindowUsage: checkpoint.contextWindow ? calculateContextWindowUsage(usage, checkpoint.contextWindow) : void 0,
1395
- status: "completed"
1396
- },
1397
- step: {
1398
- ...step,
1399
- newMessages: [...step.newMessages, ...newMessages],
1400
- finishedAt: Date.now(),
1401
- usage: sumUsage(step.usage, usage)
1402
- },
1403
- text,
1404
- usage
1405
- });
1406
- }
1407
- async function classifyToolCalls(toolCalls, skillManagers) {
1408
- return Promise.all(
1409
- toolCalls.map(async (tc) => {
1410
- const skillManager = await getSkillManagerByToolName(skillManagers, tc.toolName);
1411
- return {
1412
- toolCallId: tc.toolCallId,
1413
- toolName: tc.toolName,
1414
- input: tc.input,
1415
- skillManager
1416
- };
1417
- })
1418
- );
1419
- }
1420
- function sortToolCallsByPriority(toolCalls) {
1421
- const priority = { mcp: 0, delegate: 1, interactive: 2 };
1422
- return [...toolCalls].sort(
1423
- (a, b) => (priority[a.skillManager.type] ?? 99) - (priority[b.skillManager.type] ?? 99)
1424
- );
1425
- }
1426
- function buildToolCallParts(toolCalls) {
1427
- return toolCalls.map((tc) => ({
1428
- type: "toolCallPart",
1429
- toolCallId: tc.toolCallId,
1430
- toolName: tc.toolName,
1431
- args: tc.input
1432
- }));
1433
- }
1434
- function buildToolCalls(toolCalls) {
1435
- return toolCalls.map((tc) => ({
1436
- id: tc.toolCallId,
1437
- skillName: tc.skillManager.name,
1438
- toolName: tc.toolName,
1439
- args: tc.input
1440
- }));
1441
- }
1442
- async function generatingToolCallLogic({
1443
- setting,
1444
- checkpoint,
1445
- skillManagers
1446
- }) {
1447
- const { messages } = checkpoint;
1448
- const model = getModel(setting.model, setting.providerConfig);
1449
- let result;
1450
- try {
1451
- result = await generateText({
1452
- model,
1453
- messages: messages.map(messageToCoreMessage),
1454
- temperature: setting.temperature,
1455
- maxRetries: setting.maxRetries,
1456
- tools: await getToolSet(skillManagers),
1457
- toolChoice: "required",
1458
- abortSignal: AbortSignal.timeout(setting.timeout)
1459
- });
1460
- } catch (error) {
1461
- if (error instanceof Error) {
1462
- const reason = JSON.stringify({ error: error.name, message: error.message });
1463
- return retry(setting, checkpoint, {
1464
- reason,
1465
- newMessages: [createUserMessage([{ type: "textPart", text: reason }])],
1466
- usage: createEmptyUsage()
1467
- });
1468
- }
1469
- throw error;
1470
- }
1471
- const usage = usageFromGenerateTextResult(result);
1472
- const { text, toolCalls, finishReason } = result;
1473
- if (toolCalls.length === 0) {
1474
- const reason = JSON.stringify({
1475
- error: "Error: No tool call generated",
1476
- message: "You must generate a tool call. Try again."
1477
- });
1478
- return retry(setting, checkpoint, {
1479
- reason,
1480
- newMessages: [createUserMessage([{ type: "textPart", text: reason }])],
1481
- usage
1482
- });
1483
- }
1484
- const classified = await classifyToolCalls(toolCalls, skillManagers);
1485
- const sorted = sortToolCallsByPriority(classified);
1486
- if (finishReason === "tool-calls" || finishReason === "stop") {
1487
- const toolCallParts = buildToolCallParts(sorted);
1488
- const contents = [...toolCallParts];
1489
- if (text) {
1490
- contents.push({ type: "textPart", text });
1491
- }
1492
- const allToolCalls = buildToolCalls(sorted);
1493
- return callTools(setting, checkpoint, {
1494
- newMessage: createExpertMessage(contents),
1495
- toolCalls: allToolCalls,
1496
- usage
1497
- });
1498
- }
1499
- if (finishReason === "length") {
1500
- const firstToolCall = sorted[0];
1501
- if (!firstToolCall) {
1502
- throw new Error("No tool call found");
1503
- }
1504
- const reason = JSON.stringify({
1505
- error: "Error: Tool call generation failed",
1506
- message: "Generation length exceeded. Try again."
1507
- });
1508
- return retry(setting, checkpoint, {
1509
- reason,
1510
- newMessages: [
1511
- createExpertMessage([
1512
- {
1513
- type: "toolCallPart",
1514
- toolCallId: firstToolCall.toolCallId,
1515
- toolName: firstToolCall.toolName,
1516
- args: firstToolCall.input
1517
- }
1518
- ]),
1519
- createToolMessage([
1520
- {
1521
- type: "toolResultPart",
1522
- toolCallId: firstToolCall.toolCallId,
1523
- toolName: firstToolCall.toolName,
1524
- contents: [{ type: "textPart", text: reason }]
1525
- }
1526
- ])
1527
- ],
1528
- toolCalls: [
1529
- {
1530
- id: firstToolCall.toolCallId,
1531
- skillName: firstToolCall.skillManager.name,
1532
- toolName: firstToolCall.toolName,
1533
- args: firstToolCall.input
1534
- }
1535
- ],
1536
- toolResults: [
1537
- {
1538
- id: firstToolCall.toolCallId,
1539
- skillName: firstToolCall.skillManager.name,
1540
- toolName: firstToolCall.toolName,
1541
- result: [{ type: "textPart", id: createId(), text: reason }]
32
+ ok: false,
33
+ error: {
34
+ type: "cli-not-found",
35
+ message: "perstack-runtime CLI is not available."
1542
36
  }
1543
- ],
1544
- usage
1545
- });
1546
- }
1547
- throw new Error(`Unexpected finish reason: ${finishReason}`);
1548
- }
1549
- function getMetaInstruction(startedAt) {
1550
- return dedent`
1551
- IMPORTANT:
1552
- Based on the user's initial message, you must determine what needs to be done.
1553
- You must iterate through hypothesis and verification to fulfill the task.
1554
- YOU MUST CONTINUE TO CALL TOOLS UNTIL THE TASK IS COMPLETE.
1555
- If you do not call tools, the task will be considered complete, and the agent loop will end.
1556
-
1557
- You operate in an agent loop, iteratively completing tasks through these steps:
1558
- 1. Analyze Events: Understand user needs and current state through the event stream, focusing on the latest user messages and execution results
1559
- 2. Select Tools: Choose the next tool call based on current state, task planning, relevant knowledge, and available data APIs
1560
- 3. Wait for Execution: The selected tool action will be executed by the sandbox environment with new observations added to the event stream
1561
- 4. Iterate: Choose only one tool call per iteration, patiently repeat the above steps until task completion
1562
- 5. Notify Task Completion: Call the attemptCompletion tool to inform the user when the task is complete
1563
- 6. Generate Final Results: Produce a final result that clearly describes each task you performed, step by step
1564
-
1565
- Conditions for ending the agent loop:
1566
- If any of the following apply, **immediately call the attemptCompletion tool**.
1567
- When the agent loop must end, calling any tool other than attemptCompletion is highly dangerous.
1568
- Under all circumstances, strictly follow this rule.
1569
- - When the task is complete
1570
- - When the user's request is outside your expertise
1571
- - When the user's request is unintelligible
1572
-
1573
- Rules for requests outside your area of expertise:
1574
- - Tell your area of expertise to the user in final results
1575
-
1576
- Environment information:
1577
- - Current time is ${new Date(startedAt).toISOString()}
1578
- - Current working directory is ${process.cwd()}
1579
- `;
1580
- }
1581
- function createInstructionMessage(expert, experts, startedAt) {
1582
- const instruction = dedent`
1583
- You are Perstack, an AI expert that tackles tasks requested by users by utilizing all available tools.
1584
-
1585
- (The following information describes your nature and role as an AI, the mechanisms of the AI system, and other meta-cognitive aspects.)
1586
-
1587
- ${getMetaInstruction(startedAt)}
1588
-
1589
- ---
1590
- (The following describes the objective, steps, rules, etc. regarding your expert task.)
1591
-
1592
- ${expert.instruction}
1593
-
1594
- ---
1595
- (The following is an overview of each skill and the rules for calling tools.)
1596
-
1597
- ${getSkillRules(expert)}
1598
-
1599
- ---
1600
- (The following is an overview of each delegate expert and the rules for calling tools.)
1601
-
1602
- You can delegate tasks to the following experts by calling delegate expert name as a tool:
1603
-
1604
- ${getDelegateRules(expert, experts)}
1605
- `;
1606
- return {
1607
- type: "instructionMessage",
1608
- contents: [
1609
- {
1610
- id: createId(),
1611
- type: "textPart",
1612
- text: instruction
1613
- }
1614
- ],
1615
- id: createId(),
1616
- cache: true
1617
- };
1618
- }
1619
- function getSkillRules(expert) {
1620
- return Object.values(expert.skills).reduce((acc, skill) => {
1621
- if (!skill.rule) {
1622
- return acc;
1623
- }
1624
- return dedent`
1625
- ${acc}
1626
-
1627
- "${skill.name}" skill rules:
1628
- ${skill.rule}
1629
- `.trim();
1630
- }, "");
1631
- }
1632
- function getDelegateRules(expert, experts) {
1633
- return expert.delegates.reduce((acc, delegateExpertName) => {
1634
- const delegate = experts[delegateExpertName];
1635
- if (!delegate) {
1636
- return acc;
1637
- }
1638
- return dedent`
1639
- ${acc}
1640
-
1641
- About "${delegate.name}":
1642
- ${delegate.description}
1643
- `.trim();
1644
- }, "");
1645
- }
1646
-
1647
- // src/states/init.ts
1648
- async function initLogic({
1649
- setting,
1650
- checkpoint
1651
- }) {
1652
- const { expertKey, experts } = setting;
1653
- const expert = experts[expertKey];
1654
- switch (checkpoint.status) {
1655
- case "init": {
1656
- if (!setting.input.text) {
1657
- throw new Error("Input message is undefined");
1658
- }
1659
- return startRun(setting, checkpoint, {
1660
- initialCheckpoint: checkpoint,
1661
- inputMessages: [
1662
- createInstructionMessage(expert, experts, setting.startedAt),
1663
- createUserMessage([{ type: "textPart", text: setting.input.text }])
1664
- ]
1665
- });
1666
- }
1667
- case "stoppedByDelegate":
1668
- case "stoppedByInteractiveTool": {
1669
- if (!setting.input.interactiveToolCallResult) {
1670
- throw new Error("Interactive tool call result is undefined");
1671
- }
1672
- const { toolCallId, toolName, skillName, text } = setting.input.interactiveToolCallResult;
1673
- const pendingToolCalls = checkpoint.pendingToolCalls ?? [];
1674
- const newToolResult = {
1675
- id: toolCallId,
1676
- skillName,
1677
- toolName,
1678
- result: [{ type: "textPart", id: createId(), text }]
1679
- };
1680
- const updatedPartialResults = [...checkpoint.partialToolResults ?? [], newToolResult];
1681
- const updatedPendingToolCalls = pendingToolCalls.filter((tc) => tc.id !== toolCallId);
1682
- const updatedCheckpoint = {
1683
- ...checkpoint,
1684
- partialToolResults: updatedPartialResults,
1685
- pendingToolCalls: updatedPendingToolCalls.length > 0 ? updatedPendingToolCalls : void 0
1686
37
  };
1687
- return startRun(setting, updatedCheckpoint, {
1688
- initialCheckpoint: updatedCheckpoint,
1689
- inputMessages: []
1690
- });
1691
38
  }
1692
- default:
1693
- if (!setting.input.text) {
1694
- throw new Error("Input message is undefined");
1695
- }
1696
- return startRun(setting, checkpoint, {
1697
- initialCheckpoint: checkpoint,
1698
- inputMessages: [createUserMessage([{ type: "textPart", text: setting.input.text }])]
1699
- });
39
+ return { ok: true };
1700
40
  }
1701
- }
1702
- async function preparingForStepLogic({
1703
- setting,
1704
- checkpoint
1705
- }) {
1706
- if (checkpoint.pendingToolCalls && checkpoint.pendingToolCalls.length > 0) {
1707
- return resumeToolCalls(setting, checkpoint, {
1708
- pendingToolCalls: checkpoint.pendingToolCalls,
1709
- partialToolResults: checkpoint.partialToolResults ?? []
1710
- });
1711
- }
1712
- if (checkpoint.partialToolResults && checkpoint.partialToolResults.length > 0) {
1713
- const toolResultParts = checkpoint.partialToolResults.map((tr) => ({
1714
- type: "toolResultPart",
1715
- toolCallId: tr.id,
1716
- toolName: tr.toolName,
1717
- contents: tr.result.filter(
1718
- (part) => part.type === "textPart" || part.type === "imageInlinePart" || part.type === "fileInlinePart"
1719
- )
1720
- }));
1721
- return finishAllToolCalls(setting, checkpoint, {
1722
- newMessages: [createToolMessage(toolResultParts)]
1723
- });
41
+ convertExpert(expert) {
42
+ return { instruction: expert.instruction };
1724
43
  }
1725
- return startGeneration(setting, checkpoint, {
1726
- messages: checkpoint.messages
1727
- });
1728
- }
1729
- async function resolvingToolResultLogic({
1730
- setting,
1731
- checkpoint,
1732
- step
1733
- }) {
1734
- if (!step.toolCalls || !step.toolResults || step.toolResults.length === 0) {
1735
- throw new Error("No tool calls or tool results found");
44
+ async run(params) {
45
+ if (this.useDirectExecution) {
46
+ return this.runDirect(params);
47
+ }
48
+ return this.runViaCli(params);
1736
49
  }
1737
- const toolResultParts = step.toolResults.map((toolResult) => {
1738
- const toolCall = step.toolCalls?.find((tc) => tc.id === toolResult.id);
1739
- return {
1740
- type: "toolResultPart",
1741
- toolCallId: toolResult.id,
1742
- toolName: toolCall?.toolName ?? toolResult.toolName,
1743
- contents: toolResult.result.filter(
1744
- (part) => part.type === "textPart" || part.type === "imageInlinePart" || part.type === "fileInlinePart"
1745
- )
50
+ async runDirect(params) {
51
+ const events = [];
52
+ const eventListener = (event) => {
53
+ events.push(event);
54
+ params.eventListener?.(event);
1746
55
  };
1747
- });
1748
- return finishToolCall(setting, checkpoint, {
1749
- newMessages: [createToolMessage(toolResultParts)]
1750
- });
1751
- }
1752
-
1753
- // src/states/resolving-thought.ts
1754
- async function resolvingThoughtLogic(context) {
1755
- return resolvingToolResultLogic(context);
1756
- }
1757
-
1758
- // src/runtime-state-machine.ts
1759
- var runtimeStateMachine = setup({
1760
- types: {
1761
- input: {},
1762
- context: {},
1763
- events: {}
1764
- }
1765
- }).createMachine({
1766
- /** @xstate-layout N4IgpgJg5mDOIC5QCUCuA7AdASXQSwBcBiWAgQwCcC10BtABgF1FQAHAe1kL3fRZAAeiAKz0AzJgDsAJgAsksZNlixs+gDZZAGhABPRIszCAjGIAcY4crPHhJyQF8HOmpgAKFMK0p50UAGLsFADKBF4k5FQA4mDoYBRkBDx0TPwcXEm8-EIIZmaymPRK8saSopL00tI6+gjq9JjSltLG9HnqZkqdTi4YmDFxCUl+ACrs7AA2AMJkExNEngQUugzMSCDp3FnrOcam0pgAnHbS9GeSZsIq6jWIh2KHUsYd9MbSwl2S6j0grgPxiV8UDGkxmcyIAGNZhMQRNVmlOFs+DtEPt1Jg3odpGVbFcrLcENIzOjbGYipIZMpnmYfn9YgDhsDxtNoZDobgwgkIUkAG5gWHw9abTLI0C7ImSTDqYTY2S45TCCwE54HUr1WRXSqHcRVWl9f5DIGwsHzKFzAAiYAmYCgiTAgrYiJF2VRYnoj1ehwskkOZTEz2EBMMJnMFnezxUF2+zl+fRNRuZCzgkz5sOQcFQEwIDo2TuSLoQpWJUvowl9Flk0sqBLlZik2Mkpjyan90d6WHjo0TnlgKf5AAt2KgoP3s6khXntmLUT6JL6PkoFJX1MZtHoRKIpdrK3sqorpG3Yx3oQnJknexM+W4IAAzfx4a054X5lGFn2PMTYlSyWQH32WAl1DKKV1Q6L1V1EWQ9WPOZT3mHs+2wABbMgYHvR9x0dDIX2nN8zEeOVjGORsmgsMQCTJYRMGJORwP9MwDxpGNXE7Jkz0SMIkNYAgpnYLjrRFJ9J1FQQZzJTBPxUfDDjlN1OgJIkSUVUtDlXCpsUPVx0wvHk4O0zNiBvXw8FgftjWhITsKnUTCU-SU92JMlfSaGtjDrGQKSbfJxGeaDMG0lMjUHYdRyIIz8FM8y5kspECyabFGneRz3RkN011qOwGnUbcVzeJKDz8gLLyBa87wfMAwuMyLmRNGLnVfeL7KSl5nPI9c6mA9RQPwmwNVLQrk2KvxkNQsB0Iq8KTLMmqLMw3MrJEnJGsSxUWtSgkfSU-JDmIqNKj8g1AT8Gh9KzSE+NYASwBoOqcJs+K61sDo5SKBifwU4tSRUtTKi+KDmLjE9hvQTkyG5PBU0TUh2FYGgACFdA5AFwchyZbus3YyklYRst-DUlDEaU2tqFUMS+NyPmlPZDiAvzWMta1bTCCIYfh3QGZtO10cWmcLiOQnG3qHHiSDbGvM-Ex1EjYk-PvCL+yBUJwghXhhlQfl2AAOTAAQCCV1hubiqxjCMRUZXMGSvVUZV6A1LcuorDp8I0WWqoVvx9ZZ2GMARgBRAQITASBIAAWTIAR9dgQ2Gs0Ki8nyK4bCuDVRZNzRwxse4GK6pwY3QdgIDgfgaARBaCxUBosU0cw5TLUiCQAWlkR4nPdbVCeyld-vbHB8AIUvYtfRQ6yr6xa6xcwCTLaiadj1dKxk1phD8jwvB8PxAhCMJWAH+rcKAyuyVaa5Wi+KeW9njV59xpeDvpQ0u1BaFd7u3Y2keDpO7LW2FDLc+Z6AnsPEO4ZYAxghMOCL8MaojsJKYi8cHZ5C9EGNoRg1CdH3GcZeYD-KDV0o-CYp1+4TjLg1cQkp-SfjDE9Ew1R2o-hNnsUs1MfR-UuANHSQUhwjmIVhQeuFTg-ilF8GUblTgKDoRlBQhQJEaiJMnfCHDAp+FKuNKBPNCSlgaOIeQttWxS0DO1bKdY9HvH9D+FotMcFFXwVAEaaFyrqLirbBo1M7KvFUDJCiRJGgalUM3LEMk5R30GEdKAJ0MxZicWQlQEklAyillWVoZgUF1isG0MoDEsF0yBnYkGyNeQa0mNE3CtgiIYnEEBToFwvS+mVPkU20l8JWG1OIHJsE-AcyZmAEpNlbBqEwLo7yVRbY3HatPfCXUOgfCsGSTQmk+hyymorbevSloykeFYCkrxfqrmJiIaRRQ7JbP8Q8PyoQYasEgGsxA2JWiNB-jjdQTRqb1IkP6OwRF8KMSAbnBwQA */
1767
- id: "Run",
1768
- initial: "Init",
1769
- context: ({ input }) => ({
1770
- setting: input.setting,
1771
- checkpoint: input.initialCheckpoint,
1772
- step: {
1773
- stepNumber: input.initialCheckpoint.stepNumber,
1774
- inputMessages: [],
1775
- newMessages: [],
1776
- usage: createEmptyUsage(),
1777
- startedAt: Date.now()
1778
- },
1779
- eventListener: input.eventListener,
1780
- skillManagers: input.skillManagers
1781
- }),
1782
- states: {
1783
- Init: {
1784
- on: {
1785
- startRun: {
1786
- target: "PreparingForStep",
1787
- actions: assign({
1788
- checkpoint: ({ context, event }) => ({
1789
- ...context.checkpoint,
1790
- status: "proceeding",
1791
- messages: [...context.checkpoint.messages, ...event.inputMessages],
1792
- pendingToolCalls: event.initialCheckpoint.pendingToolCalls,
1793
- partialToolResults: event.initialCheckpoint.partialToolResults
1794
- }),
1795
- step: ({ context, event }) => ({
1796
- ...context.step,
1797
- inputMessages: event.inputMessages
1798
- })
1799
- })
1800
- }
1801
- }
1802
- },
1803
- PreparingForStep: {
1804
- on: {
1805
- startGeneration: {
1806
- target: "GeneratingToolCall",
1807
- actions: assign({
1808
- step: ({ context, event }) => ({
1809
- stepNumber: context.checkpoint.stepNumber,
1810
- inputMessages: context.step.inputMessages ?? [],
1811
- newMessages: [],
1812
- usage: createEmptyUsage(),
1813
- startedAt: event.timestamp
1814
- })
1815
- })
1816
- },
1817
- resumeToolCalls: {
1818
- target: "CallingTool",
1819
- actions: assign({
1820
- step: ({ context, event }) => ({
1821
- stepNumber: context.checkpoint.stepNumber,
1822
- inputMessages: context.step.inputMessages ?? [],
1823
- newMessages: context.step.newMessages,
1824
- toolCalls: context.step.toolCalls,
1825
- toolResults: event.partialToolResults,
1826
- pendingToolCalls: event.pendingToolCalls,
1827
- usage: context.step.usage,
1828
- startedAt: context.step.startedAt
1829
- })
1830
- })
1831
- },
1832
- finishAllToolCalls: {
1833
- target: "FinishingStep",
1834
- actions: assign({
1835
- checkpoint: ({ context, event }) => ({
1836
- ...context.checkpoint,
1837
- messages: [...context.checkpoint.messages, ...event.newMessages],
1838
- pendingToolCalls: void 0,
1839
- partialToolResults: void 0
1840
- }),
1841
- step: ({ context, event }) => ({
1842
- ...context.step,
1843
- newMessages: [...context.step.newMessages, ...event.newMessages],
1844
- toolResults: context.checkpoint.partialToolResults,
1845
- pendingToolCalls: void 0
1846
- })
1847
- })
1848
- }
1849
- }
1850
- },
1851
- GeneratingToolCall: {
1852
- on: {
1853
- retry: {
1854
- target: "FinishingStep",
1855
- actions: assign({
1856
- checkpoint: ({ context, event }) => ({
1857
- ...context.checkpoint,
1858
- messages: [...context.checkpoint.messages, ...event.newMessages],
1859
- usage: sumUsage(context.checkpoint.usage, event.usage)
1860
- }),
1861
- step: ({ context, event }) => ({
1862
- ...context.step,
1863
- newMessages: event.newMessages,
1864
- toolCalls: event.toolCalls,
1865
- toolResults: event.toolResults,
1866
- usage: sumUsage(context.step.usage, event.usage)
1867
- })
1868
- })
1869
- },
1870
- callTools: {
1871
- target: "CallingTool",
1872
- actions: assign({
1873
- checkpoint: ({ context, event }) => ({
1874
- ...context.checkpoint,
1875
- messages: [...context.checkpoint.messages, event.newMessage],
1876
- usage: sumUsage(context.checkpoint.usage, event.usage),
1877
- contextWindowUsage: context.checkpoint.contextWindow ? calculateContextWindowUsage(event.usage, context.checkpoint.contextWindow) : void 0
1878
- }),
1879
- step: ({ context, event }) => ({
1880
- ...context.step,
1881
- newMessages: [event.newMessage],
1882
- toolCalls: event.toolCalls,
1883
- usage: sumUsage(context.step.usage, event.usage)
1884
- })
1885
- })
1886
- },
1887
- callInteractiveTool: {
1888
- target: "CallingInteractiveTool",
1889
- actions: assign({
1890
- checkpoint: ({ context, event }) => ({
1891
- ...context.checkpoint,
1892
- messages: [...context.checkpoint.messages, event.newMessage],
1893
- usage: sumUsage(context.checkpoint.usage, event.usage),
1894
- contextWindowUsage: context.checkpoint.contextWindow ? calculateContextWindowUsage(event.usage, context.checkpoint.contextWindow) : void 0
1895
- }),
1896
- step: ({ context, event }) => ({
1897
- ...context.step,
1898
- newMessages: [event.newMessage],
1899
- toolCalls: [event.toolCall],
1900
- usage: sumUsage(context.step.usage, event.usage)
1901
- })
1902
- })
1903
- },
1904
- callDelegate: {
1905
- target: "CallingDelegate",
1906
- actions: assign({
1907
- checkpoint: ({ context, event }) => ({
1908
- ...context.checkpoint,
1909
- messages: [...context.checkpoint.messages, event.newMessage],
1910
- usage: sumUsage(context.checkpoint.usage, event.usage),
1911
- contextWindowUsage: context.checkpoint.contextWindow ? calculateContextWindowUsage(event.usage, context.checkpoint.contextWindow) : void 0
1912
- }),
1913
- step: ({ context, event }) => ({
1914
- ...context.step,
1915
- newMessages: [event.newMessage],
1916
- toolCalls: event.toolCalls,
1917
- usage: sumUsage(context.step.usage, event.usage)
1918
- })
1919
- })
1920
- }
1921
- }
1922
- },
1923
- CallingTool: {
1924
- on: {
1925
- resolveToolResults: {
1926
- target: "ResolvingToolResult",
1927
- actions: assign({
1928
- step: ({ context, event }) => ({
1929
- ...context.step,
1930
- toolResults: event.toolResults,
1931
- pendingToolCalls: void 0
1932
- })
1933
- })
1934
- },
1935
- resolveThought: {
1936
- target: "ResolvingThought",
1937
- actions: assign({
1938
- step: ({ context, event }) => ({
1939
- ...context.step,
1940
- toolResults: [event.toolResult]
1941
- })
1942
- })
1943
- },
1944
- attemptCompletion: {
1945
- target: "GeneratingRunResult",
1946
- actions: assign({
1947
- step: ({ context, event }) => ({
1948
- ...context.step,
1949
- toolResults: [event.toolResult]
1950
- })
1951
- })
1952
- },
1953
- callDelegate: {
1954
- target: "CallingDelegate",
1955
- actions: assign({
1956
- step: ({ context }) => ({
1957
- ...context.step,
1958
- toolCalls: context.step.toolCalls,
1959
- toolResults: context.step.toolResults,
1960
- pendingToolCalls: context.step.pendingToolCalls,
1961
- partialToolResults: context.step.partialToolResults
1962
- })
1963
- })
1964
- },
1965
- callInteractiveTool: {
1966
- target: "CallingInteractiveTool",
1967
- actions: assign({
1968
- step: ({ context }) => ({
1969
- ...context.step,
1970
- toolCalls: context.step.toolCalls,
1971
- toolResults: context.step.toolResults,
1972
- pendingToolCalls: context.step.pendingToolCalls,
1973
- partialToolResults: context.step.partialToolResults
1974
- })
1975
- })
1976
- }
1977
- }
1978
- },
1979
- ResolvingToolResult: {
1980
- on: {
1981
- finishToolCall: {
1982
- target: "FinishingStep",
1983
- actions: assign({
1984
- checkpoint: ({ context, event }) => ({
1985
- ...context.checkpoint,
1986
- messages: [...context.checkpoint.messages, ...event.newMessages]
1987
- }),
1988
- step: ({ context, event }) => ({
1989
- ...context.step,
1990
- newMessages: [...context.step.newMessages, ...event.newMessages]
1991
- })
1992
- })
1993
- }
1994
- }
1995
- },
1996
- ResolvingThought: {
1997
- on: {
1998
- finishToolCall: {
1999
- target: "FinishingStep",
2000
- actions: assign({
2001
- checkpoint: ({ context, event }) => ({
2002
- ...context.checkpoint,
2003
- messages: [...context.checkpoint.messages, ...event.newMessages]
2004
- }),
2005
- step: ({ context, event }) => ({
2006
- ...context.step,
2007
- newMessages: [...context.step.newMessages, ...event.newMessages]
2008
- })
2009
- })
2010
- }
2011
- }
2012
- },
2013
- GeneratingRunResult: {
2014
- on: {
2015
- retry: {
2016
- target: "FinishingStep",
2017
- actions: assign({
2018
- checkpoint: ({ context, event }) => ({
2019
- ...context.checkpoint,
2020
- messages: [...context.checkpoint.messages, ...event.newMessages],
2021
- usage: sumUsage(context.checkpoint.usage, event.usage)
2022
- }),
2023
- step: ({ context, event }) => ({
2024
- ...context.step,
2025
- newMessages: event.newMessages,
2026
- toolCalls: event.toolCalls,
2027
- toolResults: event.toolResults,
2028
- usage: sumUsage(context.step.usage, event.usage)
2029
- })
2030
- })
2031
- },
2032
- completeRun: {
2033
- target: "Stopped",
2034
- actions: assign({
2035
- checkpoint: ({ event }) => event.checkpoint,
2036
- step: ({ event }) => ({
2037
- ...event.step,
2038
- inputMessages: void 0
2039
- })
2040
- })
2041
- }
2042
- }
2043
- },
2044
- CallingInteractiveTool: {
2045
- on: {
2046
- stopRunByInteractiveTool: {
2047
- target: "Stopped",
2048
- actions: assign({
2049
- checkpoint: ({ event }) => event.checkpoint,
2050
- step: ({ event }) => ({
2051
- ...event.step,
2052
- inputMessages: void 0
2053
- })
2054
- })
2055
- }
2056
- }
2057
- },
2058
- CallingDelegate: {
2059
- on: {
2060
- stopRunByDelegate: {
2061
- target: "Stopped",
2062
- actions: assign({
2063
- checkpoint: ({ event }) => event.checkpoint,
2064
- step: ({ event }) => ({
2065
- ...event.step,
2066
- inputMessages: void 0
2067
- })
2068
- })
2069
- }
2070
- }
2071
- },
2072
- FinishingStep: {
2073
- on: {
2074
- continueToNextStep: {
2075
- target: "PreparingForStep",
2076
- actions: assign({
2077
- checkpoint: ({ event }) => event.nextCheckpoint,
2078
- step: ({ event }) => ({
2079
- ...event.step,
2080
- inputMessages: void 0
2081
- })
2082
- }),
2083
- reenter: true
2084
- },
2085
- stopRunByExceededMaxSteps: {
2086
- target: "Stopped",
2087
- actions: assign({
2088
- checkpoint: ({ event }) => event.checkpoint,
2089
- step: ({ event }) => ({
2090
- ...event.step,
2091
- inputMessages: void 0
2092
- })
2093
- })
2094
- }
56
+ const checkpoint = await run(
57
+ { setting: params.setting, checkpoint: params.checkpoint },
58
+ {
59
+ eventListener,
60
+ storeCheckpoint: params.storeCheckpoint,
61
+ retrieveCheckpoint: params.retrieveCheckpoint
2095
62
  }
2096
- },
2097
- Stopped: {
2098
- type: "final"
2099
- }
2100
- }
2101
- });
2102
- var StateMachineLogics = {
2103
- Init: initLogic,
2104
- PreparingForStep: preparingForStepLogic,
2105
- GeneratingToolCall: generatingToolCallLogic,
2106
- CallingTool: callingToolLogic,
2107
- ResolvingToolResult: resolvingToolResultLogic,
2108
- ResolvingThought: resolvingThoughtLogic,
2109
- GeneratingRunResult: generatingRunResultLogic,
2110
- CallingInteractiveTool: callingInteractiveToolLogic,
2111
- CallingDelegate: callingDelegateLogic,
2112
- FinishingStep: finishingStepLogic
2113
- };
2114
-
2115
- // src/execute-state-machine.ts
2116
- async function executeStateMachine(params) {
2117
- const {
2118
- setting,
2119
- initialCheckpoint,
2120
- eventListener,
2121
- skillManagers,
2122
- eventEmitter,
2123
- storeCheckpoint,
2124
- shouldContinueRun
2125
- } = params;
2126
- const runActor = createActor(runtimeStateMachine, {
2127
- input: {
2128
- setting,
2129
- initialCheckpoint,
2130
- eventListener,
2131
- skillManagers
2132
- }
2133
- });
2134
- return new Promise((resolve, reject) => {
2135
- runActor.subscribe(async (runState) => {
2136
- try {
2137
- if (runState.value === "Stopped") {
2138
- const { checkpoint, skillManagers: skillManagers2 } = runState.context;
2139
- if (!checkpoint) {
2140
- throw new Error("Checkpoint is undefined");
2141
- }
2142
- await closeSkillManagers(skillManagers2);
2143
- resolve(checkpoint);
2144
- } else {
2145
- const event = await StateMachineLogics[runState.value](runState.context);
2146
- if ("checkpoint" in event) {
2147
- await storeCheckpoint(event.checkpoint);
2148
- }
2149
- await eventEmitter.emit(event);
2150
- if (shouldContinueRun) {
2151
- const shouldContinue = await shouldContinueRun(
2152
- runState.context.setting,
2153
- runState.context.checkpoint,
2154
- runState.context.step
2155
- );
2156
- if (!shouldContinue) {
2157
- runActor.stop();
2158
- await closeSkillManagers(runState.context.skillManagers);
2159
- resolve(runState.context.checkpoint);
2160
- return;
63
+ );
64
+ return { checkpoint, events };
65
+ }
66
+ async runViaCli(params) {
67
+ const { setting, eventListener } = params;
68
+ const events = [];
69
+ const args = this.buildCliArgs(setting);
70
+ const maxSteps = setting.maxSteps ?? 100;
71
+ const processTimeout = (setting.timeout ?? 6e4) * maxSteps;
72
+ const result = await this.executeRuntimeCli(args, processTimeout, (event) => {
73
+ events.push(event);
74
+ eventListener?.(event);
75
+ });
76
+ if (result.exitCode !== 0) {
77
+ throw new Error(`perstack-runtime CLI failed with exit code ${result.exitCode}`);
78
+ }
79
+ const terminalEventTypes = [
80
+ "completeRun",
81
+ "stopRunByInteractiveTool",
82
+ "stopRunByDelegate",
83
+ "stopRunByExceededMaxSteps"
84
+ ];
85
+ const terminalEvent = events.find((e) => terminalEventTypes.includes(e.type));
86
+ if (!terminalEvent?.checkpoint) {
87
+ throw new Error("No terminal event with checkpoint received from CLI");
88
+ }
89
+ return { checkpoint: terminalEvent.checkpoint, events };
90
+ }
91
+ buildCliArgs(setting) {
92
+ const args = ["run"];
93
+ if (setting.jobId) {
94
+ args.push("--job-id", setting.jobId);
95
+ }
96
+ if (setting.runId) {
97
+ args.push("--run-id", setting.runId);
98
+ }
99
+ if (setting.maxSteps !== void 0) {
100
+ args.push("--max-steps", String(setting.maxSteps));
101
+ }
102
+ if (setting.maxRetries !== void 0) {
103
+ args.push("--max-retries", String(setting.maxRetries));
104
+ }
105
+ if (setting.timeout !== void 0) {
106
+ args.push("--timeout", String(setting.timeout));
107
+ }
108
+ if (setting.temperature !== void 0) {
109
+ args.push("--temperature", String(setting.temperature));
110
+ }
111
+ if (setting.model) {
112
+ args.push("--model", setting.model);
113
+ }
114
+ if (setting.providerConfig?.providerName) {
115
+ args.push("--provider", setting.providerConfig.providerName);
116
+ }
117
+ args.push(setting.expertKey, setting.input.text ?? "");
118
+ return args;
119
+ }
120
+ executeRuntimeCli(args, timeout, eventListener) {
121
+ const proc = spawn("perstack-runtime", args, {
122
+ cwd: process.cwd(),
123
+ env: { ...process.env },
124
+ stdio: ["pipe", "pipe", "pipe"]
125
+ });
126
+ proc.stdin.end();
127
+ return this.executeWithStreaming(proc, timeout, eventListener);
128
+ }
129
+ executeWithStreaming(proc, timeout, eventListener) {
130
+ return new Promise((resolve, reject) => {
131
+ let stdout = "";
132
+ let stderr = "";
133
+ let buffer = "";
134
+ const timer = setTimeout(() => {
135
+ proc.kill("SIGTERM");
136
+ reject(new Error(`perstack-runtime timed out after ${timeout}ms`));
137
+ }, timeout);
138
+ proc.stdout?.on("data", (data) => {
139
+ const chunk = data.toString();
140
+ stdout += chunk;
141
+ buffer += chunk;
142
+ const lines = buffer.split("\n");
143
+ buffer = lines.pop() ?? "";
144
+ for (const line of lines) {
145
+ const trimmed = line.trim();
146
+ if (!trimmed) continue;
147
+ try {
148
+ const parsed = JSON.parse(trimmed);
149
+ const terminalEventTypes = [
150
+ "completeRun",
151
+ "stopRunByInteractiveTool",
152
+ "stopRunByDelegate",
153
+ "stopRunByExceededMaxSteps"
154
+ ];
155
+ if (terminalEventTypes.includes(parsed.type) && "checkpoint" in parsed) {
156
+ const checkpointData = parsed.checkpoint;
157
+ parsed.checkpoint = checkpointSchema.parse(checkpointData);
2161
158
  }
159
+ eventListener(parsed);
160
+ } catch {
2162
161
  }
2163
- runActor.send(event);
2164
- }
2165
- } catch (error) {
2166
- await closeSkillManagers(skillManagers).catch(() => {
2167
- });
2168
- reject(error);
2169
- }
2170
- });
2171
- runActor.start();
2172
- });
2173
- }
2174
- async function resolveExpertToRun(expertKey, experts, clientOptions) {
2175
- if (experts[expertKey]) {
2176
- return experts[expertKey];
2177
- }
2178
- const client = new ApiV1Client({
2179
- baseUrl: clientOptions.perstackApiBaseUrl,
2180
- apiKey: clientOptions.perstackApiKey
2181
- });
2182
- const { expert } = await client.registry.experts.get({ expertKey });
2183
- return toRuntimeExpert(expert);
2184
- }
2185
- function toRuntimeExpert(expert) {
2186
- const skills = Object.fromEntries(
2187
- Object.entries(expert.skills).map(([name, skill]) => {
2188
- switch (skill.type) {
2189
- case "mcpStdioSkill":
2190
- return [name, { ...skill, name }];
2191
- case "mcpSseSkill":
2192
- return [name, { ...skill, name }];
2193
- case "interactiveSkill":
2194
- return [name, { ...skill, name }];
2195
- default: {
2196
- throw new Error(`Unknown skill type: ${skill.type}`);
2197
162
  }
2198
- }
2199
- })
2200
- );
2201
- return { ...expert, skills };
2202
- }
2203
-
2204
- // src/setup-experts.ts
2205
- async function setupExperts(setting, resolveExpertToRun2 = resolveExpertToRun) {
2206
- const { expertKey } = setting;
2207
- const experts = { ...setting.experts };
2208
- const clientOptions = {
2209
- perstackApiBaseUrl: setting.perstackApiBaseUrl,
2210
- perstackApiKey: setting.perstackApiKey
2211
- };
2212
- const expertToRun = await resolveExpertToRun2(expertKey, experts, clientOptions);
2213
- experts[expertKey] = expertToRun;
2214
- for (const delegateName of expertToRun.delegates) {
2215
- const delegate = await resolveExpertToRun2(delegateName, experts, clientOptions);
2216
- if (!delegate) {
2217
- throw new Error(`Delegate ${delegateName} not found`);
2218
- }
2219
- experts[delegateName] = delegate;
2220
- }
2221
- return { expertToRun, experts };
2222
- }
2223
-
2224
- // src/runtime.ts
2225
- async function run(runInput, options) {
2226
- const runParams = runParamsSchema.parse(runInput);
2227
- const eventListener = getEventListener(options);
2228
- const retrieveCheckpoint = options?.retrieveCheckpoint ?? defaultRetrieveCheckpoint;
2229
- const storeCheckpoint = options?.storeCheckpoint ?? defaultStoreCheckpoint;
2230
- const eventEmitter = new RunEventEmitter();
2231
- eventEmitter.subscribe(eventListener);
2232
- let { setting, checkpoint } = runParams;
2233
- const contextWindow = getContextWindow(setting.providerConfig.providerName, setting.model);
2234
- const getRunDir = options?.getRunDir ?? defaultGetRunDir;
2235
- await storeRunSetting(setting, options?.fileSystem, getRunDir);
2236
- let job = retrieveJob(setting.jobId) ?? createInitialJob(setting.jobId, setting.expertKey, setting.maxSteps);
2237
- if (job.status !== "running") {
2238
- job = { ...job, status: "running", finishedAt: void 0 };
2239
- }
2240
- storeJob(job);
2241
- while (true) {
2242
- const { expertToRun, experts } = await setupExperts(setting, options?.resolveExpertToRun);
2243
- if (options?.eventListener) {
2244
- const initEvent = createRuntimeEvent("initializeRuntime", setting.jobId, setting.runId, {
2245
- runtimeVersion: package_default.version,
2246
- expertName: expertToRun.name,
2247
- experts: Object.keys(experts),
2248
- model: setting.model,
2249
- temperature: setting.temperature,
2250
- maxSteps: setting.maxSteps,
2251
- maxRetries: setting.maxRetries,
2252
- timeout: setting.timeout,
2253
- query: setting.input.text,
2254
- interactiveToolCall: setting.input.interactiveToolCallResult
2255
163
  });
2256
- options.eventListener(initEvent);
2257
- }
2258
- const skillManagers = await getSkillManagers(
2259
- expertToRun,
2260
- experts,
2261
- setting,
2262
- options?.eventListener,
2263
- { isDelegatedRun: !!checkpoint?.delegatedBy }
2264
- );
2265
- const initialCheckpoint = checkpoint ? createNextStepCheckpoint(createId(), checkpoint) : createInitialCheckpoint(createId(), {
2266
- jobId: setting.jobId,
2267
- runId: setting.runId,
2268
- expertKey: setting.expertKey,
2269
- expert: expertToRun,
2270
- contextWindow
2271
- });
2272
- const runResultCheckpoint = await executeStateMachine({
2273
- setting: { ...setting, experts },
2274
- initialCheckpoint,
2275
- eventListener,
2276
- skillManagers,
2277
- eventEmitter,
2278
- storeCheckpoint,
2279
- shouldContinueRun: options?.shouldContinueRun
164
+ proc.stderr?.on("data", (data) => {
165
+ stderr += data.toString();
166
+ });
167
+ proc.on("close", (code) => {
168
+ clearTimeout(timer);
169
+ resolve({ stdout, stderr, exitCode: code ?? 127 });
170
+ });
171
+ proc.on("error", (err) => {
172
+ clearTimeout(timer);
173
+ reject(err);
174
+ });
2280
175
  });
2281
- job = {
2282
- ...job,
2283
- totalSteps: runResultCheckpoint.stepNumber,
2284
- usage: runResultCheckpoint.usage
2285
- };
2286
- switch (runResultCheckpoint.status) {
2287
- case "completed": {
2288
- if (options?.returnOnDelegationComplete) {
2289
- storeJob(job);
2290
- return runResultCheckpoint;
2291
- }
2292
- if (runResultCheckpoint.delegatedBy) {
2293
- storeJob(job);
2294
- const parentCheckpoint = await retrieveCheckpoint(
2295
- setting.jobId,
2296
- runResultCheckpoint.delegatedBy.checkpointId
2297
- );
2298
- const result = buildDelegationReturnState(setting, runResultCheckpoint, parentCheckpoint);
2299
- setting = result.setting;
2300
- checkpoint = result.checkpoint;
2301
- break;
2302
- }
2303
- storeJob({ ...job, status: "completed", finishedAt: Date.now() });
2304
- return runResultCheckpoint;
2305
- }
2306
- case "stoppedByInteractiveTool": {
2307
- storeJob({ ...job, status: "stoppedByInteractiveTool" });
2308
- return runResultCheckpoint;
2309
- }
2310
- case "stoppedByDelegate": {
2311
- storeJob(job);
2312
- const { delegateTo } = runResultCheckpoint;
2313
- if (!delegateTo || delegateTo.length === 0) {
2314
- throw new Error("No delegations found in checkpoint");
2315
- }
2316
- if (delegateTo.length === 1) {
2317
- const result = buildDelegateToState(setting, runResultCheckpoint, expertToRun);
2318
- setting = result.setting;
2319
- checkpoint = result.checkpoint;
2320
- break;
2321
- }
2322
- const firstDelegation = delegateTo[0];
2323
- const remainingDelegations = delegateTo.slice(1);
2324
- const [firstResult, ...restResults] = await Promise.all(
2325
- delegateTo.map(
2326
- (delegation) => runDelegate(delegation, setting, runResultCheckpoint, expertToRun, options)
2327
- )
2328
- );
2329
- const allResults = [firstResult, ...restResults];
2330
- const aggregatedUsage = allResults.reduce(
2331
- (acc, result) => sumUsage(acc, result.deltaUsage),
2332
- runResultCheckpoint.usage
2333
- );
2334
- const maxStepNumber = Math.max(...allResults.map((r) => r.stepNumber));
2335
- const restToolResults = restResults.map((result) => ({
2336
- id: result.toolCallId,
2337
- skillName: `delegate/${result.expertKey}`,
2338
- toolName: result.toolName,
2339
- result: [{ type: "textPart", id: createId(), text: result.text }]
2340
- }));
2341
- const processedToolCallIds = new Set(remainingDelegations.map((d) => d.toolCallId));
2342
- const remainingToolCalls = runResultCheckpoint.pendingToolCalls?.filter(
2343
- (tc) => !processedToolCallIds.has(tc.id) && tc.id !== firstDelegation.toolCallId
2344
- );
2345
- setting = {
2346
- ...setting,
2347
- expertKey: expertToRun.key,
2348
- input: {
2349
- interactiveToolCallResult: {
2350
- toolCallId: firstResult.toolCallId,
2351
- toolName: firstResult.toolName,
2352
- skillName: `delegate/${firstResult.expertKey}`,
2353
- text: firstResult.text
2354
- }
2355
- }
2356
- };
2357
- checkpoint = {
2358
- ...runResultCheckpoint,
2359
- status: "stoppedByDelegate",
2360
- delegateTo: void 0,
2361
- stepNumber: maxStepNumber,
2362
- usage: aggregatedUsage,
2363
- pendingToolCalls: remainingToolCalls?.length ? remainingToolCalls : void 0,
2364
- partialToolResults: [
2365
- ...runResultCheckpoint.partialToolResults ?? [],
2366
- ...restToolResults
2367
- ]
2368
- };
2369
- break;
2370
- }
2371
- case "stoppedByExceededMaxSteps": {
2372
- storeJob({ ...job, status: "stoppedByMaxSteps", finishedAt: Date.now() });
2373
- return runResultCheckpoint;
2374
- }
2375
- case "stoppedByError": {
2376
- storeJob({ ...job, status: "stoppedByError", finishedAt: Date.now() });
2377
- return runResultCheckpoint;
2378
- }
2379
- default:
2380
- throw new Error("Run stopped by unknown reason");
2381
- }
2382
- }
2383
- }
2384
- function getEventListener(options) {
2385
- const listener = options?.eventListener ?? ((e) => console.log(JSON.stringify(e)));
2386
- return async (event) => {
2387
- if ("stepNumber" in event) {
2388
- await defaultStoreEvent(event);
2389
- }
2390
- listener(event);
2391
- };
2392
- }
2393
- async function runDelegate(delegation, parentSetting, parentCheckpoint, parentExpert, options) {
2394
- const { expert, toolCallId, toolName, query } = delegation;
2395
- const delegateRunId = createId();
2396
- const delegateSetting = {
2397
- ...parentSetting,
2398
- runId: delegateRunId,
2399
- expertKey: expert.key,
2400
- input: { text: query }
2401
- };
2402
- const delegateCheckpoint = {
2403
- id: createId(),
2404
- jobId: parentSetting.jobId,
2405
- runId: delegateRunId,
2406
- status: "init",
2407
- stepNumber: parentCheckpoint.stepNumber,
2408
- messages: [],
2409
- expert: {
2410
- key: expert.key,
2411
- name: expert.name,
2412
- version: expert.version
2413
- },
2414
- delegatedBy: {
2415
- expert: {
2416
- key: parentExpert.key,
2417
- name: parentExpert.name,
2418
- version: parentExpert.version
2419
- },
2420
- toolCallId,
2421
- toolName,
2422
- checkpointId: parentCheckpoint.id
2423
- },
2424
- usage: createEmptyUsage(),
2425
- contextWindow: parentCheckpoint.contextWindow
2426
- };
2427
- const resultCheckpoint = await run(
2428
- { setting: delegateSetting, checkpoint: delegateCheckpoint },
2429
- { ...options, returnOnDelegationComplete: true }
2430
- );
2431
- const lastMessage = resultCheckpoint.messages[resultCheckpoint.messages.length - 1];
2432
- if (!lastMessage || lastMessage.type !== "expertMessage") {
2433
- throw new Error("Delegation error: delegation result message is incorrect");
2434
176
  }
2435
- const textPart = lastMessage.contents.find((c) => c.type === "textPart");
2436
- if (!textPart || textPart.type !== "textPart") {
2437
- throw new Error("Delegation error: delegation result message does not contain text");
2438
- }
2439
- return {
2440
- toolCallId,
2441
- toolName,
2442
- expertKey: expert.key,
2443
- text: textPart.text,
2444
- stepNumber: resultCheckpoint.stepNumber,
2445
- deltaUsage: resultCheckpoint.usage
2446
- };
2447
- }
177
+ };
2448
178
 
2449
179
  // src/index.ts
180
+ registerAdapter("perstack", () => new PerstackAdapter());
2450
181
  var runtimeVersion = package_default.version;
2451
182
 
2452
- export { StateMachineLogics, calculateContextWindowUsage, createInitialJob, getAllJobs, getAllRuns, getCheckpointDir, getCheckpointPath, getCheckpointsByJobId, getContextWindow, getEventContents, getEventsByRun, getJobDir, getJobsDir, getModel, defaultGetRunDir as getRunDir, retrieveJob, run, runtimeStateMachine, runtimeVersion, storeJob };
183
+ export { PerstackAdapter, runtimeVersion };
2453
184
  //# sourceMappingURL=index.js.map
2454
185
  //# sourceMappingURL=index.js.map