From 05acf56fbef13a3b18cbe3bce27d25e8071d0a53 Mon Sep 17 00:00:00 2001
From: wangqiang <wangqiang@lattebank.com>
Date: Thu, 5 Mar 2026 18:04:07 +0800
Subject: [PATCH] fix(moonshot): Enhance moonshot provider configuration:
 prioritize baseUrl from config for api.moonshot.cn integration.

---
 src/agents/pi-embedded-runner/run.ts | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/src/agents/pi-embedded-runner/run.ts b/src/agents/pi-embedded-runner/run.ts
index bfda498f5e3..19213f47287 100644
--- a/src/agents/pi-embedded-runner/run.ts
+++ b/src/agents/pi-embedded-runner/run.ts
@@ -318,6 +318,14 @@ export async function runEmbeddedPiAgent(
         });
       }
 
+      // Moonshot: prefer config baseUrl so .cn key is used with api.moonshot.cn (not .ai).
+      if (normalizeProviderId(provider) === "moonshot") {
+        const moonshotBaseUrl = params.config?.models?.providers?.moonshot?.baseUrl;
+        if (typeof moonshotBaseUrl === "string" && moonshotBaseUrl.trim()) {
+          (model as { baseUrl?: string }).baseUrl = moonshotBaseUrl.trim();
+        }
+      }
+
       const ctxInfo = resolveContextWindowInfo({
         cfg: params.config,
         provider,