From a63851135e70a3125b3c7cb331149b5758030d1b Mon Sep 17 00:00:00 2001
From: zengdard <61885397+zengdard@users.noreply.github.com>
Date: Sun, 19 Jul 2026 22:41:32 +0200
Subject: [PATCH] feat(pi): add DeepSeek provider support

Register DeepSeek as a runtime provider when DEEPSEEK_API_KEY is set,
enabling deepseek/deepseek-* models to be used directly without the
Vercel AI Gateway.

- Detect DEEPSEEK_API_KEY env var at Pi session creation
- Register deepseek provider with configurable base URL
- Support deepseek/deepseek-v4-flash and similar model references
---
 packages/processor/src/agents/pi-sdk.ts | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/packages/processor/src/agents/pi-sdk.ts b/packages/processor/src/agents/pi-sdk.ts
index c888ca3..d63bf32 100644
--- a/packages/processor/src/agents/pi-sdk.ts
+++ b/packages/processor/src/agents/pi-sdk.ts
@@ -465,6 +465,17 @@ async function createPiSession(projectRoot: string, cfg: PiAgentConfig): Promise
   const modelRegistry = ModelRegistry.create(authStorage, path.join(getAgentDir(), "models.json"));
   configureProviderOverrides(modelRegistry, cfg);
 
+  if (process.env.DEEPSEEK_API_KEY) {
+    const dsKey = process.env.DEEPSEEK_API_KEY;
+    const dsBaseUrl = process.env.DEEPSEEK_BASE_URL || "https://api.deepseek.com/v1";
+    authStorage.setRuntimeApiKey("deepseek", dsKey);
+    modelRegistry.registerProvider("deepseek", {
+      baseUrl: dsBaseUrl,
+      apiKey: dsKey,
+      api: "openai-completions",
+    });
+  }
+
   const modelName = cfg.model ?? DEFAULT_MODEL;
   const model = await resolvePiModelWithDynamicGateway(modelRegistry, modelName, cfg);
   const agentDir = getAgentDir();