@zhexin/pi-deepseek 0.0.1 → 0.0.2

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@zhexin/pi-deepseek",
   "description": "DeepSeek Enhancement for Pi",
-  "version": "0.0.1",
+  "version": "0.0.2",
   "type": "module",
   "license": "MIT",
   "keywords": [

package/src/index.ts

@@ -1,10 +1,30 @@
-import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
+import type { ExtensionAPI, ExtensionContext } from "@earendil-works/pi-coding-agent";
 import type { Static } from "typebox";
 
 import { fetchInput, fetchOutput, searchInput, searchOutput } from "./schema.ts";
 
 const unauthorized = new Error("Unauthorized. Run `/login deepseek` to authenticate.");
 
+async function resolveDeepSeekRequest(ctx: ExtensionContext, path: string) {
+  const authResult = await ctx.modelRegistry.getProviderAuth("deepseek");
+  const apiKey = authResult?.auth.apiKey;
+  if (apiKey === undefined) throw unauthorized;
+
+  const baseUrl =
+    authResult?.auth.baseUrl ??
+    ctx.modelRegistry.getProvider("deepseek")?.baseUrl ??
+    "https://api.deepseek.com";
+
+  return {
+    url: new URL(path, baseUrl),
+    headers: {
+      ...authResult?.auth.headers,
+      Authorization: `Bearer ${apiKey}`,
+      "Content-Type": "application/json",
+    },
+  };
+}
+
 export default function (pi: ExtensionAPI): void {
   pi.registerTool({
     name: "web-search",
@@ -13,15 +33,11 @@ export default function (pi: ExtensionAPI): void {
     parameters: searchInput,
     outputSchema: searchOutput,
     async execute(_toolCallId, params, signal, _onUpdate, ctx) {
-      const apiKey = await ctx.modelRegistry.getApiKeyForProvider("deepseek");
-      if (apiKey === undefined) throw unauthorized;
+      const { url, headers } = await resolveDeepSeekRequest(ctx, "/v1/tools/search");
 
-      const response = await fetch("https://api.deepseek.com/v1/tools/search", {
+      const response = await fetch(url, {
         method: "POST",
-        headers: {
-          Authorization: `Bearer ${apiKey}`,
-          "Content-Type": "application/json",
-        },
+        headers,
         body: JSON.stringify(params),
         signal: signal ?? null,
       });
@@ -63,15 +79,11 @@ export default function (pi: ExtensionAPI): void {
     parameters: fetchInput,
     outputSchema: fetchOutput,
     async execute(_toolCallId, params, signal, _onUpdate, ctx) {
-      const apiKey = await ctx.modelRegistry.getApiKeyForProvider("deepseek");
-      if (apiKey === undefined) throw unauthorized;
+      const { url, headers } = await resolveDeepSeekRequest(ctx, "/v1/tools/fetch");
 
-      const response = await fetch("https://api.deepseek.com/v1/tools/fetch", {
+      const response = await fetch(url, {
         method: "POST",
-        headers: {
-          Authorization: `Bearer ${apiKey}`,
-          "Content-Type": "application/json",
-        },
+        headers,
         body: JSON.stringify(params),
         signal: signal ?? null,
       });