@capekai/core 1.0.10 → 1.0.11

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@capekai/core",
-  "version": "1.0.10",
+  "version": "1.0.11",
   "description": "Bun-native composable agent runtime and framework for Capek.",
   "license": "MIT",
   "repository": {

package/src/adapters/ai-sdk.ts

@@ -1,15 +1,16 @@
 import { createOpenAI } from '@ai-sdk/openai';
 import {
+  wrapLanguageModel,
   dynamicTool,
   jsonSchema,
   streamText,
   type JSONSchema7,
-  type LanguageModel,
   type Tool,
 } from 'ai';
 import { getModelWithMetadata } from '../core/model-utils';
 import { openAiModelOmitsTemperature } from '../core/provider-utils';
 import type { ModelFactoryResult } from '../providers/types';
+import { codexNetworkRetryMiddleware } from './codex-network-retry';
 
 export interface TextModelRequest {
   modelId?: string;
@@ -55,7 +56,10 @@ export function createOpenAiResponsesModel(request: OpenAiResponsesModelRequest)
     fetch: request.fetch,
   });
   return {
-    model: openai.responses(request.modelId) as unknown as LanguageModel,
+    model: wrapLanguageModel({
+      model: openai.responses(request.modelId),
+      middleware: codexNetworkRetryMiddleware,
+    }),
     useProviderInstructions: true,
     omitMaxOutputTokens: true,
     omitTemperature: openAiModelOmitsTemperature(request.modelId),

package/src/adapters/codex-network-retry.ts

@@ -0,0 +1,25 @@
+import { APICallError, type LanguageModelMiddleware } from 'ai';
+import { ApiErrorType, classifyApiError } from '../utils/errors';
+
+export const codexNetworkRetryMiddleware: LanguageModelMiddleware = {
+  specificationVersion: 'v3',
+  wrapStream: async ({ doStream, params }) => {
+    try {
+      return await doStream();
+    } catch (error: unknown) {
+      if (params.abortSignal?.aborted || APICallError.isInstance(error)
+        || (error instanceof Error && error.name === 'AbortError')) throw error;
+      const classified = classifyApiError(error);
+      if (classified.type !== ApiErrorType.Network) throw error;
+      // OpenAI's early stream probe can throw a raw socket error. Normalize
+      // only before stream handoff so SDK retries this request, not prior tools.
+      throw new APICallError({
+        message: classified.message,
+        url: 'https://chatgpt.com/backend-api/codex/responses',
+        requestBodyValues: undefined,
+        cause: error,
+        isRetryable: true,
+      });
+    }
+  },
+};