@lota-sdk/core 0.4.21 → 0.4.22

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lota-sdk/core",
-  "version": "0.4.21",
+  "version": "0.4.22",
   "files": [
     "src",
     "infrastructure/schema"
@@ -31,7 +31,7 @@
     "@ai-sdk/openai": "^3.0.54",
     "@chat-adapter/slack": "^4.26.0",
     "@chat-adapter/state-ioredis": "^4.26.0",
-    "@lota-sdk/shared": "0.4.21",
+    "@lota-sdk/shared": "0.4.22",
     "@mendable/firecrawl-js": "^4.20.0",
     "@surrealdb/node": "^3.0.3",
     "ai": "^6.0.170",

package/src/ai-gateway/ai-gateway.ts

@@ -892,6 +892,81 @@ function addAiGatewayReasoningRawChunks(
   return { ...params, includeRawChunks: true }
 }
 
+const JSON_SCHEMA_PROPERTY_MAP_KEYS = new Set([
+  '$defs',
+  'definitions',
+  'dependentSchemas',
+  'patternProperties',
+  'properties',
+])
+
+function stripJsonSchemaFormatKeywords(value: unknown): unknown {
+  if (Array.isArray(value)) {
+    return value.map((item) => stripJsonSchemaFormatKeywords(item))
+  }
+
+  if (!isRecord(value)) {
+    return value
+  }
+
+  let changed = false
+  const nextValue: Record<string, unknown> = {}
+
+  for (const [key, child] of Object.entries(value)) {
+    if (key === 'format') {
+      changed = true
+      continue
+    }
+
+    if (JSON_SCHEMA_PROPERTY_MAP_KEYS.has(key) && isRecord(child)) {
+      let mapChanged = false
+      const nextMap: Record<string, unknown> = {}
+      for (const [propertyName, propertySchema] of Object.entries(child)) {
+        const nextPropertySchema = stripJsonSchemaFormatKeywords(propertySchema)
+        mapChanged ||= nextPropertySchema !== propertySchema
+        nextMap[propertyName] = nextPropertySchema
+      }
+      changed ||= mapChanged
+      nextValue[key] = mapChanged ? nextMap : child
+      continue
+    }
+
+    const nextChild = stripJsonSchemaFormatKeywords(child)
+    changed ||= nextChild !== child
+    nextValue[key] = nextChild
+  }
+
+  return changed ? nextValue : value
+}
+
+export function normalizeAiGatewayJsonSchemas(params: AiGatewayCallOptions): AiGatewayCallOptions {
+  let nextParams = params
+
+  if (params.responseFormat?.type === 'json' && params.responseFormat.schema) {
+    const nextSchema = stripJsonSchemaFormatKeywords(params.responseFormat.schema)
+    if (nextSchema !== params.responseFormat.schema) {
+      nextParams = { ...nextParams, responseFormat: { ...params.responseFormat, schema: nextSchema } }
+    }
+  }
+
+  const sourceTools = nextParams.tools
+  if (!Array.isArray(sourceTools)) {
+    return nextParams
+  }
+
+  const nextTools = sourceTools.map((tool) => {
+    if (!isRecord(tool) || tool.type !== 'function') {
+      return tool
+    }
+
+    const inputSchema = (tool as { inputSchema: unknown }).inputSchema
+    const nextInputSchema = stripJsonSchemaFormatKeywords(inputSchema)
+    return { ...tool, inputSchema: nextInputSchema }
+  })
+
+  return { ...nextParams, tools: nextTools as AiGatewayCallOptions['tools'] }
+}
+
 function resolveProviderModel(
   provider: ReturnType<typeof createOpenAI>,
   modelId: string,
@@ -942,7 +1017,9 @@ function createAiGatewayLanguageModelMiddleware(
       Promise.resolve(
         withDefaultAiGatewayCacheHeaders(
           addAiGatewayReasoningRawChunks(
-            providerId === OPENAI_CHAT_PROVIDER_ID ? normalizeAiGatewayChatProviderOptions(params, modelId) : params,
+            normalizeAiGatewayJsonSchemas(
+              providerId === OPENAI_CHAT_PROVIDER_ID ? normalizeAiGatewayChatProviderOptions(params, modelId) : params,
+            ),
             type,
           ),
         ),

package/src/ai-gateway/index.ts

@@ -14,6 +14,7 @@ export {
   injectAiGatewayChatReasoningContent,
   injectAiGatewayChatReasoningStream,
   normalizeAiGatewayChatProviderOptions,
+  normalizeAiGatewayJsonSchemas,
   normalizeAiGatewayUrl,
 } from './ai-gateway'
 export type { AiGatewayDeps, AiGatewayModels, RuntimeBridge } from './ai-gateway'