From 8d288e2dfda1c55ff4ae763010600830ed92b7b1 Mon Sep 17 00:00:00 2001
From: Peter Steinberger <steipete@gmail.com>
Date: Fri, 1 May 2026 11:49:31 +0100
Subject: [PATCH] refactor: trim stream helper internal exports

---
 extensions/openai/native-web-search.ts | 8 +++-----
 extensions/xai/stream.ts               | 3 +--
 extensions/xai/test-helpers.ts         | 6 +++---
 3 files changed, 7 insertions(+), 10 deletions(-)

diff --git a/extensions/openai/native-web-search.ts b/extensions/openai/native-web-search.ts
index c1c7abc74d2..f26f8c87022 100644
--- 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
index e3878bce5d8..6ed093fdf17 100644
--- 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
index 637c4b7dc22..366019f4adf 100644
--- 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: [