@ai-sdk/harness-pi 1.0.36 → 1.0.38

package/CHANGELOG.md

@@ -1,5 +1,24 @@
 # @ai-sdk/harness-pi
 
+## 1.0.38
+
+### Patch Changes
+
+- f5cdb2d: chore(harness): update primary SDK dependencies
+- Updated dependencies [02ffdcb]
+- Updated dependencies [76cb673]
+  - @ai-sdk/provider-utils@5.0.12
+  - @ai-sdk/harness@1.0.38
+
+## 1.0.37
+
+### Patch Changes
+
+- 81c3aa5: feat(harness-pi): add `agentDir` option to `createPi` for reusing CLI config
+- Updated dependencies [b460541]
+- Updated dependencies [079591e]
+  - @ai-sdk/harness@1.0.37
+
 ## 1.0.36
 
 ### Patch Changes

package/dist/index.d.ts

@@ -45,6 +45,13 @@ type PiHarnessSettings = {
      * `thinkingLevel` option on `createAgentSession`.
      */
     readonly thinkingLevel?: PiThinkingLevel;
+    /**
+     * Directory holding Pi's global agent config (auth.json, models.json,
+     * settings.json). When omitted, a per-session temp dir is used. Pass the
+     * user's agent dir (e.g. `~/.pi/agent/`) to reuse their CLI auth and
+     * model settings.
+     */
+    readonly agentDir?: string;
 };
 declare const PI_BUILTIN_TOOLS: {
     readonly read: HarnessV1BuiltinTool<{

package/dist/index.js

@@ -90,11 +90,11 @@ async function pullSessionFileFromSandbox(args) {
 
 // src/pi-session.ts
 import {
-  AuthStorage,
   createAgentSession,
   DefaultResourceLoader,
   defineTool,
   ModelRegistry,
+  ModelRuntime,
   SessionManager,
   SettingsManager
 } from "@earendil-works/pi-coding-agent";
@@ -108,7 +108,7 @@ import { resolveSandboxHomeDir } from "@ai-sdk/harness/utils";
 import { getAiGatewayAuthFromEnv } from "@ai-sdk/harness/utils";
 
 // src/version.ts
-var VERSION = true ? "1.0.36" : "0.0.0-test";
+var VERSION = true ? "1.0.38" : "0.0.0-test";
 
 // src/pi-auth.ts
 var DEFAULT_GATEWAY_BASE_URL = "https://ai-gateway.vercel.sh";
@@ -130,9 +130,14 @@ function createGatewayProviderConfig({
     }
   };
 }
-function register(registries, provider, apiKey, config) {
-  registries.authStorage.setRuntimeApiKey(provider, apiKey);
+async function register({
+  registries,
+  provider,
+  apiKey,
+  config
+}) {
   registries.modelRegistry.registerProvider(provider, config);
+  await registries.modelRuntime.setRuntimeApiKey(provider, apiKey);
 }
 function hasConfiguredValue(value) {
   if (value == null) return false;
@@ -142,17 +147,11 @@ function hasConfiguredValue(value) {
 }
 function resolvePiEnv({
   options,
-  env,
-  registries,
-  clientApp = HARNESS_CLIENT_APP
+  env
 }) {
   const customEnvConfigured = hasConfiguredValue(options?.customEnv);
   if (customEnvConfigured) {
-    return applyCustomEnv({
-      customEnv: options.customEnv ?? {},
-      registries,
-      clientApp
-    });
+    return resolveCustomEnv({ customEnv: options.customEnv ?? {} });
   }
   const gatewayConfigured = hasConfiguredValue(options?.gateway);
   const gatewayAuthFromEnv = getAiGatewayAuthFromEnv({ env });
@@ -160,31 +159,11 @@ function resolvePiEnv({
     const apiKey = options.gateway?.apiKey ?? gatewayAuthFromEnv.apiKey;
     const baseUrl = options.gateway?.baseUrl ?? gatewayAuthFromEnv.baseUrl;
     if (apiKey) {
-      register(
-        registries,
-        "vercel-ai-gateway",
-        apiKey,
-        createGatewayProviderConfig({
-          apiKey,
-          baseUrl,
-          clientApp
-        })
-      );
       return { AI_GATEWAY_API_KEY: apiKey, AI_GATEWAY_BASE_URL: baseUrl };
     }
     return {};
   }
   if (gatewayAuthFromEnv.apiKey) {
-    register(
-      registries,
-      "vercel-ai-gateway",
-      gatewayAuthFromEnv.apiKey,
-      createGatewayProviderConfig({
-        apiKey: gatewayAuthFromEnv.apiKey,
-        baseUrl: gatewayAuthFromEnv.baseUrl,
-        clientApp
-      })
-    );
     return {
       AI_GATEWAY_API_KEY: gatewayAuthFromEnv.apiKey,
       AI_GATEWAY_BASE_URL: gatewayAuthFromEnv.baseUrl
@@ -192,46 +171,87 @@ function resolvePiEnv({
   }
   return {};
 }
-function applyCustomEnv({
+async function registerPiProviders({
+  options,
+  resolvedEnv,
+  registries,
+  clientApp = HARNESS_CLIENT_APP
+}) {
+  if (hasConfiguredValue(options?.customEnv)) {
+    await registerCustomProviders({
+      customEnv: options.customEnv ?? {},
+      registries,
+      clientApp
+    });
+    return;
+  }
+  const apiKey = resolvedEnv.AI_GATEWAY_API_KEY;
+  const baseUrl = resolvedEnv.AI_GATEWAY_BASE_URL;
+  if (!apiKey || !baseUrl) return;
+  await register({
+    registries,
+    provider: "vercel-ai-gateway",
+    apiKey,
+    config: createGatewayProviderConfig({ apiKey, baseUrl, clientApp })
+  });
+}
+function resolveCustomEnv({
+  customEnv
+}) {
+  const apiKey = customEnv.AI_GATEWAY_API_KEY;
+  if (!apiKey) return {};
+  return {
+    AI_GATEWAY_API_KEY: apiKey,
+    AI_GATEWAY_BASE_URL: customEnv.AI_GATEWAY_BASE_URL ?? DEFAULT_GATEWAY_BASE_URL
+  };
+}
+async function registerCustomProviders({
   customEnv,
   registries,
   clientApp
 }) {
-  const out = {};
   const gatewayKey = customEnv.AI_GATEWAY_API_KEY;
   if (gatewayKey) {
     const baseUrl = customEnv.AI_GATEWAY_BASE_URL ?? DEFAULT_GATEWAY_BASE_URL;
-    register(
+    await register({
       registries,
-      "vercel-ai-gateway",
-      gatewayKey,
-      createGatewayProviderConfig({
+      provider: "vercel-ai-gateway",
+      apiKey: gatewayKey,
+      config: createGatewayProviderConfig({
         apiKey: gatewayKey,
         baseUrl,
         clientApp
       })
-    );
-    out.AI_GATEWAY_API_KEY = gatewayKey;
-    out.AI_GATEWAY_BASE_URL = baseUrl;
+    });
   }
   if (customEnv.OPENAI_API_KEY) {
     const baseUrl = customEnv.OPENAI_BASE_URL ?? DEFAULT_OPENAI_BASE_URL;
-    register(registries, "openai", customEnv.OPENAI_API_KEY, {
+    await register({
+      registries,
+      provider: "openai",
       apiKey: customEnv.OPENAI_API_KEY,
-      baseUrl,
-      authHeader: true
+      config: {
+        apiKey: customEnv.OPENAI_API_KEY,
+        baseUrl,
+        authHeader: true
+      }
     });
   }
   if (customEnv.ANTHROPIC_API_KEY) {
     const baseUrl = customEnv.ANTHROPIC_BASE_URL ?? DEFAULT_ANTHROPIC_BASE_URL;
-    register(registries, "anthropic", customEnv.ANTHROPIC_API_KEY, {
+    await register({
+      registries,
+      provider: "anthropic",
       apiKey: customEnv.ANTHROPIC_API_KEY,
-      baseUrl,
-      ...customEnv.ANTHROPIC_AUTH_TOKEN ? {
-        headers: {
-          authorization: `Bearer ${customEnv.ANTHROPIC_AUTH_TOKEN}`
-        }
-      } : {}
+      config: {
+        apiKey: customEnv.ANTHROPIC_API_KEY,
+        baseUrl,
+        ...customEnv.ANTHROPIC_AUTH_TOKEN ? {
+          headers: {
+            authorization: `Bearer ${customEnv.ANTHROPIC_AUTH_TOKEN}`
+          }
+        } : {}
+      }
     });
   }
   for (const [name, apiKey] of Object.entries(customEnv)) {
@@ -247,13 +267,17 @@ function applyCustomEnv({
     if (!baseUrl) {
       continue;
     }
-    register(registries, provider, apiKey, {
+    await register({
+      registries,
+      provider,
       apiKey,
-      baseUrl,
-      authHeader: true
+      config: {
+        apiKey,
+        baseUrl,
+        authHeader: true
+      }
     });
   }
-  return out;
 }
 
 // src/pi-events.ts
@@ -344,7 +368,10 @@ function extractAssistantText(message) {
 // src/pi-model-resolver.ts
 import { getAiGatewayAuthFromEnv as getAiGatewayAuthFromEnv2 } from "@ai-sdk/harness/utils";
 var DEFAULT_PI_GATEWAY_MODEL_ID = "anthropic/claude-sonnet-4.6";
-function createPiModelResolver(modelRegistry, env = process.env) {
+function createPiModelResolver({
+  modelRegistry,
+  env = process.env
+}) {
   let cachedModels;
   const loadModels = () => {
     if (cachedModels) {
@@ -1617,22 +1644,31 @@ async function createPiSession(input) {
     sandboxWorkDir: sessionWorkDir,
     readableRoots: sandboxSkillRootDir ? [{ sandboxDir: sandboxSkillRootDir }] : []
   });
-  const authStorage = AuthStorage.create(path7.join(hostAgentDir, "auth.json"));
-  const modelRegistry = ModelRegistry.create(
-    authStorage,
-    path7.join(hostAgentDir, "models.json")
-  );
-  const settingsManager = SettingsManager.inMemory();
+  const agentDir = input.agentDir ?? hostAgentDir;
+  const modelRuntime = await ModelRuntime.create({
+    authPath: path7.join(agentDir, "auth.json"),
+    modelsPath: path7.join(agentDir, "models.json"),
+    allowModelNetwork: false
+  });
+  const modelRegistry = new ModelRegistry(modelRuntime);
+  const settingsManager = input.agentDir != null ? SettingsManager.create(hostWorkDir, agentDir) : SettingsManager.inMemory();
   const resolverEnv = resolvePiEnv({
     options: input.settings.auth,
-    env: process.env,
+    env: process.env
+  });
+  await registerPiProviders({
+    options: input.settings.auth,
+    resolvedEnv: resolverEnv,
     registries: {
-      authStorage,
-      modelRegistry
+      modelRegistry,
+      modelRuntime
     },
     clientApp: input.clientApp
   });
-  const resolveModel = createPiModelResolver(modelRegistry, resolverEnv);
+  const resolveModel = createPiModelResolver({
+    modelRegistry,
+    env: resolverEnv
+  });
   const resolvedModel = resolveModel(input.settings.model);
   const resourceLoader = new DefaultResourceLoader({
     cwd: sessionWorkDir,
@@ -1806,8 +1842,7 @@ async function createPiSession(input) {
     const { session } = await createAgentSession({
       cwd: sessionWorkDir,
       agentDir: hostAgentDir,
-      authStorage,
-      modelRegistry,
+      modelRuntime,
       sessionManager,
       settingsManager,
       resourceLoader,
@@ -2400,7 +2435,8 @@ function createPi(settings = {}) {
         permissionMode: startOpts.permissionMode,
         builtinToolFiltering: startOpts.builtinToolFiltering,
         ...resumeData?.sessionFileName ? { resumeSessionFileName: resumeData.sessionFileName } : {},
-        ...startOpts.abortSignal ? { abortSignal: startOpts.abortSignal } : {}
+        ...startOpts.abortSignal ? { abortSignal: startOpts.abortSignal } : {},
+        ...settings.agentDir ? { agentDir: settings.agentDir } : {}
       });
     }
   };