@aliou/pi-synthetic 0.4.2 → 0.4.3

package/CHANGELOG.md

@@ -1,5 +1,11 @@
 # @aliou/pi-synthetic
 
+## 0.4.3
+
+### Patch Changes
+
+- 7dc1d80: Defer subscription check to session_start for non-blocking extension init.
+
 ## 0.4.2
 
 ### Patch Changes

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aliou/pi-synthetic",
-  "version": "0.4.2",
+  "version": "0.4.3",
   "repository": {
     "type": "git",
     "url": "https://github.com/aliou/pi-synthetic"
@@ -14,6 +14,9 @@
     ],
     "video": "https://assets.aliou.me/pi-extensions/demos/pi-synthetic.mp4"
   },
+  "peerDependencies": {
+    "@mariozechner/pi-coding-agent": ">=0.51.0"
+  },
   "devDependencies": {
     "@biomejs/biome": "^2.3.13",
     "@changesets/cli": "^2.27.11",

package/src/commands/quotas.ts

@@ -1,9 +1,9 @@
 import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
 import type { Component } from "@mariozechner/pi-tui";
-import { QuotasDisplayComponent } from "../components/quotas-display.js";
-import { QuotasErrorComponent } from "../components/quotas-error.js";
-import { QuotasLoadingComponent } from "../components/quotas-loading.js";
-import type { QuotasResponse } from "../types/quotas.js";
+import { QuotasDisplayComponent } from "../components/quotas-display";
+import { QuotasErrorComponent } from "../components/quotas-error";
+import { QuotasLoadingComponent } from "../components/quotas-loading";
+import type { QuotasResponse } from "../types/quotas";
 
 export function registerQuotasCommand(pi: ExtensionAPI): void {
   pi.registerCommand("synthetic:quotas", {
@@ -19,7 +19,7 @@ export function registerQuotasCommand(pi: ExtensionAPI): void {
         return;
       }
 
-      await ctx.ui.custom<void>((tui, theme, _kb, done) => {
+      const result = await ctx.ui.custom<void>((tui, theme, _kb, done) => {
         let currentComponent: Component = new QuotasLoadingComponent(theme);
 
         fetchQuotas()
@@ -50,6 +50,16 @@ export function registerQuotasCommand(pi: ExtensionAPI): void {
           },
         };
       });
+
+      // RPC fallback: custom() returned undefined
+      if (result === undefined) {
+        const quotas = await fetchQuotas();
+        if (!quotas) {
+          ctx.ui.notify("Failed to fetch quotas", "error");
+          return;
+        }
+        ctx.ui.notify(formatQuotasPlain(quotas), "info");
+      }
     },
   });
 }

package/src/components/quotas-display.ts

@@ -1,7 +1,7 @@
 import type { Theme } from "@mariozechner/pi-coding-agent";
 import { DynamicBorder } from "@mariozechner/pi-coding-agent";
 import { type Component, Container, Text } from "@mariozechner/pi-tui";
-import type { QuotasResponse } from "../types/quotas.js";
+import type { QuotasResponse } from "../types/quotas";
 
 export class QuotasDisplayComponent implements Component {
   private container: Container;

package/src/index.ts

@@ -1,7 +1,7 @@
 import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
-import { registerQuotasCommand } from "./commands/quotas.js";
-import { registerSyntheticProvider } from "./providers/index.js";
-import { registerSyntheticWebSearchTool } from "./tools/search.js";
+import { registerQuotasCommand } from "./commands/quotas";
+import { registerSyntheticProvider } from "./providers/index";
+import { registerSyntheticWebSearchTool } from "./tools/search";
 
 export default async function (pi: ExtensionAPI) {
   registerSyntheticProvider(pi);
@@ -9,6 +9,6 @@ export default async function (pi: ExtensionAPI) {
   // Only register quotas command and web search tool if API key is available
   if (process.env.SYNTHETIC_API_KEY) {
     registerQuotasCommand(pi);
-    await registerSyntheticWebSearchTool(pi);
+    registerSyntheticWebSearchTool(pi);
   }
 }

package/src/providers/index.ts

@@ -1,5 +1,5 @@
 import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
-import { SYNTHETIC_MODELS } from "./models.js";
+import { SYNTHETIC_MODELS } from "./models";
 
 export function registerSyntheticProvider(pi: ExtensionAPI): void {
   pi.registerProvider("synthetic", {

package/src/tools/search.ts

@@ -73,27 +73,30 @@ async function checkSubscriptionAccess(
 }
 
 // Tool Registration
-export async function registerSyntheticWebSearchTool(pi: ExtensionAPI) {
-  // Check for API key
+export function registerSyntheticWebSearchTool(pi: ExtensionAPI) {
   const apiKey = process.env.SYNTHETIC_API_KEY;
   if (!apiKey) {
     return;
   }
 
-  // Only register if user has subscription access (search is subscription-only)
-  const access = await checkSubscriptionAccess(apiKey);
-  if (!access.ok) {
-    pi.on("session_start", async (_event, ctx) => {
+  // Check subscription on session start, only register tool if access is granted
+  pi.on("session_start", async (_event, ctx) => {
+    const access = await checkSubscriptionAccess(apiKey);
+    if (!access.ok) {
       if (ctx.hasUI) {
         ctx.ui.notify(
           `Synthetic web search disabled: ${access.reason}`,
           "warning",
         );
       }
-    });
-    return;
-  }
+      return;
+    }
 
+    registerTool(pi, apiKey);
+  });
+}
+
+function registerTool(pi: ExtensionAPI, apiKey: string) {
   pi.registerTool<typeof SearchParams, WebSearchDetails>({
     name: "synthetic_web_search",
     label: "Synthetic: Web Search",
@@ -105,21 +108,13 @@ export async function registerSyntheticWebSearchTool(pi: ExtensionAPI) {
       _toolCallId: string,
       params: SearchParamsType,
       signal: AbortSignal | undefined,
-      onUpdate: (result: AgentToolResult<WebSearchDetails>) => void,
+      onUpdate:
+        | ((result: AgentToolResult<WebSearchDetails>) => void)
+        | undefined,
       _ctx: ExtensionContext,
     ): Promise<AgentToolResult<WebSearchDetails>> {
-      // Check for API key
-      const apiKey = process.env.SYNTHETIC_API_KEY;
-      if (!apiKey) {
-        const error = "SYNTHETIC_API_KEY environment variable is required";
-        return {
-          content: [{ type: "text", text: `Error: ${error}` }],
-          details: { error, isError: true },
-        };
-      }
-
       // Send progress update
-      onUpdate({
+      onUpdate?.({
         content: [{ type: "text", text: "Searching..." }],
         details: { query: params.query },
       });