@lota-sdk/core 0.4.22 → 0.4.24

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lota-sdk/core",
-  "version": "0.4.22",
+  "version": "0.4.24",
   "files": [
     "src",
     "infrastructure/schema"
@@ -29,9 +29,10 @@
   "dependencies": {
     "@ai-sdk/devtools": "^0.0.16",
     "@ai-sdk/openai": "^3.0.54",
+    "@ai-sdk/provider": "^3.0.9",
     "@chat-adapter/slack": "^4.26.0",
     "@chat-adapter/state-ioredis": "^4.26.0",
-    "@lota-sdk/shared": "0.4.22",
+    "@lota-sdk/shared": "0.4.24",
     "@mendable/firecrawl-js": "^4.20.0",
     "@surrealdb/node": "^3.0.3",
     "ai": "^6.0.170",

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

@@ -1,5 +1,6 @@
 import { devToolsMiddleware } from '@ai-sdk/devtools'
 import { createOpenAI } from '@ai-sdk/openai'
+import type { JSONSchema7 } from '@ai-sdk/provider'
 import { wrapEmbeddingModel, wrapLanguageModel } from 'ai'
 import type { LanguageModelMiddleware } from 'ai'
 import { Cause, Clock, Context, Duration, Effect, Fiber, Layer, Semaphore } from 'effect'
@@ -16,6 +17,7 @@ type WrapStreamOptions = Parameters<NonNullable<LanguageModelMiddleware['wrapStr
 type AiGatewayLanguageModel = Parameters<typeof wrapLanguageModel>[0]['model']
 type AiGatewayEmbeddingModel = Parameters<typeof wrapEmbeddingModel>[0]['model']
 type AiGatewayCallOptions = WrapStreamOptions['params']
+type AiGatewayFunctionTool = Extract<NonNullable<AiGatewayCallOptions['tools']>[number], { type: 'function' }>
 type AiGatewayGenerateResult = Awaited<ReturnType<WrapStreamOptions['doGenerate']>>
 type AiGatewayStreamResult = Awaited<ReturnType<WrapStreamOptions['doStream']>>
 type AiGatewayGeneratedContent = AiGatewayGenerateResult['content'][number]
@@ -72,6 +74,17 @@ function isAiGenerationError(error: unknown): error is AiGenerationError {
   return isRecord(error) && error._tag === ERROR_TAGS.AiGenerationError
 }
 
+export function isAiGenerationContentFilterError(error: unknown): error is AiGenerationError {
+  if (!isAiGenerationError(error)) return false
+
+  const text = [error.message, error.providerData, error.responseBody]
+    .filter((part): part is string => typeof part === 'string' && part.length > 0)
+    .join('\n')
+    .toLowerCase()
+
+  return text.includes('content_filter') || text.includes('content management policy')
+}
+
 function getNumericField(value: Record<string, unknown>, key: string): number | null {
   const field = value[key]
   if (typeof field === 'number' && Number.isFinite(field)) return field
@@ -900,52 +913,55 @@ const JSON_SCHEMA_PROPERTY_MAP_KEYS = new Set([
   'properties',
 ])
 
-function stripJsonSchemaFormatKeywords(value: unknown): unknown {
+function removeJsonSchemaFormatKeywords(value: unknown): void {
   if (Array.isArray(value)) {
-    return value.map((item) => stripJsonSchemaFormatKeywords(item))
+    for (const item of value) {
+      removeJsonSchemaFormatKeywords(item)
+    }
+    return
   }
 
   if (!isRecord(value)) {
-    return value
+    return
   }
 
-  let changed = false
-  const nextValue: Record<string, unknown> = {}
+  delete value.format
 
   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
+      for (const propertySchema of Object.values(child)) {
+        removeJsonSchemaFormatKeywords(propertySchema)
       }
-      changed ||= mapChanged
-      nextValue[key] = mapChanged ? nextMap : child
       continue
     }
 
-    const nextChild = stripJsonSchemaFormatKeywords(child)
-    changed ||= nextChild !== child
-    nextValue[key] = nextChild
+    removeJsonSchemaFormatKeywords(child)
   }
+}
+
+function cloneAiGatewayJsonSchema(schema: JSONSchema7): JSONSchema7 {
+  // eslint-disable-next-line typescript-eslint/no-unsafe-assignment -- structuredClone preserves the imported JSONSchema7 shape.
+  const nextSchema: JSONSchema7 = structuredClone(schema)
+  removeJsonSchemaFormatKeywords(nextSchema)
+  return nextSchema
+}
 
-  return changed ? nextValue : value
+function isAiGatewayFunctionTool(
+  tool: NonNullable<AiGatewayCallOptions['tools']>[number],
+): tool is AiGatewayFunctionTool {
+  return isRecord(tool) && tool.type === 'function'
 }
 
 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 } }
+    // eslint-disable-next-line typescript-eslint/no-unsafe-assignment -- cloneAiGatewayJsonSchema returns the same JSONSchema7 type.
+    const responseSchema: JSONSchema7 = cloneAiGatewayJsonSchema(params.responseFormat.schema)
+    nextParams = {
+      ...nextParams,
+      // eslint-disable-next-line typescript-eslint/no-unsafe-assignment -- responseSchema is JSONSchema7 for responseFormat.schema.
+      responseFormat: { ...params.responseFormat, schema: responseSchema },
     }
   }
 
