fix: resolve CI failures (cohere TS2722 + surface report budget)

xydt-tanshanshan · xydt-tanshanshan · commit 4b9afeea003c · 2026-06-18T11:52:19.000+08:00
diff --git a/extensions/cohere/index.test.ts b/extensions/cohere/index.test.ts
@@ -36,9 +36,6 @@ function captureCoherePayload(context: Context): Record<string, unknown> {
   };
 
   const wrappedStreamFn = createCohereCompletionsWrapper(baseStreamFn);
-  if (!wrappedStreamFn) {
-    throw new Error("Cohere wrapper did not return a stream function");
-  }
   void wrappedStreamFn(requireCohereModel(), context, {
     onPayload: (payload) => {
       captured = payload as Record<string, unknown>;
diff --git a/extensions/cohere/stream.ts b/extensions/cohere/stream.ts
@@ -1,8 +1,7 @@
-import type { ProviderWrapStreamFnContext } from "openclaw/plugin-sdk/plugin-entry";
+import type { StreamFn } from "openclaw/plugin-sdk/agent-core";
 import { createPayloadPatchStreamWrapper } from "openclaw/plugin-sdk/provider-stream-shared";
 
 function patchCoherePayload(payload: Record<string, unknown>): void {
-  // Cohere's Compatibility API uses developer, not system, for instructions.
   if (Array.isArray(payload.messages)) {
     payload.messages = payload.messages.map((message) =>
       message &&
@@ -13,13 +12,10 @@ function patchCoherePayload(payload: Record<string, unknown>): void {
     );
   }
 
-  // Cohere lets tool-capable models choose a tool when tool_choice is omitted.
   delete payload.tool_choice;
 }
 
-export function createCohereCompletionsWrapper(
-  baseStreamFn: ProviderWrapStreamFnContext["streamFn"],
-): ProviderWrapStreamFnContext["streamFn"] {
+export function createCohereCompletionsWrapper(baseStreamFn: StreamFn | undefined): StreamFn {
   return createPayloadPatchStreamWrapper(baseStreamFn, ({ payload }) =>
     patchCoherePayload(payload),
   );
diff --git a/scripts/plugin-sdk-surface-report.mjs b/scripts/plugin-sdk-surface-report.mjs
@@ -162,7 +162,7 @@ try {
   budgets = {
     publicEntrypoints: readBudgetEnv("OPENCLAW_PLUGIN_SDK_MAX_PUBLIC_ENTRYPOINTS", 319),
     publicExports: readBudgetEnv("OPENCLAW_PLUGIN_SDK_MAX_PUBLIC_EXPORTS", 10291),
-    publicFunctionExports: readBudgetEnv("OPENCLAW_PLUGIN_SDK_MAX_PUBLIC_FUNCTION_EXPORTS", 5166),
+    publicFunctionExports: readBudgetEnv("OPENCLAW_PLUGIN_SDK_MAX_PUBLIC_FUNCTION_EXPORTS", 5168),
     publicDeprecatedExports: readBudgetEnv(
       "OPENCLAW_PLUGIN_SDK_MAX_PUBLIC_DEPRECATED_EXPORTS",
       3244,
