refactor: trim stream helper internal exports

steipete · steipete · commit 8d288e2dfda1 · 2026-05-01T11:49:37.000+01:00
diff --git a/extensions/openai/native-web-search.ts b/extensions/openai/native-web-search.ts
@@ -7,7 +7,7 @@ import { isOpenAIApiBaseUrl } from "./base-url.js";
 
 const OPENAI_WEB_SEARCH_TOOL = { type: "web_search" } as const;
 
-export type OpenAINativeWebSearchPatchResult =
+type OpenAINativeWebSearchPatchResult =
   | "payload_not_object"
   | "native_tool_already_present"
   | "injected";
@@ -38,7 +38,7 @@ function shouldUseOpenAINativeWebSearchProvider(config: OpenClawConfig | undefin
   return normalized === "" || normalized === "auto" || normalized === "openai";
 }
 
-export function shouldEnableOpenAINativeWebSearch(params: {
+function shouldEnableOpenAINativeWebSearch(params: {
   config?: OpenClawConfig;
   model: { api?: unknown; provider?: unknown; baseUrl?: unknown };
 }): boolean {
@@ -65,9 +65,7 @@ function raiseMinimalReasoningForOpenAINativeWebSearch(payload: Record<string, u
   reasoning.effort = "low";
 }
 
-export function patchOpenAINativeWebSearchPayload(
-  payload: unknown,
-): OpenAINativeWebSearchPatchResult {
+function patchOpenAINativeWebSearchPayload(payload: unknown): OpenAINativeWebSearchPatchResult {
   if (!isRecord(payload)) {
     return "payload_not_object";
   }
diff --git a/extensions/xai/stream.ts b/extensions/xai/stream.ts
@@ -201,8 +201,7 @@ export function createXaiFastModeWrapper(
   };
 }
 
-export const createXaiToolCallArgumentDecodingWrapper =
-  createHtmlEntityToolCallArgumentDecodingWrapper;
+const createXaiToolCallArgumentDecodingWrapper = createHtmlEntityToolCallArgumentDecodingWrapper;
 
 export function wrapXaiProviderStream(ctx: ProviderWrapStreamFnContext): StreamFn | undefined {
   const extraParams = ctx.extraParams;
diff --git a/extensions/xai/test-helpers.ts b/extensions/xai/test-helpers.ts
@@ -2,16 +2,16 @@ import type { StreamFn } from "@mariozechner/pi-agent-core";
 import type { Context, Model } from "@mariozechner/pi-ai";
 import { expect } from "vitest";
 
-export type XaiToolPayloadFunction = {
+type XaiToolPayloadFunction = {
   function?: Record<string, unknown>;
 };
 
-export type XaiTestPayload = Record<string, unknown> & {
+type XaiTestPayload = Record<string, unknown> & {
   tools?: Array<{ type?: string; function?: Record<string, unknown> }>;
   input?: unknown[];
 };
 
-export function createXaiToolStreamPayload(): XaiTestPayload {
+function createXaiToolStreamPayload(): XaiTestPayload {
   return {
     reasoning: { effort: "high" },
     tools: [

Original file line number	Diff line number	Diff line change
`@@ -201,8 +201,7 @@ export function createXaiFastModeWrapper(`
`201`	`201`	`};`
`202`	`202`	`}`
`203`	`203`
`204`		`-export const createXaiToolCallArgumentDecodingWrapper =`
`205`		`- createHtmlEntityToolCallArgumentDecodingWrapper;`
	`204`	`+const createXaiToolCallArgumentDecodingWrapper = createHtmlEntityToolCallArgumentDecodingWrapper;`
`206`	`205`
`207`	`206`	`export function wrapXaiProviderStream(ctx: ProviderWrapStreamFnContext): StreamFn \| undefined {`
`208`	`207`	`const extraParams = ctx.extraParams;`