@@ -955,13 +971,14 @@ export function normalizeAiGatewayJsonSchemas(params: AiGatewayCallOptions): AiG
   }
 
   const nextTools = sourceTools.map((tool) => {
-    if (!isRecord(tool) || tool.type !== 'function') {
+    if (!isAiGatewayFunctionTool(tool)) {
       return tool
     }
 
-    const inputSchema = (tool as { inputSchema: unknown }).inputSchema
-    const nextInputSchema = stripJsonSchemaFormatKeywords(inputSchema)
-    return { ...tool, inputSchema: nextInputSchema }
+    // eslint-disable-next-line typescript-eslint/no-unsafe-assignment -- function tool inputSchema is JSONSchema7.
+    const inputSchema: JSONSchema7 = cloneAiGatewayJsonSchema(tool.inputSchema)
+    // eslint-disable-next-line typescript-eslint/no-unsafe-assignment -- inputSchema is JSONSchema7 for function tool inputSchema.
+    return { ...tool, inputSchema }
   })
 
   return { ...nextParams, tools: nextTools as AiGatewayCallOptions['tools'] }

package/src/ai-gateway/index.ts

@@ -13,6 +13,7 @@ export {
   extractAiGatewayChatReasoningText,
   injectAiGatewayChatReasoningContent,
   injectAiGatewayChatReasoningStream,
+  isAiGenerationContentFilterError,
   normalizeAiGatewayChatProviderOptions,
   normalizeAiGatewayJsonSchemas,
   normalizeAiGatewayUrl,

package/src/services/recent-activity-title.service.ts

@@ -1,9 +1,10 @@
 import { Context, Effect, Layer } from 'effect'
 
-import { AiGatewayModelsTag } from '../ai-gateway/ai-gateway'
+import { AiGatewayModelsTag, isAiGenerationContentFilterError } from '../ai-gateway/ai-gateway'
 import type { AiGatewayModels } from '../ai-gateway/ai-gateway'
 import type { ResolvedAgentConfig } from '../config/agent-defaults'
-import { ServiceError } from '../effect/errors'
+import { chatLogger } from '../config/logger'
+import { ERROR_TAGS, ServiceError } from '../effect/errors'
 import { AgentConfigServiceTag } from '../effect/services'
 import type { HelperModelRuntime } from '../runtime/helper-model'
 import { HelperModelTag } from '../runtime/helper-model'
@@ -62,20 +63,34 @@ export function makeRecentActivityTitleService(
         return
       }
 
-      const refinedTitle = normalizeTitle(
-        yield* Effect.tryPromise({
-          try: () =>
-            helperModelRuntime.generateHelperText({
-              tag: 'recent-activity-title-refinement',
-              createAgent: refinerAgentFactory,
-              defaultSystemPrompt: RECENT_ACTIVITY_TITLE_REFINER_PROMPT,
-              timeoutMs: RECENT_ACTIVITY_TITLE_TIMEOUT_MS,
-              messages: [{ role: 'user', content: promptInput }],
-            }),
-          catch: (cause) =>
-            new ServiceError({ message: 'Failed to generate recent activity title refinement.', cause }),
-        }),
+      const maybeRefinedTitle = yield* Effect.tryPromise({
+        try: () =>
+          helperModelRuntime.generateHelperText({
+            tag: 'recent-activity-title-refinement',
+            createAgent: refinerAgentFactory,
+            defaultSystemPrompt: RECENT_ACTIVITY_TITLE_REFINER_PROMPT,
+            timeoutMs: RECENT_ACTIVITY_TITLE_TIMEOUT_MS,
+            messages: [{ role: 'user', content: promptInput }],
+          }),
+        catch: (cause) =>
+          isAiGenerationContentFilterError(cause)
+            ? cause
+            : new ServiceError({ message: 'Failed to generate recent activity title refinement.', cause }),
+      }).pipe(
+        Effect.catchTag(ERROR_TAGS.AiGenerationError, (error) =>
+          isAiGenerationContentFilterError(error)
+            ? Effect.sync(() => {
+                chatLogger.warn`Skipping recent activity title refinement after provider content filter (activityId=${activityId})`
+                return null
+              })
+            : Effect.fail(error),
+        ),
       )
+      if (maybeRefinedTitle === null) {
+        return
+      }
+
+      const refinedTitle = normalizeTitle(maybeRefinedTitle)
       if (
         !recentActivityService.isAgentTitleUseful({
           currentTitle: candidate.title,