shariq-pi-extensions 0.2.2 → 0.2.4

package/extensions/antigravity-provider/antigravity/index.ts

@@ -66,7 +66,12 @@ export default function antigravityProviderExtension(pi: ExtensionAPI) {
 		models: ANTIGRAVITY_MODELS,
 		oauth: {
 			name: PROVIDER_NAME,
-			login: loginAntigravity as any,
+			login: ((callbacks: Parameters<typeof loginAntigravity>[0]) => {
+				// Pi replaces its single stored provider credential after login. Archive
+				// the current account first so signing into another account cannot lose it.
+				migrateStoredCredential();
+				return loginAntigravity(callbacks);
+			}) as any,
 			refreshToken: refreshAntigravityToken as any,
 			getApiKey: getApiKey as any,
 		},

package/extensions/subagents/index.ts

@@ -480,7 +480,7 @@ export default function (pi: ExtensionAPI) {
             childCwd: cwd,
             parentTrusted: ctx.isProjectTrusted(),
           }),
-          inheritedModel: ctx.model ? { provider: ctx.model.provider, id: ctx.model.id } : undefined,
+          inheritedModel: ctx.model ?? undefined,
           inheritedThinkingLevel: pi.getThinkingLevel(),
           inheritedMessages,
           parentSessionFile: ctx.sessionManager.getSessionFile(),

package/extensions/subagents/src/backends/pi.ts

@@ -186,14 +186,17 @@ function createParentBridgeTools(task: SpawnTask): ToolDefinition[] {
  * then must be unambiguous across providers. No hint inherits the parent
  * model; with nothing to inherit, the SDK default applies.
  */
-function resolvePiModel(
+export function resolvePiModel(
   registry: ModelRegistry,
   hint: string | undefined,
-  inherited: { provider: string; id: string } | undefined,
+  inherited: Model<any> | undefined,
 ): Model<any> | undefined {
   if (!hint) {
     if (!inherited) return undefined;
-    return registry.find(inherited.provider, inherited.id) ?? undefined;
+    return registry.find(inherited.provider, inherited.id) ?? inherited;
+  }
+  if (inherited?.id === hint) {
+    return registry.find(inherited.provider, hint) ?? inherited;
   }
   const slash = hint.indexOf("/");
   if (slash > 0) {
@@ -201,11 +204,13 @@ function resolvePiModel(
     const id = hint.slice(slash + 1);
     const found = registry.find(provider, id);
     if (found) return found;
+    if (inherited?.provider === provider && inherited.id === id) return inherited;
     throw new Error(`Unknown model "${hint}".`);
   }
   if (inherited) {
     const found = registry.find(inherited.provider, hint);
     if (found) return found;
+    if (inherited.id === hint) return inherited;
   }
   const matches = registry.getAll().filter((m) => m.id === hint);
   if (matches.length === 1) return matches[0];

package/extensions/subagents/src/domain.ts

@@ -6,7 +6,7 @@
  * `SubagentEvent` union.
  */
 
-import type { Message } from "@earendil-works/pi-ai";
+import type { Message, Model } from "@earendil-works/pi-ai";
 import type { ModelRegistry } from "@earendil-works/pi-coding-agent";
 import { Data } from "effect";
 import type { CapabilityMode, IsolationMode } from "./config.ts";
@@ -50,7 +50,7 @@ export interface ParentContext {
   /** In-process bridge used by child-only message_parent/ask_parent tools. */
   readonly bridge?: ParentBridge;
   /** Parent pi model, for the pi backend's "inherit" default. */
-  readonly inheritedModel?: { readonly provider: string; readonly id: string };
+  readonly inheritedModel?: Model<any>;
   readonly inheritedThinkingLevel?: string;
   /** Parent model registry; required by the pi backend to resolve models. */
   readonly modelRegistry?: ModelRegistry;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "shariq-pi-extensions",
-  "version": "0.2.2",
+  "version": "0.2.4",
   "description": "Cross-platform extension suite for the Pi coding agent.",
   "license": "MIT",
   "author": "Shariq Riaz",