@pasko70/pibo 1.16.6 → 2.1.1

package/dist/core/profiles.js

@@ -1,6 +1,22 @@
+import { normalizePiboToolDefinition, } from "../tools/contract.js";
+export function normalizeToolProfile(tool) {
+    const { definition, createDefinition, ...profile } = tool;
+    return {
+        ...profile,
+        ...(definition ? { definition: normalizePiboToolDefinition(definition) } : {}),
+        ...(createDefinition
+            ? {
+                createDefinition: (context) => normalizePiboToolDefinition(createDefinition(context)),
+            }
+            : {}),
+    };
+}
+export const DEFAULT_AGENT_RUNTIME_INSTANCE_ID = "pi";
 export const DEFAULT_BUILTIN_TOOL_NAMES = ["read", "bash", "edit", "write"];
 export class InitialSessionContext {
     profileName;
+    runtimeInstanceId;
+    runtimeOptions;
     sessionId;
     parentSessionId;
     model;
@@ -24,6 +40,8 @@ export class InitialSessionContext {
     toolPackages;
     constructor(options) {
         this.profileName = options.profileName;
+        this.runtimeInstanceId = options.runtimeInstanceId ?? DEFAULT_AGENT_RUNTIME_INSTANCE_ID;
+        this.runtimeOptions = structuredClone(options.runtimeOptions ?? {});
         this.sessionId = options.sessionId;
         this.parentSessionId = options.parentSessionId;
         this.model = options.model ? { ...options.model } : undefined;
@@ -36,7 +54,7 @@ export class InitialSessionContext {
         this.mainFast = options.mainFast;
         this.subagentFast = options.subagentFast;
         this.skills = [...(options.skills ?? [])];
-        this.tools = [...(options.tools ?? [])];
+        this.tools = (options.tools ?? []).map(normalizeToolProfile);
         this.subagents = [...(options.subagents ?? [])];
         this.mcpServers = [...(options.mcpServers ?? [])];
         this.piPackages = [...(options.piPackages ?? [])];
@@ -49,6 +67,8 @@ export class InitialSessionContext {
 }
 export class InitialSessionContextBuilder {
     profileName;
+    runtimeInstanceId = DEFAULT_AGENT_RUNTIME_INSTANCE_ID;
+    runtimeOptions = {};
     sessionId;
     parentSessionId;
     model;
@@ -73,6 +93,11 @@ export class InitialSessionContextBuilder {
     constructor(profileName) {
         this.profileName = profileName;
     }
+    withAgentRuntime(instanceId, options = {}) {
+        this.runtimeInstanceId = instanceId;
+        this.runtimeOptions = structuredClone(options);
+        return this;
+    }
     withSessionId(sessionId) {
         this.sessionId = sessionId;
         return this;
@@ -142,11 +167,11 @@ export class InitialSessionContextBuilder {
         return this;
     }
     addTool(tool) {
-        this.tools.push(tool);
+        this.tools.push(normalizeToolProfile(tool));
         return this;
     }
     addTools(tools) {
-        this.tools.push(...tools);
+        this.tools.push(...tools.map(normalizeToolProfile));
         return this;
     }
     addSubagent(subagent) {
@@ -184,6 +209,8 @@ export class InitialSessionContextBuilder {
     createSession() {
         return new InitialSessionContext({
             profileName: this.profileName,
+            runtimeInstanceId: this.runtimeInstanceId,
+            runtimeOptions: this.runtimeOptions,
             sessionId: this.sessionId,
             parentSessionId: this.parentSessionId,
             model: this.model,