@ai-sdk/harness-pi 1.0.103 → 1.0.104

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ai-sdk/harness-pi",
-  "version": "1.0.103",
+  "version": "1.0.104",
   "type": "module",
   "license": "Apache-2.0",
   "sideEffects": false,
@@ -26,7 +26,7 @@
     }
   },
   "dependencies": {
-    "@ai-sdk/harness": "1.0.101",
+    "@ai-sdk/harness": "1.0.102",
     "@ai-sdk/provider-utils": "5.0.36",
     "@earendil-works/pi-ai": "0.74.2",
     "@earendil-works/pi-coding-agent": "^0.84.3",
@@ -37,7 +37,7 @@
     "zod": "^3.25.76 || ^4.1.8"
   },
   "devDependencies": {
-    "@ai-sdk/sandbox-just-bash": "1.0.101",
+    "@ai-sdk/sandbox-just-bash": "1.0.102",
     "@types/node": "22.19.19",
     "@vercel/ai-tsconfig": "0.0.0",
     "tsup": "^8.5.1",

package/src/pi-auth.ts

@@ -174,16 +174,19 @@ function createGatewayProviderConfig({
   apiKey,
   baseUrl,
   clientApp,
+  headers,
 }: {
   apiKey: string;
   baseUrl: string;
   clientApp: string;
+  headers?: Readonly<Record<string, string>>;
 }): ProviderConfigInput {
   return {
     apiKey,
     baseUrl,
     authHeader: true,
     headers: {
+      ...headers,
       'User-Agent': clientApp,
       'x-client-app': clientApp,
     },
@@ -311,11 +314,13 @@ export async function registerPiProviders({
   resolvedEnv,
   registries,
   clientApp = HARNESS_CLIENT_APP,
+  headers,
 }: {
   options: PiAuthenticationMode | undefined;
   resolvedEnv: Record<string, string>;
   registries: PiRegistries;
   clientApp?: string;
+  headers?: Readonly<Record<string, string>>;
 }): Promise<void> {
   const suppliedEnvironment = isHarnessAuthenticationEnvironment(options);
   const authenticationEnvironment = suppliedEnvironment ? options : process.env;
@@ -328,6 +333,7 @@ export async function registerPiProviders({
         customEnv: { ...pickOpenAIEnv(authenticationEnvironment), ...env },
         registries,
         clientApp,
+        headers,
       });
       return;
     }
@@ -337,6 +343,7 @@ export async function registerPiProviders({
         customEnv: { ...pickAnthropicEnv(authenticationEnvironment), ...env },
         registries,
         clientApp,
+        headers,
       });
       return;
     }
@@ -347,6 +354,7 @@ export async function registerPiProviders({
         customEnv: { ...pickProviderEnv(authenticationEnvironment), ...env },
         registries,
         clientApp,
+        headers,
       });
       return;
     }
@@ -367,6 +375,7 @@ export async function registerPiProviders({
           apiKey: gatewayApiKey,
           baseUrl: gatewayBaseUrl,
           clientApp,
+          headers,
         }),
       });
       return;
@@ -391,6 +400,7 @@ export async function registerPiProviders({
             apiKey: gatewayApiKey,
             baseUrl: gatewayBaseUrl,
             clientApp,
+            headers,
           }),
         });
         return;
@@ -400,6 +410,7 @@ export async function registerPiProviders({
         customEnv: { ...pickProviderEnv(authenticationEnvironment), ...env },
         registries,
         clientApp,
+        headers,
       });
       return;
     }
@@ -455,10 +466,12 @@ async function registerCustomProviders({
   customEnv,
   registries,
   clientApp,
+  headers,
 }: {
   customEnv: Record<string, string>;
   registries: PiRegistries;
   clientApp: string;
+  headers?: Readonly<Record<string, string>>;
 }): Promise<void> {
   const gatewayKey = customEnv.AI_GATEWAY_API_KEY;
   if (gatewayKey) {
@@ -471,6 +484,7 @@ async function registerCustomProviders({
         apiKey: gatewayKey,
         baseUrl,
         clientApp,
+        headers,
       }),
     });
   }
@@ -485,6 +499,7 @@ async function registerCustomProviders({
         apiKey: customEnv.OPENAI_API_KEY,
         baseUrl,
         authHeader: true,
+        ...(headers ? { headers } : {}),
       },
     });
   }
