@lota-sdk/core 0.4.17 → 0.4.19

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lota-sdk/core",
-  "version": "0.4.17",
+  "version": "0.4.19",
   "type": "module",
   "main": "./src/index.ts",
   "types": "./src/index.ts",
@@ -31,7 +31,7 @@
     "@ai-sdk/openai": "^3.0.53",
     "@chat-adapter/slack": "^4.26.0",
     "@chat-adapter/state-ioredis": "^4.26.0",
-    "@lota-sdk/shared": "0.4.17",
+    "@lota-sdk/shared": "0.4.19",
     "@mendable/firecrawl-js": "^4.18.3",
     "@surrealdb/node": "^3.0.3",
     "ai": "^6.0.168",

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

@@ -21,7 +21,6 @@ type AiGatewayGenerateResult = Awaited<ReturnType<WrapStreamOptions['doGenerate'
 type AiGatewayStreamResult = Awaited<ReturnType<WrapStreamOptions['doStream']>>
 type AiGatewayGeneratedContent = AiGatewayGenerateResult['content'][number]
 type AiGatewayStreamPart = AiGatewayStreamResult['stream'] extends ReadableStream<infer T> ? T : never
-type AiGatewayProviderOptions = NonNullable<AiGatewayCallOptions['providerOptions']>
 type AiGatewayAttemptResult<A> = { source: string; result: A }
 // eslint-disable-next-line @typescript-eslint/no-redundant-type-constituents
 type AiGatewayRunFork = <A, E>(effect: Effect.Effect<A, E, never>) => Fiber.Fiber<A, E | unknown>
@@ -867,10 +866,6 @@ function executeStreamAttemptPlan(
   )
 }
 
