kuzushi 0.1.0
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/LICENSE +21 -0
- package/README.md +410 -0
- package/dist/agent-runtime/claude.d.ts +8 -0
- package/dist/agent-runtime/claude.js +124 -0
- package/dist/agent-runtime/claude.js.map +1 -0
- package/dist/agent-runtime/index.d.ts +9 -0
- package/dist/agent-runtime/index.js +31 -0
- package/dist/agent-runtime/index.js.map +1 -0
- package/dist/agent-runtime/model-spec.d.ts +8 -0
- package/dist/agent-runtime/model-spec.js +17 -0
- package/dist/agent-runtime/model-spec.js.map +1 -0
- package/dist/agent-runtime/pi-ai.d.ts +8 -0
- package/dist/agent-runtime/pi-ai.js +365 -0
- package/dist/agent-runtime/pi-ai.js.map +1 -0
- package/dist/agent-runtime/tools.d.ts +3 -0
- package/dist/agent-runtime/tools.js +330 -0
- package/dist/agent-runtime/tools.js.map +1 -0
- package/dist/agent-runtime/types.d.ts +72 -0
- package/dist/agent-runtime/types.js +2 -0
- package/dist/agent-runtime/types.js.map +1 -0
- package/dist/agents/index.d.ts +7 -0
- package/dist/agents/index.js +24 -0
- package/dist/agents/index.js.map +1 -0
- package/dist/agents/registry.d.ts +14 -0
- package/dist/agents/registry.js +97 -0
- package/dist/agents/registry.js.map +1 -0
- package/dist/agents/scanner-adapter.d.ts +5 -0
- package/dist/agents/scanner-adapter.js +18 -0
- package/dist/agents/scanner-adapter.js.map +1 -0
- package/dist/agents/tasks/augur-analyze.d.ts +22 -0
- package/dist/agents/tasks/augur-analyze.js +418 -0
- package/dist/agents/tasks/augur-analyze.js.map +1 -0
- package/dist/agents/tasks/augur-extraction-agent.d.ts +44 -0
- package/dist/agents/tasks/augur-extraction-agent.js +507 -0
- package/dist/agents/tasks/augur-extraction-agent.js.map +1 -0
- package/dist/agents/tasks/augur-label.d.ts +21 -0
- package/dist/agents/tasks/augur-label.js +627 -0
- package/dist/agents/tasks/augur-label.js.map +1 -0
- package/dist/agents/tasks/augur-preflight.d.ts +36 -0
- package/dist/agents/tasks/augur-preflight.js +471 -0
- package/dist/agents/tasks/augur-preflight.js.map +1 -0
- package/dist/agents/tasks/augur-types.d.ts +111 -0
- package/dist/agents/tasks/augur-types.js +169 -0
- package/dist/agents/tasks/augur-types.js.map +1 -0
- package/dist/agents/tasks/context-gatherer.d.ts +6 -0
- package/dist/agents/tasks/context-gatherer.js +320 -0
- package/dist/agents/tasks/context-gatherer.js.map +1 -0
- package/dist/agents/types.d.ts +28 -0
- package/dist/agents/types.js +2 -0
- package/dist/agents/types.js.map +1 -0
- package/dist/bus/adapters/google-pubsub.d.ts +13 -0
- package/dist/bus/adapters/google-pubsub.js +26 -0
- package/dist/bus/adapters/google-pubsub.js.map +1 -0
- package/dist/bus/adapters/in-process.d.ts +12 -0
- package/dist/bus/adapters/in-process.js +118 -0
- package/dist/bus/adapters/in-process.js.map +1 -0
- package/dist/bus/adapters/index.d.ts +6 -0
- package/dist/bus/adapters/index.js +22 -0
- package/dist/bus/adapters/index.js.map +1 -0
- package/dist/bus/adapters/nats.d.ts +13 -0
- package/dist/bus/adapters/nats.js +26 -0
- package/dist/bus/adapters/nats.js.map +1 -0
- package/dist/bus/adapters/redis.d.ts +13 -0
- package/dist/bus/adapters/redis.js +26 -0
- package/dist/bus/adapters/redis.js.map +1 -0
- package/dist/bus/events.d.ts +295 -0
- package/dist/bus/events.js +2 -0
- package/dist/bus/events.js.map +1 -0
- package/dist/bus/helpers.d.ts +7 -0
- package/dist/bus/helpers.js +16 -0
- package/dist/bus/helpers.js.map +1 -0
- package/dist/bus/index.d.ts +30 -0
- package/dist/bus/index.js +66 -0
- package/dist/bus/index.js.map +1 -0
- package/dist/bus/orchestrator.d.ts +51 -0
- package/dist/bus/orchestrator.js +1350 -0
- package/dist/bus/orchestrator.js.map +1 -0
- package/dist/bus/types.d.ts +23 -0
- package/dist/bus/types.js +2 -0
- package/dist/bus/types.js.map +1 -0
- package/dist/bus/workers/audit-worker.d.ts +9 -0
- package/dist/bus/workers/audit-worker.js +125 -0
- package/dist/bus/workers/audit-worker.js.map +1 -0
- package/dist/bus/workers/poc-harness-worker.d.ts +9 -0
- package/dist/bus/workers/poc-harness-worker.js +96 -0
- package/dist/bus/workers/poc-harness-worker.js.map +1 -0
- package/dist/bus/workers/report-worker.d.ts +11 -0
- package/dist/bus/workers/report-worker.js +235 -0
- package/dist/bus/workers/report-worker.js.map +1 -0
- package/dist/bus/workers/scan-worker.d.ts +13 -0
- package/dist/bus/workers/scan-worker.js +223 -0
- package/dist/bus/workers/scan-worker.js.map +1 -0
- package/dist/bus/workers/store-worker.d.ts +10 -0
- package/dist/bus/workers/store-worker.js +62 -0
- package/dist/bus/workers/store-worker.js.map +1 -0
- package/dist/bus/workers/triage-worker.d.ts +13 -0
- package/dist/bus/workers/triage-worker.js +129 -0
- package/dist/bus/workers/triage-worker.js.map +1 -0
- package/dist/bus/workers/verification-worker.d.ts +9 -0
- package/dist/bus/workers/verification-worker.js +91 -0
- package/dist/bus/workers/verification-worker.js.map +1 -0
- package/dist/cli.d.ts +2 -0
- package/dist/cli.js +513 -0
- package/dist/cli.js.map +1 -0
- package/dist/config.d.ts +17 -0
- package/dist/config.js +866 -0
- package/dist/config.js.map +1 -0
- package/dist/context.d.ts +5 -0
- package/dist/context.js +26 -0
- package/dist/context.js.map +1 -0
- package/dist/deps.d.ts +15 -0
- package/dist/deps.js +18 -0
- package/dist/deps.js.map +1 -0
- package/dist/llm/anthropic.d.ts +3 -0
- package/dist/llm/anthropic.js +21 -0
- package/dist/llm/anthropic.js.map +1 -0
- package/dist/poc-harness.d.ts +49 -0
- package/dist/poc-harness.js +420 -0
- package/dist/poc-harness.js.map +1 -0
- package/dist/report-markdown.d.ts +3 -0
- package/dist/report-markdown.js +197 -0
- package/dist/report-markdown.js.map +1 -0
- package/dist/report-sarif.d.ts +3 -0
- package/dist/report-sarif.js +253 -0
- package/dist/report-sarif.js.map +1 -0
- package/dist/report.d.ts +18 -0
- package/dist/report.js +146 -0
- package/dist/report.js.map +1 -0
- package/dist/retry.d.ts +6 -0
- package/dist/retry.js +25 -0
- package/dist/retry.js.map +1 -0
- package/dist/scanner/claude-adk.d.ts +26 -0
- package/dist/scanner/claude-adk.js +265 -0
- package/dist/scanner/claude-adk.js.map +1 -0
- package/dist/scanner/resolve.d.ts +10 -0
- package/dist/scanner/resolve.js +99 -0
- package/dist/scanner/resolve.js.map +1 -0
- package/dist/scanner.d.ts +47 -0
- package/dist/scanner.js +123 -0
- package/dist/scanner.js.map +1 -0
- package/dist/scanners/agentic.d.ts +11 -0
- package/dist/scanners/agentic.js +71 -0
- package/dist/scanners/agentic.js.map +1 -0
- package/dist/scanners/claude-adk.d.ts +11 -0
- package/dist/scanners/claude-adk.js +74 -0
- package/dist/scanners/claude-adk.js.map +1 -0
- package/dist/scanners/codeql.d.ts +15 -0
- package/dist/scanners/codeql.js +97 -0
- package/dist/scanners/codeql.js.map +1 -0
- package/dist/scanners/finding-selection.d.ts +7 -0
- package/dist/scanners/finding-selection.js +120 -0
- package/dist/scanners/finding-selection.js.map +1 -0
- package/dist/scanners/index.d.ts +4 -0
- package/dist/scanners/index.js +14 -0
- package/dist/scanners/index.js.map +1 -0
- package/dist/scanners/registry.d.ts +12 -0
- package/dist/scanners/registry.js +70 -0
- package/dist/scanners/registry.js.map +1 -0
- package/dist/scanners/resolve-codeql.d.ts +9 -0
- package/dist/scanners/resolve-codeql.js +38 -0
- package/dist/scanners/resolve-codeql.js.map +1 -0
- package/dist/scanners/resolve-semgrep.d.ts +10 -0
- package/dist/scanners/resolve-semgrep.js +96 -0
- package/dist/scanners/resolve-semgrep.js.map +1 -0
- package/dist/scanners/run-agentic.d.ts +33 -0
- package/dist/scanners/run-agentic.js +267 -0
- package/dist/scanners/run-agentic.js.map +1 -0
- package/dist/scanners/run-claude-adk.d.ts +33 -0
- package/dist/scanners/run-claude-adk.js +267 -0
- package/dist/scanners/run-claude-adk.js.map +1 -0
- package/dist/scanners/run-codeql.d.ts +100 -0
- package/dist/scanners/run-codeql.js +538 -0
- package/dist/scanners/run-codeql.js.map +1 -0
- package/dist/scanners/run-semgrep.d.ts +41 -0
- package/dist/scanners/run-semgrep.js +115 -0
- package/dist/scanners/run-semgrep.js.map +1 -0
- package/dist/scanners/scoring.d.ts +7 -0
- package/dist/scanners/scoring.js +14 -0
- package/dist/scanners/scoring.js.map +1 -0
- package/dist/scanners/semgrep.d.ts +11 -0
- package/dist/scanners/semgrep.js +64 -0
- package/dist/scanners/semgrep.js.map +1 -0
- package/dist/scanners/types.d.ts +36 -0
- package/dist/scanners/types.js +2 -0
- package/dist/scanners/types.js.map +1 -0
- package/dist/scanners.d.ts +25 -0
- package/dist/scanners.js +88 -0
- package/dist/scanners.js.map +1 -0
- package/dist/store.d.ts +106 -0
- package/dist/store.js +609 -0
- package/dist/store.js.map +1 -0
- package/dist/triage.d.ts +74 -0
- package/dist/triage.js +314 -0
- package/dist/triage.js.map +1 -0
- package/dist/types.d.ts +166 -0
- package/dist/types.js +2 -0
- package/dist/types.js.map +1 -0
- package/dist/utils.d.ts +6 -0
- package/dist/utils.js +19 -0
- package/dist/utils.js.map +1 -0
- package/dist/verify.d.ts +56 -0
- package/dist/verify.js +298 -0
- package/dist/verify.js.map +1 -0
- package/package.json +55 -0
|
@@ -0,0 +1,365 @@
|
|
|
1
|
+
import { complete, getModel, validateToolCall, } from "@mariozechner/pi-ai";
|
|
2
|
+
import { isRecord, normalizeCostUsd, toErrorMessage } from "../utils.js";
|
|
3
|
+
import { parsePiAiModelSpec } from "./model-spec.js";
|
|
4
|
+
import { executeTool, PI_AI_TOOL_DEFINITIONS } from "./tools.js";
|
|
5
|
+
const DEFAULT_MAX_TURNS = 10;
|
|
6
|
+
const KNOWN_TOOL_NAMES = new Set(["Read", "Glob", "Grep", "Bash", "Write"]);
|
|
7
|
+
export class PiAiAgentRuntime {
|
|
8
|
+
id = "pi-ai";
|
|
9
|
+
runtimeConfig;
|
|
10
|
+
constructor(runtimeConfig) {
|
|
11
|
+
this.runtimeConfig = runtimeConfig;
|
|
12
|
+
}
|
|
13
|
+
async *query(request) {
|
|
14
|
+
const restoreRuntimeEnv = applyRuntimeEnvOverrides(this.runtimeConfig);
|
|
15
|
+
const startedAtMs = Date.now();
|
|
16
|
+
const maxTurns = normalizeMaxTurns(request.options.maxTurns);
|
|
17
|
+
const signal = request.options.abortController?.signal;
|
|
18
|
+
const maxBudgetUsd = normalizeBudget(request.options.maxBudgetUsd);
|
|
19
|
+
let totalCostUsd = 0;
|
|
20
|
+
let generatedToolCallCounter = 0;
|
|
21
|
+
try {
|
|
22
|
+
const model = resolveModel(request.options.model);
|
|
23
|
+
const enabledTools = resolveEnabledTools(request.options.allowedTools, request.options.disallowedTools);
|
|
24
|
+
const context = {
|
|
25
|
+
systemPrompt: buildSystemPrompt(request.options.outputFormat?.schema),
|
|
26
|
+
messages: [
|
|
27
|
+
{
|
|
28
|
+
role: "user",
|
|
29
|
+
content: request.prompt,
|
|
30
|
+
timestamp: Date.now(),
|
|
31
|
+
},
|
|
32
|
+
],
|
|
33
|
+
...(enabledTools.length > 0 ? { tools: enabledTools } : {}),
|
|
34
|
+
};
|
|
35
|
+
const toolsByName = new Map(enabledTools.map((tool) => [tool.name, tool]));
|
|
36
|
+
for (let turn = 1; turn <= maxTurns; turn++) {
|
|
37
|
+
if (signal?.aborted) {
|
|
38
|
+
yield errorResult("error_aborted", "pi-ai agent run aborted", startedAtMs, turn - 1, totalCostUsd);
|
|
39
|
+
return;
|
|
40
|
+
}
|
|
41
|
+
const assistant = await complete(model, context, { signal });
|
|
42
|
+
context.messages.push(assistant);
|
|
43
|
+
totalCostUsd += extractCostUsd(assistant);
|
|
44
|
+
const assistantContent = mapAssistantContent(assistant);
|
|
45
|
+
if (assistantContent.length > 0) {
|
|
46
|
+
yield {
|
|
47
|
+
type: "assistant",
|
|
48
|
+
content: assistantContent,
|
|
49
|
+
};
|
|
50
|
+
}
|
|
51
|
+
if (assistant.stopReason === "aborted" || signal?.aborted) {
|
|
52
|
+
yield errorResult("error_aborted", assistant.errorMessage ?? "pi-ai agent run aborted", startedAtMs, turn, totalCostUsd);
|
|
53
|
+
return;
|
|
54
|
+
}
|
|
55
|
+
if (assistant.stopReason === "error") {
|
|
56
|
+
yield errorResult("error_runtime", assistant.errorMessage ?? "pi-ai request failed", startedAtMs, turn, totalCostUsd);
|
|
57
|
+
return;
|
|
58
|
+
}
|
|
59
|
+
const toolCalls = assistant.content.filter((block) => (block.type === "toolCall"));
|
|
60
|
+
if (toolCalls.length === 0) {
|
|
61
|
+
const resultText = extractText(assistant);
|
|
62
|
+
const structuredOutput = request.options.outputFormat
|
|
63
|
+
? parseStructuredOutput(resultText)
|
|
64
|
+
: undefined;
|
|
65
|
+
yield {
|
|
66
|
+
type: "result",
|
|
67
|
+
subtype: "success",
|
|
68
|
+
result: resultText,
|
|
69
|
+
structuredOutput,
|
|
70
|
+
durationMs: Date.now() - startedAtMs,
|
|
71
|
+
numTurns: turn,
|
|
72
|
+
totalCostUsd: normalizeCostUsd(totalCostUsd),
|
|
73
|
+
};
|
|
74
|
+
return;
|
|
75
|
+
}
|
|
76
|
+
if (maxBudgetUsd != null && totalCostUsd > maxBudgetUsd) {
|
|
77
|
+
yield errorResult("error_budget_exceeded", `pi-ai budget exceeded: cost $${normalizeCostUsd(totalCostUsd).toFixed(4)} > max $${maxBudgetUsd.toFixed(4)}`, startedAtMs, turn, totalCostUsd);
|
|
78
|
+
return;
|
|
79
|
+
}
|
|
80
|
+
if (turn >= maxTurns) {
|
|
81
|
+
yield errorResult("error_max_turns", `pi-ai agent exceeded max turns (${maxTurns}) before producing a final response`, startedAtMs, turn, totalCostUsd);
|
|
82
|
+
return;
|
|
83
|
+
}
|
|
84
|
+
for (const toolCall of toolCalls) {
|
|
85
|
+
const toolUseId = toolCall.id || `${toolCall.name}_${Date.now()}_${++generatedToolCallCounter}`;
|
|
86
|
+
const startedToolMs = Date.now();
|
|
87
|
+
const tool = toolsByName.get(toolCall.name);
|
|
88
|
+
if (!tool) {
|
|
89
|
+
const unknownToolMessage = `Tool "${toolCall.name}" is not enabled`;
|
|
90
|
+
const elapsedSeconds = Math.max(0, (Date.now() - startedToolMs) / 1000);
|
|
91
|
+
yield {
|
|
92
|
+
type: "tool-progress",
|
|
93
|
+
toolUseId,
|
|
94
|
+
toolName: toolCall.name,
|
|
95
|
+
elapsedSeconds,
|
|
96
|
+
};
|
|
97
|
+
yield {
|
|
98
|
+
type: "tool-summary",
|
|
99
|
+
summary: summarizeToolResult(toolCall.name, unknownToolMessage, true),
|
|
100
|
+
precedingToolUseIds: [toolUseId],
|
|
101
|
+
};
|
|
102
|
+
context.messages.push({
|
|
103
|
+
role: "toolResult",
|
|
104
|
+
toolCallId: toolUseId,
|
|
105
|
+
toolName: toolCall.name,
|
|
106
|
+
content: [{ type: "text", text: unknownToolMessage }],
|
|
107
|
+
isError: true,
|
|
108
|
+
timestamp: Date.now(),
|
|
109
|
+
});
|
|
110
|
+
continue;
|
|
111
|
+
}
|
|
112
|
+
let validatedArgs;
|
|
113
|
+
try {
|
|
114
|
+
validatedArgs = validateToolCall(enabledTools, toolCall);
|
|
115
|
+
}
|
|
116
|
+
catch (error) {
|
|
117
|
+
const validationError = toErrorMessage(error);
|
|
118
|
+
const elapsedSeconds = Math.max(0, (Date.now() - startedToolMs) / 1000);
|
|
119
|
+
yield {
|
|
120
|
+
type: "tool-progress",
|
|
121
|
+
toolUseId,
|
|
122
|
+
toolName: toolCall.name,
|
|
123
|
+
elapsedSeconds,
|
|
124
|
+
};
|
|
125
|
+
yield {
|
|
126
|
+
type: "tool-summary",
|
|
127
|
+
summary: summarizeToolResult(toolCall.name, validationError, true),
|
|
128
|
+
precedingToolUseIds: [toolUseId],
|
|
129
|
+
};
|
|
130
|
+
context.messages.push({
|
|
131
|
+
role: "toolResult",
|
|
132
|
+
toolCallId: toolUseId,
|
|
133
|
+
toolName: toolCall.name,
|
|
134
|
+
content: [{ type: "text", text: validationError }],
|
|
135
|
+
isError: true,
|
|
136
|
+
timestamp: Date.now(),
|
|
137
|
+
});
|
|
138
|
+
continue;
|
|
139
|
+
}
|
|
140
|
+
const permission = await canUseTool(request, toolCall.name, validatedArgs, signal, toolUseId);
|
|
141
|
+
if (permission.behavior === "deny") {
|
|
142
|
+
yield errorResult("error_permission_denied", permission.message ?? `Permission denied for tool "${toolCall.name}"`, startedAtMs, turn, totalCostUsd);
|
|
143
|
+
return;
|
|
144
|
+
}
|
|
145
|
+
let isToolError = false;
|
|
146
|
+
let toolResultText = "";
|
|
147
|
+
try {
|
|
148
|
+
toolResultText = await executeTool(toolCall.name, validatedArgs, request.options.cwd);
|
|
149
|
+
}
|
|
150
|
+
catch (error) {
|
|
151
|
+
isToolError = true;
|
|
152
|
+
toolResultText = toErrorMessage(error);
|
|
153
|
+
}
|
|
154
|
+
const elapsedSeconds = Math.max(0, (Date.now() - startedToolMs) / 1000);
|
|
155
|
+
yield {
|
|
156
|
+
type: "tool-progress",
|
|
157
|
+
toolUseId,
|
|
158
|
+
toolName: toolCall.name,
|
|
159
|
+
elapsedSeconds,
|
|
160
|
+
};
|
|
161
|
+
yield {
|
|
162
|
+
type: "tool-summary",
|
|
163
|
+
summary: summarizeToolResult(toolCall.name, toolResultText, isToolError),
|
|
164
|
+
precedingToolUseIds: [toolUseId],
|
|
165
|
+
};
|
|
166
|
+
context.messages.push({
|
|
167
|
+
role: "toolResult",
|
|
168
|
+
toolCallId: toolUseId,
|
|
169
|
+
toolName: toolCall.name,
|
|
170
|
+
content: [{ type: "text", text: toolResultText }],
|
|
171
|
+
isError: isToolError,
|
|
172
|
+
timestamp: Date.now(),
|
|
173
|
+
});
|
|
174
|
+
}
|
|
175
|
+
}
|
|
176
|
+
yield errorResult("error_max_turns", `pi-ai agent exceeded max turns (${maxTurns})`, startedAtMs, maxTurns, totalCostUsd);
|
|
177
|
+
}
|
|
178
|
+
catch (error) {
|
|
179
|
+
const aborted = signal?.aborted === true;
|
|
180
|
+
yield errorResult(aborted ? "error_aborted" : "error_runtime", toErrorMessage(error), startedAtMs, 0, totalCostUsd);
|
|
181
|
+
}
|
|
182
|
+
finally {
|
|
183
|
+
restoreRuntimeEnv();
|
|
184
|
+
}
|
|
185
|
+
}
|
|
186
|
+
}
|
|
187
|
+
function applyRuntimeEnvOverrides(runtimeConfig) {
|
|
188
|
+
const shouldOverrideApiKey = Boolean(runtimeConfig?.apiKey);
|
|
189
|
+
const shouldOverrideBaseUrl = Boolean(runtimeConfig?.baseUrl);
|
|
190
|
+
const originalApiKey = process.env["ANTHROPIC_API_KEY"];
|
|
191
|
+
const originalBaseUrl = process.env["ANTHROPIC_BASE_URL"];
|
|
192
|
+
if (shouldOverrideApiKey && runtimeConfig?.apiKey) {
|
|
193
|
+
process.env["ANTHROPIC_API_KEY"] = runtimeConfig.apiKey;
|
|
194
|
+
}
|
|
195
|
+
if (shouldOverrideBaseUrl && runtimeConfig?.baseUrl) {
|
|
196
|
+
process.env["ANTHROPIC_BASE_URL"] = runtimeConfig.baseUrl;
|
|
197
|
+
}
|
|
198
|
+
return () => {
|
|
199
|
+
if (shouldOverrideApiKey) {
|
|
200
|
+
if (originalApiKey === undefined) {
|
|
201
|
+
delete process.env["ANTHROPIC_API_KEY"];
|
|
202
|
+
}
|
|
203
|
+
else {
|
|
204
|
+
process.env["ANTHROPIC_API_KEY"] = originalApiKey;
|
|
205
|
+
}
|
|
206
|
+
}
|
|
207
|
+
if (shouldOverrideBaseUrl) {
|
|
208
|
+
if (originalBaseUrl === undefined) {
|
|
209
|
+
delete process.env["ANTHROPIC_BASE_URL"];
|
|
210
|
+
}
|
|
211
|
+
else {
|
|
212
|
+
process.env["ANTHROPIC_BASE_URL"] = originalBaseUrl;
|
|
213
|
+
}
|
|
214
|
+
}
|
|
215
|
+
};
|
|
216
|
+
}
|
|
217
|
+
function resolveModel(modelSpec) {
|
|
218
|
+
const { provider, modelId } = parsePiAiModelSpec(modelSpec);
|
|
219
|
+
try {
|
|
220
|
+
return getModel(provider, modelId);
|
|
221
|
+
}
|
|
222
|
+
catch (error) {
|
|
223
|
+
throw new Error(`Unable to resolve pi-ai model "${modelSpec}": ${toErrorMessage(error)}`);
|
|
224
|
+
}
|
|
225
|
+
}
|
|
226
|
+
function resolveEnabledTools(allowedTools, disallowedTools) {
|
|
227
|
+
const allowSet = new Set();
|
|
228
|
+
if (Array.isArray(allowedTools) && allowedTools.length > 0) {
|
|
229
|
+
for (const toolName of allowedTools) {
|
|
230
|
+
if (KNOWN_TOOL_NAMES.has(toolName)) {
|
|
231
|
+
allowSet.add(toolName);
|
|
232
|
+
}
|
|
233
|
+
}
|
|
234
|
+
}
|
|
235
|
+
else {
|
|
236
|
+
for (const toolName of KNOWN_TOOL_NAMES) {
|
|
237
|
+
allowSet.add(toolName);
|
|
238
|
+
}
|
|
239
|
+
}
|
|
240
|
+
if (Array.isArray(disallowedTools)) {
|
|
241
|
+
for (const toolName of disallowedTools) {
|
|
242
|
+
allowSet.delete(toolName);
|
|
243
|
+
}
|
|
244
|
+
}
|
|
245
|
+
return PI_AI_TOOL_DEFINITIONS.filter((tool) => allowSet.has(tool.name));
|
|
246
|
+
}
|
|
247
|
+
function buildSystemPrompt(schema) {
|
|
248
|
+
const lines = [
|
|
249
|
+
"You are a code security agent operating on a local repository.",
|
|
250
|
+
"Use available tools (Read, Glob, Grep, Bash, Write) when code inspection or file manipulation is required.",
|
|
251
|
+
"When you call tools, pass valid JSON arguments.",
|
|
252
|
+
];
|
|
253
|
+
if (schema) {
|
|
254
|
+
lines.push("", "Return the final answer as JSON only (no markdown fences).", "The JSON must match this schema exactly:", JSON.stringify(schema));
|
|
255
|
+
}
|
|
256
|
+
return lines.join("\n");
|
|
257
|
+
}
|
|
258
|
+
function mapAssistantContent(assistant) {
|
|
259
|
+
const content = [];
|
|
260
|
+
for (const block of assistant.content) {
|
|
261
|
+
if (block.type === "text") {
|
|
262
|
+
content.push({
|
|
263
|
+
type: "text",
|
|
264
|
+
text: block.text,
|
|
265
|
+
});
|
|
266
|
+
continue;
|
|
267
|
+
}
|
|
268
|
+
if (block.type === "thinking") {
|
|
269
|
+
content.push({
|
|
270
|
+
type: "text",
|
|
271
|
+
text: block.thinking,
|
|
272
|
+
});
|
|
273
|
+
continue;
|
|
274
|
+
}
|
|
275
|
+
if (block.type === "toolCall") {
|
|
276
|
+
content.push({
|
|
277
|
+
type: "tool_use",
|
|
278
|
+
id: block.id,
|
|
279
|
+
name: block.name,
|
|
280
|
+
input: isRecord(block.arguments) ? block.arguments : {},
|
|
281
|
+
});
|
|
282
|
+
}
|
|
283
|
+
}
|
|
284
|
+
return content;
|
|
285
|
+
}
|
|
286
|
+
function extractText(assistant) {
|
|
287
|
+
const textBlocks = assistant.content
|
|
288
|
+
.filter((block) => (block.type === "text"))
|
|
289
|
+
.map((block) => block.text.trim())
|
|
290
|
+
.filter(Boolean);
|
|
291
|
+
return textBlocks.join("\n\n");
|
|
292
|
+
}
|
|
293
|
+
function parseStructuredOutput(resultText) {
|
|
294
|
+
const extracted = extractJsonObject(resultText);
|
|
295
|
+
if (!extracted) {
|
|
296
|
+
return undefined;
|
|
297
|
+
}
|
|
298
|
+
try {
|
|
299
|
+
return JSON.parse(extracted);
|
|
300
|
+
}
|
|
301
|
+
catch {
|
|
302
|
+
return undefined;
|
|
303
|
+
}
|
|
304
|
+
}
|
|
305
|
+
function extractJsonObject(text) {
|
|
306
|
+
const trimmed = text.trim();
|
|
307
|
+
if (!trimmed) {
|
|
308
|
+
return null;
|
|
309
|
+
}
|
|
310
|
+
const fenceMatch = trimmed.match(/```(?:json)?\s*\n?([\s\S]*?)\n?```/);
|
|
311
|
+
if (fenceMatch) {
|
|
312
|
+
return fenceMatch[1].trim();
|
|
313
|
+
}
|
|
314
|
+
const start = trimmed.indexOf("{");
|
|
315
|
+
const end = trimmed.lastIndexOf("}");
|
|
316
|
+
if (start !== -1 && end > start) {
|
|
317
|
+
return trimmed.slice(start, end + 1);
|
|
318
|
+
}
|
|
319
|
+
return trimmed;
|
|
320
|
+
}
|
|
321
|
+
function extractCostUsd(message) {
|
|
322
|
+
return normalizeCostUsd(message.usage?.cost?.total);
|
|
323
|
+
}
|
|
324
|
+
function normalizeMaxTurns(value) {
|
|
325
|
+
if (!Number.isFinite(value) || value == null || value < 1) {
|
|
326
|
+
return DEFAULT_MAX_TURNS;
|
|
327
|
+
}
|
|
328
|
+
return Math.floor(value);
|
|
329
|
+
}
|
|
330
|
+
function normalizeBudget(value) {
|
|
331
|
+
if (!Number.isFinite(value) || value == null || value < 0) {
|
|
332
|
+
return undefined;
|
|
333
|
+
}
|
|
334
|
+
return value;
|
|
335
|
+
}
|
|
336
|
+
async function canUseTool(request, toolName, args, signal, toolUseId) {
|
|
337
|
+
if (!request.options.canUseTool) {
|
|
338
|
+
return { behavior: "allow" };
|
|
339
|
+
}
|
|
340
|
+
return request.options.canUseTool(toolName, args, {
|
|
341
|
+
signal: signal ?? new AbortController().signal,
|
|
342
|
+
toolUseId,
|
|
343
|
+
});
|
|
344
|
+
}
|
|
345
|
+
function summarizeToolResult(toolName, toolResult, isError) {
|
|
346
|
+
const firstLine = toolResult
|
|
347
|
+
.split("\n")
|
|
348
|
+
.map((line) => line.trim())
|
|
349
|
+
.find(Boolean);
|
|
350
|
+
if (isError) {
|
|
351
|
+
return `${toolName} failed${firstLine ? `: ${firstLine}` : ""}`;
|
|
352
|
+
}
|
|
353
|
+
return `${toolName} completed${firstLine ? `: ${firstLine}` : ""}`;
|
|
354
|
+
}
|
|
355
|
+
function errorResult(subtype, error, startedAtMs, numTurns, totalCostUsd) {
|
|
356
|
+
return {
|
|
357
|
+
type: "result",
|
|
358
|
+
subtype,
|
|
359
|
+
errors: [error],
|
|
360
|
+
durationMs: Date.now() - startedAtMs,
|
|
361
|
+
numTurns,
|
|
362
|
+
totalCostUsd: normalizeCostUsd(totalCostUsd),
|
|
363
|
+
};
|
|
364
|
+
}
|
|
365
|
+
//# sourceMappingURL=pi-ai.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"pi-ai.js","sourceRoot":"","sources":["../../src/agent-runtime/pi-ai.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,QAAQ,EACR,QAAQ,EAMR,gBAAgB,GACjB,MAAM,qBAAqB,CAAC;AAE7B,OAAO,EAAE,QAAQ,EAAE,gBAAgB,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AACzE,OAAO,EAAE,kBAAkB,EAAE,MAAM,iBAAiB,CAAC;AACrD,OAAO,EAAE,WAAW,EAAE,sBAAsB,EAAE,MAAM,YAAY,CAAC;AAGjE,MAAM,iBAAiB,GAAG,EAAE,CAAC;AAC7B,MAAM,gBAAgB,GAAG,IAAI,GAAG,CAAC,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC;AAE5E,MAAM,OAAO,gBAAgB;IAClB,EAAE,GAAG,OAAO,CAAC;IACL,aAAa,CAAsB;IAEpD,YAAY,aAAkC;QAC5C,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;IACrC,CAAC;IAED,KAAK,CAAC,CAAC,KAAK,CAAC,OAA0B;QACrC,MAAM,iBAAiB,GAAG,wBAAwB,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QACvE,MAAM,WAAW,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC/B,MAAM,QAAQ,GAAG,iBAAiB,CAAC,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAC7D,MAAM,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,eAAe,EAAE,MAAM,CAAC;QACvD,MAAM,YAAY,GAAG,eAAe,CAAC,OAAO,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;QACnE,IAAI,YAAY,GAAG,CAAC,CAAC;QACrB,IAAI,wBAAwB,GAAG,CAAC,CAAC;QAEjC,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,YAAY,CAAC,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;YAClD,MAAM,YAAY,GAAG,mBAAmB,CACtC,OAAO,CAAC,OAAO,CAAC,YAAY,EAC5B,OAAO,CAAC,OAAO,CAAC,eAAe,CAChC,CAAC;YACF,MAAM,OAAO,GAAgB;gBAC3B,YAAY,EAAE,iBAAiB,CAAC,OAAO,CAAC,OAAO,CAAC,YAAY,EAAE,MAAM,CAAC;gBACrE,QAAQ,EAAE;oBACR;wBACE,IAAI,EAAE,MAAM;wBACZ,OAAO,EAAE,OAAO,CAAC,MAAM;wBACvB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;qBACtB;iBACF;gBACD,GAAG,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aAC5D,CAAC;YAEF,MAAM,WAAW,GAAG,IAAI,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC;YAE3E,KAAK,IAAI,IAAI,GAAG,CAAC,EAAE,IAAI,IAAI,QAAQ,EAAE,IAAI,EAAE,EAAE,CAAC;gBAC5C,IAAI,MAAM,EAAE,OAAO,EAAE,CAAC;oBACpB,MAAM,WAAW,CACf,eAAe,EACf,yBAAyB,EACzB,WAAW,EACX,IAAI,GAAG,CAAC,EACR,YAAY,CACb,CAAC;oBACF,OAAO;gBACT,CAAC;gBAED,MAAM,SAAS,GAAG,MAAM,QAAQ,CAAC,KAAK,EAAE,OAAO,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;gBAC7D,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;gBACjC,YAAY,IAAI,cAAc,CAAC,SAAS,CAAC,CAAC;gBAE1C,MAAM,gBAAgB,GAAG,mBAAmB,CAAC,SAAS,CAAC,CAAC;gBACxD,IAAI,gBAAgB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBAChC,MAAM;wBACJ,IAAI,EAAE,WAAW;wBACjB,OAAO,EAAE,gBAAgB;qBAC1B,CAAC;gBACJ,CAAC;gBAED,IAAI,SAAS,CAAC,UAAU,KAAK,SAAS,IAAI,MAAM,EAAE,OAAO,EAAE,CAAC;oBAC1D,MAAM,WAAW,CACf,eAAe,EACf,SAAS,CAAC,YAAY,IAAI,yBAAyB,EACnD,WAAW,EACX,IAAI,EACJ,YAAY,CACb,CAAC;oBACF,OAAO;gBACT,CAAC;gBAED,IAAI,SAAS,CAAC,UAAU,KAAK,OAAO,EAAE,CAAC;oBACrC,MAAM,WAAW,CACf,eAAe,EACf,SAAS,CAAC,YAAY,IAAI,sBAAsB,EAChD,WAAW,EACX,IAAI,EACJ,YAAY,CACb,CAAC;oBACF,OAAO;gBACT,CAAC;gBAED,MAAM,SAAS,GAAG,SAAS,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,KAAK,EAAyB,EAAE,CAAC,CAC3E,KAAK,CAAC,IAAI,KAAK,UAAU,CAC1B,CAAC,CAAC;gBAEH,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;oBAC3B,MAAM,UAAU,GAAG,WAAW,CAAC,SAAS,CAAC,CAAC;oBAC1C,MAAM,gBAAgB,GAAG,OAAO,CAAC,OAAO,CAAC,YAAY;wBACnD,CAAC,CAAC,qBAAqB,CAAC,UAAU,CAAC;wBACnC,CAAC,CAAC,SAAS,CAAC;oBAEd,MAAM;wBACJ,IAAI,EAAE,QAAQ;wBACd,OAAO,EAAE,SAAS;wBAClB,MAAM,EAAE,UAAU;wBAClB,gBAAgB;wBAChB,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,WAAW;wBACpC,QAAQ,EAAE,IAAI;wBACd,YAAY,EAAE,gBAAgB,CAAC,YAAY,CAAC;qBAC7C,CAAC;oBACF,OAAO;gBACT,CAAC;gBAED,IAAI,YAAY,IAAI,IAAI,IAAI,YAAY,GAAG,YAAY,EAAE,CAAC;oBACxD,MAAM,WAAW,CACf,uBAAuB,EACvB,gCAAgC,gBAAgB,CAAC,YAAY,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,WAAW,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAC7G,WAAW,EACX,IAAI,EACJ,YAAY,CACb,CAAC;oBACF,OAAO;gBACT,CAAC;gBAED,IAAI,IAAI,IAAI,QAAQ,EAAE,CAAC;oBACrB,MAAM,WAAW,CACf,iBAAiB,EACjB,mCAAmC,QAAQ,qCAAqC,EAChF,WAAW,EACX,IAAI,EACJ,YAAY,CACb,CAAC;oBACF,OAAO;gBACT,CAAC;gBAED,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;oBACjC,MAAM,SAAS,GAAG,QAAQ,CAAC,EAAE,IAAI,GAAG,QAAQ,CAAC,IAAI,IAAI,IAAI,CAAC,GAAG,EAAE,IAAI,EAAE,wBAAwB,EAAE,CAAC;oBAChG,MAAM,aAAa,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;oBAEjC,MAAM,IAAI,GAAG,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;oBAC5C,IAAI,CAAC,IAAI,EAAE,CAAC;wBACV,MAAM,kBAAkB,GAAG,SAAS,QAAQ,CAAC,IAAI,kBAAkB,CAAC;wBACpE,MAAM,cAAc,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,aAAa,CAAC,GAAG,IAAI,CAAC,CAAC;wBACxE,MAAM;4BACJ,IAAI,EAAE,eAAe;4BACrB,SAAS;4BACT,QAAQ,EAAE,QAAQ,CAAC,IAAI;4BACvB,cAAc;yBACf,CAAC;wBACF,MAAM;4BACJ,IAAI,EAAE,cAAc;4BACpB,OAAO,EAAE,mBAAmB,CAAC,QAAQ,CAAC,IAAI,EAAE,kBAAkB,EAAE,IAAI,CAAC;4BACrE,mBAAmB,EAAE,CAAC,SAAS,CAAC;yBACjC,CAAC;wBACF,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC;4BACpB,IAAI,EAAE,YAAY;4BAClB,UAAU,EAAE,SAAS;4BACrB,QAAQ,EAAE,QAAQ,CAAC,IAAI;4BACvB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,kBAAkB,EAAE,CAAC;4BACrD,OAAO,EAAE,IAAI;4BACb,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;yBACtB,CAAC,CAAC;wBACH,SAAS;oBACX,CAAC;oBAED,IAAI,aAAsC,CAAC;oBAC3C,IAAI,CAAC;wBACH,aAAa,GAAG,gBAAgB,CAAC,YAAY,EAAE,QAAQ,CAA4B,CAAC;oBACtF,CAAC;oBAAC,OAAO,KAAK,EAAE,CAAC;wBACf,MAAM,eAAe,GAAG,cAAc,CAAC,KAAK,CAAC,CAAC;wBAC9C,MAAM,cAAc,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,aAAa,CAAC,GAAG,IAAI,CAAC,CAAC;wBACxE,MAAM;4BACJ,IAAI,EAAE,eAAe;4BACrB,SAAS;4BACT,QAAQ,EAAE,QAAQ,CAAC,IAAI;4BACvB,cAAc;yBACf,CAAC;wBACF,MAAM;4BACJ,IAAI,EAAE,cAAc;4BACpB,OAAO,EAAE,mBAAmB,CAAC,QAAQ,CAAC,IAAI,EAAE,eAAe,EAAE,IAAI,CAAC;4BAClE,mBAAmB,EAAE,CAAC,SAAS,CAAC;yBACjC,CAAC;wBACF,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC;4BACpB,IAAI,EAAE,YAAY;4BAClB,UAAU,EAAE,SAAS;4BACrB,QAAQ,EAAE,QAAQ,CAAC,IAAI;4BACvB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,eAAe,EAAE,CAAC;4BAClD,OAAO,EAAE,IAAI;4BACb,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;yBACtB,CAAC,CAAC;wBACH,SAAS;oBACX,CAAC;oBAED,MAAM,UAAU,GAAG,MAAM,UAAU,CACjC,OAAO,EACP,QAAQ,CAAC,IAAI,EACb,aAAa,EACb,MAAM,EACN,SAAS,CACV,CAAC;oBAEF,IAAI,UAAU,CAAC,QAAQ,KAAK,MAAM,EAAE,CAAC;wBACnC,MAAM,WAAW,CACf,yBAAyB,EACzB,UAAU,CAAC,OAAO,IAAI,+BAA+B,QAAQ,CAAC,IAAI,GAAG,EACrE,WAAW,EACX,IAAI,EACJ,YAAY,CACb,CAAC;wBACF,OAAO;oBACT,CAAC;oBAED,IAAI,WAAW,GAAG,KAAK,CAAC;oBACxB,IAAI,cAAc,GAAG,EAAE,CAAC;oBACxB,IAAI,CAAC;wBACH,cAAc,GAAG,MAAM,WAAW,CAChC,QAAQ,CAAC,IAAI,EACb,aAAa,EACb,OAAO,CAAC,OAAO,CAAC,GAAG,CACpB,CAAC;oBACJ,CAAC;oBAAC,OAAO,KAAK,EAAE,CAAC;wBACf,WAAW,GAAG,IAAI,CAAC;wBACnB,cAAc,GAAG,cAAc,CAAC,KAAK,CAAC,CAAC;oBACzC,CAAC;oBAED,MAAM,cAAc,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,aAAa,CAAC,GAAG,IAAI,CAAC,CAAC;oBACxE,MAAM;wBACJ,IAAI,EAAE,eAAe;wBACrB,SAAS;wBACT,QAAQ,EAAE,QAAQ,CAAC,IAAI;wBACvB,cAAc;qBACf,CAAC;oBACF,MAAM;wBACJ,IAAI,EAAE,cAAc;wBACpB,OAAO,EAAE,mBAAmB,CAAC,QAAQ,CAAC,IAAI,EAAE,cAAc,EAAE,WAAW,CAAC;wBACxE,mBAAmB,EAAE,CAAC,SAAS,CAAC;qBACjC,CAAC;oBAEF,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC;wBACpB,IAAI,EAAE,YAAY;wBAClB,UAAU,EAAE,SAAS;wBACrB,QAAQ,EAAE,QAAQ,CAAC,IAAI;wBACvB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC;wBACjD,OAAO,EAAE,WAAW;wBACpB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;qBACtB,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;YAED,MAAM,WAAW,CACf,iBAAiB,EACjB,mCAAmC,QAAQ,GAAG,EAC9C,WAAW,EACX,QAAQ,EACR,YAAY,CACb,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,OAAO,GAAG,MAAM,EAAE,OAAO,KAAK,IAAI,CAAC;YACzC,MAAM,WAAW,CACf,OAAO,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,eAAe,EAC3C,cAAc,CAAC,KAAK,CAAC,EACrB,WAAW,EACX,CAAC,EACD,YAAY,CACb,CAAC;QACJ,CAAC;gBAAS,CAAC;YACT,iBAAiB,EAAE,CAAC;QACtB,CAAC;IACH,CAAC;CACF;AAED,SAAS,wBAAwB,CAAC,aAA6C;IAC7E,MAAM,oBAAoB,GAAG,OAAO,CAAC,aAAa,EAAE,MAAM,CAAC,CAAC;IAC5D,MAAM,qBAAqB,GAAG,OAAO,CAAC,aAAa,EAAE,OAAO,CAAC,CAAC;IAE9D,MAAM,cAAc,GAAG,OAAO,CAAC,GAAG,CAAC,mBAAmB,CAAC,CAAC;IACxD,MAAM,eAAe,GAAG,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC,CAAC;IAE1D,IAAI,oBAAoB,IAAI,aAAa,EAAE,MAAM,EAAE,CAAC;QAClD,OAAO,CAAC,GAAG,CAAC,mBAAmB,CAAC,GAAG,aAAa,CAAC,MAAM,CAAC;IAC1D,CAAC;IAED,IAAI,qBAAqB,IAAI,aAAa,EAAE,OAAO,EAAE,CAAC;QACpD,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC,GAAG,aAAa,CAAC,OAAO,CAAC;IAC5D,CAAC;IAED,OAAO,GAAG,EAAE;QACV,IAAI,oBAAoB,EAAE,CAAC;YACzB,IAAI,cAAc,KAAK,SAAS,EAAE,CAAC;gBACjC,OAAO,OAAO,CAAC,GAAG,CAAC,mBAAmB,CAAC,CAAC;YAC1C,CAAC;iBAAM,CAAC;gBACN,OAAO,CAAC,GAAG,CAAC,mBAAmB,CAAC,GAAG,cAAc,CAAC;YACpD,CAAC;QACH,CAAC;QAED,IAAI,qBAAqB,EAAE,CAAC;YAC1B,IAAI,eAAe,KAAK,SAAS,EAAE,CAAC;gBAClC,OAAO,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC,CAAC;YAC3C,CAAC;iBAAM,CAAC;gBACN,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC,GAAG,eAAe,CAAC;YACtD,CAAC;QACH,CAAC;IACH,CAAC,CAAC;AACJ,CAAC;AAED,SAAS,YAAY,CAAC,SAAiB;IACrC,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,GAAG,kBAAkB,CAAC,SAAS,CAAC,CAAC;IAE5D,IAAI,CAAC;QACH,OAAO,QAAQ,CAAC,QAAyB,EAAE,OAAgB,CAAC,CAAC;IAC/D,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,IAAI,KAAK,CACb,kCAAkC,SAAS,MAAM,cAAc,CAAC,KAAK,CAAC,EAAE,CACzE,CAAC;IACJ,CAAC;AACH,CAAC;AAED,SAAS,mBAAmB,CAC1B,YAAkC,EAClC,eAAqC;IAErC,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAU,CAAC;IACnC,IAAI,KAAK,CAAC,OAAO,CAAC,YAAY,CAAC,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC3D,KAAK,MAAM,QAAQ,IAAI,YAAY,EAAE,CAAC;YACpC,IAAI,gBAAgB,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;gBACnC,QAAQ,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YACzB,CAAC;QACH,CAAC;IACH,CAAC;SAAM,CAAC;QACN,KAAK,MAAM,QAAQ,IAAI,gBAAgB,EAAE,CAAC;YACxC,QAAQ,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACzB,CAAC;IACH,CAAC;IAED,IAAI,KAAK,CAAC,OAAO,CAAC,eAAe,CAAC,EAAE,CAAC;QACnC,KAAK,MAAM,QAAQ,IAAI,eAAe,EAAE,CAAC;YACvC,QAAQ,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC5B,CAAC;IACH,CAAC;IAED,OAAO,sBAAsB,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;AAC1E,CAAC;AAED,SAAS,iBAAiB,CAAC,MAA2C;IACpE,MAAM,KAAK,GAAG;QACZ,gEAAgE;QAChE,4GAA4G;QAC5G,iDAAiD;KAClD,CAAC;IAEF,IAAI,MAAM,EAAE,CAAC;QACX,KAAK,CAAC,IAAI,CACR,EAAE,EACF,4DAA4D,EAC5D,0CAA0C,EAC1C,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CACvB,CAAC;IACJ,CAAC;IAED,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC;AAED,SAAS,mBAAmB,CAC1B,SAA+B;IAE/B,MAAM,OAAO,GAAmC,EAAE,CAAC;IAEnD,KAAK,MAAM,KAAK,IAAI,SAAS,CAAC,OAAO,EAAE,CAAC;QACtC,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YAC1B,OAAO,CAAC,IAAI,CAAC;gBACX,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,KAAK,CAAC,IAAI;aACjB,CAAC,CAAC;YACH,SAAS;QACX,CAAC;QAED,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;YAC9B,OAAO,CAAC,IAAI,CAAC;gBACX,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,KAAK,CAAC,QAAQ;aACrB,CAAC,CAAC;YACH,SAAS;QACX,CAAC;QAED,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;YAC9B,OAAO,CAAC,IAAI,CAAC;gBACX,IAAI,EAAE,UAAU;gBAChB,EAAE,EAAE,KAAK,CAAC,EAAE;gBACZ,IAAI,EAAE,KAAK,CAAC,IAAI;gBAChB,KAAK,EAAE,QAAQ,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE;aACxD,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,SAAS,WAAW,CAAC,SAA+B;IAClD,MAAM,UAAU,GAAG,SAAS,CAAC,OAAO;SACjC,MAAM,CAAC,CAAC,KAAK,EAA+E,EAAE,CAAC,CAC9F,KAAK,CAAC,IAAI,KAAK,MAAM,CACtB,CAAC;SACD,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;SACjC,MAAM,CAAC,OAAO,CAAC,CAAC;IAEnB,OAAO,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;AACjC,CAAC;AAED,SAAS,qBAAqB,CAAC,UAAkB;IAC/C,MAAM,SAAS,GAAG,iBAAiB,CAAC,UAAU,CAAC,CAAC;IAChD,IAAI,CAAC,SAAS,EAAE,CAAC;QACf,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,IAAI,CAAC;QACH,OAAO,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;IAC/B,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,SAAS,CAAC;IACnB,CAAC;AACH,CAAC;AAED,SAAS,iBAAiB,CAAC,IAAY;IACrC,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;IAC5B,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,UAAU,GAAG,OAAO,CAAC,KAAK,CAAC,oCAAoC,CAAC,CAAC;IACvE,IAAI,UAAU,EAAE,CAAC;QACf,OAAO,UAAU,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;IAC9B,CAAC;IAED,MAAM,KAAK,GAAG,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;IACnC,MAAM,GAAG,GAAG,OAAO,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;IACrC,IAAI,KAAK,KAAK,CAAC,CAAC,IAAI,GAAG,GAAG,KAAK,EAAE,CAAC;QAChC,OAAO,OAAO,CAAC,KAAK,CAAC,KAAK,EAAE,GAAG,GAAG,CAAC,CAAC,CAAC;IACvC,CAAC;IAED,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,SAAS,cAAc,CAAC,OAA6B;IACnD,OAAO,gBAAgB,CAAC,OAAO,CAAC,KAAK,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC;AACtD,CAAC;AAED,SAAS,iBAAiB,CAAC,KAAyB;IAClD,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,KAAK,IAAI,IAAI,IAAI,KAAK,GAAG,CAAC,EAAE,CAAC;QAC1D,OAAO,iBAAiB,CAAC;IAC3B,CAAC;IAED,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;AAC3B,CAAC;AAED,SAAS,eAAe,CAAC,KAAyB;IAChD,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,KAAK,IAAI,IAAI,IAAI,KAAK,GAAG,CAAC,EAAE,CAAC;QAC1D,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC;AAED,KAAK,UAAU,UAAU,CACvB,OAA0B,EAC1B,QAAgB,EAChB,IAA6B,EAC7B,MAA+B,EAC/B,SAAiB;IAEjB,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC;QAChC,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAW,CAAC;IACxC,CAAC;IAED,OAAO,OAAO,CAAC,OAAO,CAAC,UAAU,CAAC,QAAQ,EAAE,IAAI,EAAE;QAChD,MAAM,EAAE,MAAM,IAAI,IAAI,eAAe,EAAE,CAAC,MAAM;QAC9C,SAAS;KACV,CAAC,CAAC;AACL,CAAC;AAED,SAAS,mBAAmB,CAAC,QAAgB,EAAE,UAAkB,EAAE,OAAgB;IACjF,MAAM,SAAS,GAAG,UAAU;SACzB,KAAK,CAAC,IAAI,CAAC;SACX,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;SAC1B,IAAI,CAAC,OAAO,CAAC,CAAC;IAEjB,IAAI,OAAO,EAAE,CAAC;QACZ,OAAO,GAAG,QAAQ,UAAU,SAAS,CAAC,CAAC,CAAC,KAAK,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC;IAClE,CAAC;IAED,OAAO,GAAG,QAAQ,aAAa,SAAS,CAAC,CAAC,CAAC,KAAK,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC;AACrE,CAAC;AAED,SAAS,WAAW,CAClB,OAAe,EACf,KAAa,EACb,WAAmB,EACnB,QAAgB,EAChB,YAAoB;IAEpB,OAAO;QACL,IAAI,EAAE,QAAQ;QACd,OAAO;QACP,MAAM,EAAE,CAAC,KAAK,CAAC;QACf,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,WAAW;QACpC,QAAQ;QACR,YAAY,EAAE,gBAAgB,CAAC,YAAY,CAAC;KAC7C,CAAC;AACJ,CAAC"}
|