@@ -498,10 +513,15 @@ async function registerCustomProviders({
       config: {
         apiKey: customEnv.ANTHROPIC_API_KEY,
         baseUrl,
-        ...(customEnv.ANTHROPIC_AUTH_TOKEN
+        ...(headers || customEnv.ANTHROPIC_AUTH_TOKEN
           ? {
               headers: {
-                authorization: `Bearer ${customEnv.ANTHROPIC_AUTH_TOKEN}`,
+                ...headers,
+                ...(customEnv.ANTHROPIC_AUTH_TOKEN
+                  ? {
+                      authorization: `Bearer ${customEnv.ANTHROPIC_AUTH_TOKEN}`,
+                    }
+                  : {}),
               },
             }
           : {}),
@@ -534,6 +554,7 @@ async function registerCustomProviders({
         apiKey,
         baseUrl,
         authHeader: true,
+        ...(headers ? { headers } : {}),
       },
     });
   }

package/src/pi-harness.ts

@@ -161,6 +161,7 @@ export function createPi(
           ...(settings.extensionFactories
             ? { extensionFactories: settings.extensionFactories }
             : {}),
+          ...(startOpts.headers ? { headers: startOpts.headers } : {}),
         },
         clientApp: PI_CLIENT_APP,
         isResume: lifecycleState != null,

package/src/pi-session.ts

@@ -33,6 +33,7 @@ import {
 import {
   getRestrictedSandboxSession,
   resolveSandboxHomeDir,
+  writeSkills,
 } from '@ai-sdk/harness/utils';
 import type { Experimental_SandboxSession as SandboxSession } from '@ai-sdk/provider-utils';
 import {
@@ -45,7 +46,7 @@ import { getPiTerminalError, parseNativeEvent } from './pi-events';
 import { createPiModelResolver } from './pi-model-resolver';
 import { createPiPathMapper } from './pi-paths';
 import { createPiRemoteOps, type PiRemoteOps } from './pi-remote-ops';
-import { writePiSkills } from './pi-skills';
+
 import {
   persistSessionFileToSandbox,
   pullSessionFileFromSandbox,
@@ -217,6 +218,7 @@ export type PiThinkingLevel =
 
 export interface PiSessionSettings {
   readonly auth?: PiAuthenticationMode;
+  readonly headers?: Readonly<Record<string, string>>;
   readonly model?: string;
   readonly thinkingLevel?: PiThinkingLevel;
   readonly mcpServers?: Record<string, unknown>;
@@ -410,6 +412,7 @@ export async function createPiSession(
       modelRuntime,
     },
     clientApp: input.clientApp,
+    headers: input.settings.headers,
   });
   const resolveModel = createPiModelResolver({
     modelRegistry,
@@ -1071,11 +1074,15 @@ export async function createPiSession(
       throw new Error('Pi session has been stopped.');
     }
 
-    const skillWriteResult = await writePiSkills({
+    const skillWriteResult = await writeSkills({
       sandbox: toolSafeSandboxSession,
-      sandboxHomeDir,
+      homePath: sandboxHomeDir,
+      skillsDir: '.agents/skills',
       skills: turnOpts.skills,
       abortSignal: turnOpts.abortSignal,
+      invalidSkillNameMessage: ({ name }) => `Invalid Pi skill name: ${name}`,
+      invalidSkillFilePathMessage: ({ skillName, filePath }) =>
+        `Invalid Pi skill file path for ${skillName}: ${filePath}`,
     });
     harnessSkills = createHarnessPiSkills({
       skills: turnOpts.skills,

package/src/pi-skills.ts

@@ -1,25 +0,0 @@
-import path from 'node:path';
-import type { HarnessV1Skill } from '@ai-sdk/harness';
-import { writeSkills, type WriteSkillsResult } from '@ai-sdk/harness/utils';
-import type { Experimental_SandboxSession } from '@ai-sdk/provider-utils';
-
-/**
- * Materialize Pi skills as files under
- * `$HOME/.agents/skills/<name>/SKILL.md` inside the sandbox.
- */
-export async function writePiSkills(args: {
-  readonly sandbox: Experimental_SandboxSession;
-  readonly sandboxHomeDir: string;
-  readonly skills: ReadonlyArray<HarnessV1Skill>;
-  readonly abortSignal?: AbortSignal;
-}): Promise<WriteSkillsResult> {
-  return writeSkills({
-    sandbox: args.sandbox,
-    rootDir: path.posix.join(args.sandboxHomeDir, '.agents', 'skills'),
-    skills: args.skills,
-    abortSignal: args.abortSignal,
-    invalidSkillNameMessage: ({ name }) => `Invalid Pi skill name: ${name}`,
-    invalidSkillFilePathMessage: ({ skillName, filePath }) =>
-      `Invalid Pi skill file path for ${skillName}: ${filePath}`,
-  });
-}