-function isOpenRouterOpenAIReasoningModel(modelId: string): boolean {
-  return modelId.trim().toLowerCase().startsWith('openrouter/openai/gpt-5')
-}
-
 function shouldCloseInjectedReasoning(chunk: AiGatewayStreamPart): boolean {
   switch (chunk.type) {
     case 'stream-start':
@@ -983,6 +978,10 @@ export function setDefaultAiGatewayDeps(deps: AiGatewayDeps): void {
   defaultAiGatewayDeps = deps
 }
 
+export function getDefaultAiGatewayRunPromise(): AiGatewayDeps['runPromise'] {
+  return resolveAiGatewayDeps(undefined).runPromise
+}
+
 function resolveAiGatewayDeps(deps: AiGatewayDeps | undefined): AiGatewayDeps {
   if (deps) return deps
   if (defaultAiGatewayDeps) return defaultAiGatewayDeps
@@ -1048,27 +1047,9 @@ function createAiGatewayLanguageModelMiddleware(
 
 export function normalizeAiGatewayChatProviderOptions(
   params: AiGatewayCallOptions,
-  modelId?: string,
+  _modelId?: string,
 ): AiGatewayCallOptions {
-  const providerOptions = isRecord(params.providerOptions)
-    ? ({ ...params.providerOptions } as AiGatewayProviderOptions)
-    : ({} as AiGatewayProviderOptions)
-  const openaiOptions = isRecord(providerOptions.openai)
-    ? { ...providerOptions.openai }
-    : ({} as Record<string, unknown>)
-
-  if (modelId && isOpenRouterOpenAIReasoningModel(modelId) && openaiOptions.forceReasoning === undefined) {
-    openaiOptions.forceReasoning = true
-  }
-
-  if (providerOptions.openai === openaiOptions || Object.keys(openaiOptions).length === 0) {
-    return params
-  }
-
-  return {
-    ...params,
-    providerOptions: { ...providerOptions, openai: openaiOptions as AiGatewayProviderOptions['openai'] },
-  }
+  return params
 }
 
 function withAiGatewayDevTools<TModel extends AiGatewayLanguageModel>(model: TModel): TModel {

package/src/system-agents/delegated-agent-factory.ts

@@ -3,6 +3,7 @@ import type { ModelMessage, LanguageModel, ToolLoopAgentSettings, ToolSet } from
 import { Schema, Effect } from 'effect'
 import { z } from 'zod'
 
+import { getDefaultAiGatewayRunPromise } from '../ai-gateway/ai-gateway'
 import type { ToolDefinition } from '../ai/definitions'
 import { aiLogger } from '../config/logger'
 import { ERROR_TAGS } from '../effect/errors'
@@ -26,7 +27,7 @@ interface DelegatedAgentDefinition {
   maxSteps?: number
   maxOutputTokens?: number
   temperature?: number
-  runPromise: <A, E>(effect: Effect.Effect<A, E, never>, options?: { signal?: AbortSignal }) => Promise<A>
+  runPromise?: <A, E>(effect: Effect.Effect<A, E, never>, options?: { signal?: AbortSignal }) => Promise<A>
 }
 
 type DelegatedAgentRunPromise = <A, E>(
@@ -42,7 +43,7 @@ interface DelegatedAgentDefinitionWithContext<TContext> extends Omit<
   createTools: (context: TContext) => ToolSet
   /** Pull runPromise off the caller's context so each tool invocation carries
    * its own runtime binding rather than closing over a boot-time one. */
-  getRunPromise: (context: TContext) => DelegatedAgentRunPromise
+  getRunPromise?: (context: TContext) => DelegatedAgentRunPromise
 }
 
 class DelegatedAgentError extends Schema.TaggedErrorClass<DelegatedAgentError>()(ERROR_TAGS.DelegatedAgentError, {
@@ -251,7 +252,7 @@ export function createDelegatedAgentTool(definition: DelegatedAgentDefinition):
         description: definition.description,
         inputSchema: z.object({ task: z.string().min(1) }),
         execute: ({ task }: { task: string }, { abortSignal }) =>
-          definition.runPromise(
+          (definition.runPromise ?? getDefaultAiGatewayRunPromise())(
             Effect.gen(function* () {
               const agentTools = definition.tools
               const createAgent = () =>
@@ -309,7 +310,7 @@ export function createDelegatedAgentToolWithContext<TContext>(
         description: definition.description,
         inputSchema: z.object({ task: z.string().min(1) }),
         execute: ({ task }: { task: string }, { abortSignal }) =>
-          definition.getRunPromise(context)(
+          (definition.getRunPromise?.(context) ?? getDefaultAiGatewayRunPromise())(
             Effect.gen(function* () {
               const agentTools = definition.createTools(context)
               const createAgent = () =>

package/src/tools/execution-plan.tool.ts

@@ -15,6 +15,7 @@ import type { CreateProjectWithPlanResultData, ExecutionPlanArgs } from '@lota-s
 import { tool } from 'ai'
 import { Effect } from 'effect'
 
+import { getDefaultAiGatewayRunPromise } from '../ai-gateway/ai-gateway'
 import { serverLogger } from '../config/logger'
 import type { RecordIdRef } from '../db/record-id'
 import { recordIdToString } from '../db/record-id'
@@ -58,7 +59,7 @@ export function createExecutionPlanTool(params: {
   agentId: string
   executionPlanService: ExecutionPlanExecutionPlanService
   threadService: ExecutionPlanThreadService
-  runPromise: ExecutionPlanToolRunPromise
+  runPromise?: ExecutionPlanToolRunPromise
   onPlanChanged?: () => void
   validateInlinePlan?: (draft: ReturnType<typeof expandAgentPlanDraft>) => void
 }) {
@@ -67,7 +68,7 @@ export function createExecutionPlanTool(params: {
       'Manage execution plans. Actions: create (inline, 1-2 nodes), create-project (dedicated project thread, 3+ nodes), replace (swap active plan), resume (resume interrupted plan), update-node (submit result for a specific node).',
     inputSchema: ExecutionPlanArgsSchema,
     execute: (input) =>
-      params.runPromise(
+      (params.runPromise ?? getDefaultAiGatewayRunPromise())(
         Effect.gen(function* () {
           const parsed = yield* parseExecutionPlanArgsEffect(input)
 
@@ -265,14 +266,14 @@ function extractAgentPlanDraftEffect(
 export function createExecutionPlanQueryTool(params: {
   threadId: RecordIdRef
   executionPlanService: Pick<ExecutionPlanExecutionPlanService, 'listActivePlanSummaries' | 'getActivePlanToolResult'>
-  runPromise: ExecutionPlanToolRunPromise
+  runPromise?: ExecutionPlanToolRunPromise
 }) {
   return tool({
     description:
       'Query execution plans. Omit runId to list all active plans. Provide runId to load a specific plan run.',
     inputSchema: ExecutionPlanQueryArgsSchema,
     execute: (input) =>
-      params.runPromise(
+      (params.runPromise ?? getDefaultAiGatewayRunPromise())(
         Effect.gen(function* () {
           if (!input.runId) {
             return yield* params.executionPlanService.listActivePlanSummaries(params.threadId)
@@ -295,7 +296,7 @@ export function createSubmitExecutionNodeResultTool(params: {
   threadId: RecordIdRef
   agentId: string
   executionPlanService: Pick<ExecutionPlanExecutionPlanService, 'submitNodeResult'>
-  runPromise: ExecutionPlanToolRunPromise
+  runPromise?: ExecutionPlanToolRunPromise
   onPlanChanged?: () => void
 }) {
   return tool({
@@ -303,7 +304,7 @@ export function createSubmitExecutionNodeResultTool(params: {
       'Submit the result for the currently running execution node. The executor validates outputs, artifacts, and completion checks before advancing the run.',
     inputSchema: SubmitExecutionNodeResultArgsSchema,
     execute: (input) =>
-      params.runPromise(
+      (params.runPromise ?? getDefaultAiGatewayRunPromise())(
         Effect.gen(function* () {
           const result = yield* params.executionPlanService.submitNodeResult({
             threadId: params.threadId,

package/src/tools/fetch-webpage.tool.ts

@@ -3,6 +3,7 @@ import { tool } from 'ai'
 import { Effect, Schema } from 'effect'
 import { z } from 'zod'
 
+import { getDefaultAiGatewayRunPromise } from '../ai-gateway/ai-gateway'
 import type { ToolDefinition } from '../ai/definitions'
 import { withTimeout } from '../utils/async'
 import { nowIsoDateTimeString } from '../utils/date-time'
@@ -12,7 +13,7 @@ import { toRecord, WEB_TOOL_TIMEOUT_MS } from './web-tool-shared'
 
 export interface FetchWebpageToolContext {
   firecrawl: Firecrawl
-  runPromise: <A, E>(effect: Effect.Effect<A, E, never>, options?: { signal?: AbortSignal }) => Promise<A>
+  runPromise?: <A, E>(effect: Effect.Effect<A, E, never>, options?: { signal?: AbortSignal }) => Promise<A>
 }
 
 class FetchWebpageToolError extends Schema.TaggedErrorClass<FetchWebpageToolError>()(
@@ -103,8 +104,9 @@ function buildFetchCitations(url: string, document: unknown): WebCitation[] {
 
 export const fetchWebpageTool = {
   name: 'fetchWebpage',
-  create: ({ firecrawl, runPromise }: FetchWebpageToolContext) =>
-    tool({
+  create: ({ firecrawl, runPromise }: FetchWebpageToolContext) => {
+    const runEffect = runPromise ?? getDefaultAiGatewayRunPromise()
+    return tool({
       description: 'Retrieve and parse a single webpage.',
       inputSchema: z
         .object({
@@ -123,7 +125,7 @@ export const fetchWebpageTool = {
         }: { url: string; formats?: z.infer<typeof FormatSchema>[]; onlyMainContent?: boolean; maxAge?: number },
         { abortSignal }: { abortSignal?: AbortSignal } = {},
       ) =>
-        runPromise(
+        runEffect(
           Effect.gen(function* () {
             const result = yield* Effect.tryPromise({
               try: () =>
@@ -142,5 +144,6 @@ export const fetchWebpageTool = {
           }).pipe(Effect.withSpan('tool.fetchWebpage.execute')),
           abortSignal ? { signal: abortSignal } : undefined,
         ),
-    }),
+    })
+  },
 } as const satisfies ToolDefinition<FetchWebpageToolContext>

package/src/tools/memory-block.tool.ts

@@ -2,6 +2,7 @@ import { tool } from 'ai'
 import { Effect } from 'effect'
 import { z } from 'zod'
 
+import { getDefaultAiGatewayRunPromise } from '../ai-gateway/ai-gateway'
 import type { RecordIdRef } from '../db/record-id'
 import {
   hasMemoryBlockEntry,
@@ -25,10 +26,11 @@ export function createMemoryBlockTool({
   threadId: RecordIdRef
   agentLabel: string
   threadService: MemoryBlockThreadService
-  runPromise: <A, E>(effect: Effect.Effect<A, E, never>) => Promise<A>
+  runPromise?: <A, E>(effect: Effect.Effect<A, E, never>) => Promise<A>
   getCurrentBlock?: () => string
   onAppend?: (value: string) => void
 }) {
+  const runEffect = runPromise ?? getDefaultAiGatewayRunPromise()
   return tool({
     description:
       'Append a critical 2-3 sentence note to the conversation memory block. Include only essential information.',
@@ -50,7 +52,7 @@ export function createMemoryBlockTool({
 
       void safeEnqueue(
         () =>
-          runPromise(
+          runEffect(
             Effect.gen(function* () {
               const updated = yield* threadService.appendMemoryBlock(threadId, prepared.formatted)
               onAppend?.(updated)

package/src/tools/plan-approval.tool.ts

@@ -2,6 +2,7 @@ import { PlanApprovalArgsSchema, PlanApprovalToolResultDataSchema } from '@lota-
 import { tool } from 'ai'
 import { Effect } from 'effect'
 
+import { getDefaultAiGatewayRunPromise } from '../ai-gateway/ai-gateway'
 import type { RecordIdRef } from '../db/record-id'
 import type { makeExecutionPlanService } from '../services/execution-plan/execution-plan.service'
 import { toToolPlanSummary } from '../services/plan/plan-run-data'
@@ -21,15 +22,16 @@ export function createPlanApprovalTool(params: {
   threadId: RecordIdRef
   actorId: string
   executionPlanService: PlanApprovalExecutionPlanService
-  runPromise: <A, E>(effect: Effect.Effect<A, E, never>, options?: { signal?: AbortSignal }) => Promise<A>
+  runPromise?: <A, E>(effect: Effect.Effect<A, E, never>, options?: { signal?: AbortSignal }) => Promise<A>
 }) {
+  const runEffect = params.runPromise ?? getDefaultAiGatewayRunPromise()
   return tool({
     description:
       'Approve, reject, or request changes to a plan that is pending approval. Use approve to start execution, reject to abort it, or modify to request a revised plan.',
     inputSchema: PlanApprovalArgsSchema,
     outputSchema: PlanApprovalToolResultDataSchema,
     execute: (input, { abortSignal }: { abortSignal?: AbortSignal } = {}) =>
-      params.runPromise(
+      runEffect(
         Effect.gen(function* () {
           switch (input.action) {
             case 'approve': {

package/src/tools/read-file-parts.tool.ts

@@ -2,6 +2,7 @@ import { tool } from 'ai'
 import { Schema, Effect } from 'effect'
 import { z } from 'zod'
 
+import { getDefaultAiGatewayRunPromise } from '../ai-gateway/ai-gateway'
 import type { ToolDefinition } from '../ai/definitions'
 import { ERROR_TAGS } from '../effect/errors'
 import type { makeAttachmentService } from '../services/attachment.service'
@@ -16,7 +17,7 @@ export interface ReadFilePartsToolContext {
   userId: string
   uploads: ReadableUploadMetadata[]
   attachmentService: ReadFilePartsAttachmentService
-  runPromise: <A, E>(effect: Effect.Effect<A, E, never>, options?: { signal?: AbortSignal }) => Promise<A>
+  runPromise?: <A, E>(effect: Effect.Effect<A, E, never>, options?: { signal?: AbortSignal }) => Promise<A>
 }
 
 function toUploadSummary(upload: ReadableUploadMetadata) {
@@ -49,8 +50,9 @@ function resolveUploadTargetEffect(params: {
 
 export const readFilePartsTool = {
   name: 'readFileParts',
-  create: ({ orgId, userId, uploads, attachmentService, runPromise }: ReadFilePartsToolContext) =>
-    tool({
+  create: ({ orgId, userId, uploads, attachmentService, runPromise }: ReadFilePartsToolContext) => {
+    const runEffect = runPromise ?? getDefaultAiGatewayRunPromise()
+    return tool({
       description:
         'Read uploaded file content by part. Call with no args to list uploads metadata, then call with storageKey and part (25 pages per part).',
       inputSchema: z
@@ -60,7 +62,7 @@ export const readFilePartsTool = {
         { storageKey, part }: { storageKey?: string; part?: number },
         { abortSignal }: { abortSignal?: AbortSignal } = {},
       ) =>
-        runPromise(
+        runEffect(
           Effect.gen(function* () {
             const availableUploads = uploads.map(toUploadSummary)
             const selected = yield* resolveUploadTargetEffect({ uploads, storageKey })
@@ -93,5 +95,6 @@ export const readFilePartsTool = {
           }),
           abortSignal ? { signal: abortSignal } : undefined,
         ),
-    }),
+    })
+  },
 } as const satisfies ToolDefinition<ReadFilePartsToolContext>

package/src/tools/remember-memory.tool.ts

@@ -2,6 +2,7 @@ import { tool } from 'ai'
 import { Effect } from 'effect'
 import { z } from 'zod'
 
+import { getDefaultAiGatewayRunPromise } from '../ai-gateway/ai-gateway'
 import type { RecordIdRef } from '../db/record-id'
 import { recordIdToString } from '../db/record-id'
 import { TABLES } from '../db/tables'
@@ -29,8 +30,9 @@ export function createRememberMemoryTool({
   orgId: RecordIdRef
   agentName: string
   memoryService: RememberMemoryService
-  runPromise: <A, E>(effect: Effect.Effect<A, E, never>) => Promise<A>
+  runPromise?: <A, E>(effect: Effect.Effect<A, E, never>) => Promise<A>
 }) {
+  const runEffect = runPromise ?? getDefaultAiGatewayRunPromise()
   return tool({
     description:
       'Save a durable memory. Importance is scored automatically from memory significance. Use targetScope="agent" for role-specific memory, or targetScope="global" for organization-wide memory.',
@@ -44,7 +46,7 @@ export function createRememberMemoryTool({
       const orgIdString = recordIdToString(orgId, TABLES.ORGANIZATION)
       void safeEnqueue(
         () =>
-          runPromise(
+          runEffect(
             Effect.gen(function* () {
               const assessment = yield* memoryService.assessMemoryCandidate({ orgId: orgIdString, content: trimmed })
               if (!assessment) {

package/src/tools/research-topic.tool.ts

@@ -2,6 +2,7 @@ import type Firecrawl from '@mendable/firecrawl-js'
 import type { Effect } from 'effect'
 
 import type { AiGatewayModels } from '../ai-gateway/ai-gateway'
+import { aiGatewayChatModel, getDefaultAiGatewayRunPromise } from '../ai-gateway/ai-gateway'
 import { buildAiGatewayStrictSemanticCacheHeaders } from '../ai-gateway/cache-headers'
 import {
   OPENROUTER_FAST_REASONING_MODEL_ID,
@@ -14,15 +15,18 @@ import { searchWebTool } from './search-web.tool'
 
 export interface ResearchTopicToolContext {
   firecrawl: Firecrawl
-  aiGatewayModels: AiGatewayModels
-  runPromise: <A, E>(effect: Effect.Effect<A, E, never>, options?: { signal?: AbortSignal }) => Promise<A>
+  aiGatewayModels?: AiGatewayModels
+  runPromise?: <A, E>(effect: Effect.Effect<A, E, never>, options?: { signal?: AbortSignal }) => Promise<A>
 }
 
 export const researchTopicTool = createDelegatedAgentToolWithContext<ResearchTopicToolContext>({
   id: 'researchTopic',
   description:
     'Delegate a research task to a dedicated research agent that searches the web, fetches pages, and returns a synthesized markdown report.',
-  model: ({ aiGatewayModels }) => aiGatewayModels.chatModel(OPENROUTER_FAST_REASONING_MODEL_ID),
+  model: ({ aiGatewayModels }) =>
+    aiGatewayModels
+      ? aiGatewayModels.chatModel(OPENROUTER_FAST_REASONING_MODEL_ID)
+      : aiGatewayChatModel(OPENROUTER_FAST_REASONING_MODEL_ID),
   providerOptions: OPENROUTER_LOW_REASONING_PROVIDER_OPTIONS,
   headers: buildAiGatewayStrictSemanticCacheHeaders('researchTopic'),
   instructions: RESEARCHER_PROMPT,
@@ -31,5 +35,5 @@ export const researchTopicTool = createDelegatedAgentToolWithContext<ResearchTop
     fetchWebpage: fetchWebpageTool.create({ firecrawl, runPromise }),
   }),
   maxSteps: 6,
-  getRunPromise: (context) => context.runPromise,
+  getRunPromise: (context) => context.runPromise ?? getDefaultAiGatewayRunPromise(),
 })

package/src/tools/search-web.tool.ts

@@ -3,6 +3,7 @@ import { tool } from 'ai'
 import { Effect, Schema } from 'effect'
 import { z } from 'zod'
 
+import { getDefaultAiGatewayRunPromise } from '../ai-gateway/ai-gateway'
 import type { ToolDefinition } from '../ai/definitions'
 import { ERROR_TAGS } from '../effect/errors'
 import { withTimeout } from '../utils/async'
@@ -13,7 +14,7 @@ import { toRecord, WEB_TOOL_TIMEOUT_MS } from './web-tool-shared'
 
 export interface SearchWebToolContext {
   firecrawl: Firecrawl
-  runPromise: <A, E>(effect: Effect.Effect<A, E, never>, options?: { signal?: AbortSignal }) => Promise<A>
+  runPromise?: <A, E>(effect: Effect.Effect<A, E, never>, options?: { signal?: AbortSignal }) => Promise<A>
 }
 
 class SearchWebToolError extends Schema.TaggedErrorClass<SearchWebToolError>()(ERROR_TAGS.SearchWebToolError, {
@@ -141,8 +142,9 @@ function buildWebCitations(results: { web?: unknown[]; news?: unknown[]; images?
 
 export const searchWebTool = {
   name: 'searchWeb',
-  create: ({ firecrawl, runPromise }: SearchWebToolContext) =>
-    tool({
+  create: ({ firecrawl, runPromise }: SearchWebToolContext) => {
+    const runEffect = runPromise ?? getDefaultAiGatewayRunPromise()
+    return tool({
       description: 'Search the web for real-time information.',
       inputSchema: z
         .object({
@@ -163,7 +165,7 @@ export const searchWebTool = {
         }: { query: string; limit?: number; sources?: ('web' | 'news' | 'images')[]; location?: string; tbs?: string },
         { abortSignal }: { abortSignal?: AbortSignal } = {},
       ) =>
-        runPromise(
+        runEffect(
           Effect.gen(function* () {
             const results = yield* Effect.tryPromise({
               try: () =>
@@ -186,5 +188,6 @@ export const searchWebTool = {
           }).pipe(Effect.withSpan('tool.searchWeb.execute')),
           abortSignal ? { signal: abortSignal } : undefined,
         ),
-    }),
+    })
+  },
 } as const satisfies ToolDefinition<SearchWebToolContext>

package/src/tools/search.tool.ts

@@ -2,6 +2,7 @@ import { tool } from 'ai'
 import { Effect } from 'effect'
 import { z } from 'zod'
 
+import { getDefaultAiGatewayRunPromise } from '../ai-gateway/ai-gateway'
 import type { ResolvedAgentConfig } from '../config/agent-defaults'
 import { isAgentName } from '../config/agent-defaults'
 import type { RecordIdRef } from '../db/record-id'
@@ -16,15 +17,33 @@ type MemorySearchService = Pick<ReturnType<typeof createMemoryService>, 'searchA
 type ConversationSearchService = Pick<ReturnType<typeof makeThreadMessageService>, 'searchMessagesEffect'>
 
 type SearchToolRunPromise = <A, E>(effect: Effect.Effect<A, E, never>, options?: { signal?: AbortSignal }) => Promise<A>
+type MemorySearchOptions = { fastMode?: boolean; allowMultiScopeRerank?: boolean }
 
 export function createMemorySearchTool(
   agentConfig: ResolvedAgentConfig,
   orgIdString: string,
   memoryService: MemorySearchService,
-  runPromise: SearchToolRunPromise,
-  agentName?: string,
-  options?: { fastMode?: boolean; allowMultiScopeRerank?: boolean },
+  runPromiseOrAgentName?: SearchToolRunPromise | string,
+  agentNameOrOptions?: string | MemorySearchOptions,
+  optionsArg?: MemorySearchOptions,
 ) {
+  const runPromise =
+    typeof runPromiseOrAgentName === 'function' ? runPromiseOrAgentName : getDefaultAiGatewayRunPromise()
+  const agentName =
+    typeof runPromiseOrAgentName === 'string'
+      ? runPromiseOrAgentName
+      : typeof agentNameOrOptions === 'string'
+        ? agentNameOrOptions
+        : undefined
+  const options =
+    typeof runPromiseOrAgentName === 'function'
+      ? typeof agentNameOrOptions === 'object'
+        ? agentNameOrOptions
+        : optionsArg
+      : typeof agentNameOrOptions === 'object'
+        ? agentNameOrOptions
+        : optionsArg
+
   return tool({
     description: 'Search organization and agent memories relevant to a query.',
     inputSchema: MemorySearchInputSchema,
@@ -55,7 +74,7 @@ export function createMemorySearchTool(
 export function createConversationSearchTool(
   threadId: RecordIdRef,
   threadMessageService: ConversationSearchService,
-  runPromise: SearchToolRunPromise,
+  runPromise: SearchToolRunPromise = getDefaultAiGatewayRunPromise(),
 ) {
   return tool({
     description: 'Search prior chat messages by role and query text.',

package/src/tools/team-think.tool.ts

@@ -3,6 +3,7 @@ import { stepCountIs } from 'ai'
 import type { ToolSet } from 'ai'
 import { Schema, Effect } from 'effect'
 
+import { getDefaultAiGatewayRunPromise } from '../ai-gateway/ai-gateway'
 import { aiLogger } from '../config/logger'
 import type { RecordIdRef } from '../db/record-id'
 import { recordIdToString } from '../db/record-id'
@@ -72,8 +73,9 @@ export function createTeamThinkTool(params: {
   context?: unknown
   toolProviders?: ToolSet
   abortSignal: AbortSignal
-  runPromise: <A, E>(effect: Effect.Effect<A, E, never>, options?: { signal?: AbortSignal }) => Promise<A>
+  runPromise?: <A, E>(effect: Effect.Effect<A, E, never>, options?: { signal?: AbortSignal }) => Promise<A>
 }) {
+  const runEffect = params.runPromise ?? getDefaultAiGatewayRunPromise()
   return Effect.gen(function* () {
     const turnHooks = yield* TurnHooksServiceTag
     const runtimeAdapters = yield* RuntimeAdaptersServiceTag
@@ -84,7 +86,7 @@ export function createTeamThinkTool(params: {
     )
     const participantRunner: TeamConsultationParticipantRunner = {
       buildParticipantAgent(agentId, runParams) {
-        return params.runPromise(
+        return runEffect(
           Effect.gen(function* () {
             // Capture the fiber context of the managed runtime and replay it
             // for observer callbacks that cross the Effect → Promise boundary