Remove live check and migrate to abort signal. Fix cli to pass abort signal

Author: jahooma

cli/src/hooks/use-send-message.ts

@@ -375,6 +375,7 @@ export const useSendMessage = ({
           previousRunState: previousRunStateRef.current,
           agentDefinitions,
           eventHandlerState,
+          signal: abortController.signal,
         })
 
         const runState = await client.run(runConfig)

cli/src/utils/create-run-config.ts

@@ -15,6 +15,7 @@ export type CreateRunConfigParams = {
   previousRunState: RunState | null
   agentDefinitions: AgentDefinition[]
   eventHandlerState: EventHandlerState
+  signal: AbortSignal
 }
 
 export const createRunConfig = (params: CreateRunConfigParams) => {
@@ -38,5 +39,6 @@ export const createRunConfig = (params: CreateRunConfigParams) => {
     maxAgentSteps: 100,
     handleStreamChunk: createStreamChunkHandler(eventHandlerState),
     handleEvent: createEventHandler(eventHandlerState),
+    signal: params.signal,
   }
 }

common/src/testing/fixtures/agent-runtime.ts

@@ -104,8 +104,6 @@ export const TEST_AGENT_RUNTIME_IMPL = Object.freeze<
 
   // Mutable State
   databaseAgentCache: new Map<string, AgentTemplate | null>(),
-  liveUserInputRecord: {},
-  sessionConnections: {},
 
   // Analytics
   trackEvent: () => {},

common/src/types/contracts/agent-runtime.ts

@@ -18,7 +18,6 @@ import type {
   GetUserInfoFromApiKeyFn,
   StartAgentRunFn,
 } from './database'
-import type { SessionRecord, UserInputRecord } from './live-user-input'
 import type {
   PromptAiSdkFn,
   PromptAiSdkStreamFn,
@@ -49,8 +48,6 @@ export type AgentRuntimeDeps = {
 
   // Mutable State
   databaseAgentCache: DatabaseAgentCache
-  liveUserInputRecord: UserInputRecord
-  sessionConnections: SessionRecord
 
   // Analytics
   trackEvent: TrackEventFn

common/src/types/contracts/live-user-input.ts

@@ -1,6 +1,5 @@
 import type { TrackEventFn } from './analytics'
 import type { SendActionFn } from './client'
-import type { CheckLiveUserInputFn } from './live-user-input'
 import type { OpenRouterProviderRoutingOptions } from '../agent-template'
 import type { ParamsExcluding } from '../function-params'
 import type { Logger } from './logger'
@@ -50,8 +49,8 @@ export type PromptAiSdkStreamFn = (
     sendAction: SendActionFn
     logger: Logger
     trackEvent: TrackEventFn
-  } & ParamsExcluding<typeof streamText, 'model' | 'messages'> &
-    ParamsExcluding<CheckLiveUserInputFn, 'clientSessionId'>,
+    signal: AbortSignal
+  } & ParamsExcluding<typeof streamText, 'model' | 'messages'>,
 ) => AsyncGenerator<StreamChunk, string | null>
 
 export type PromptAiSdkFn = (
@@ -74,8 +73,8 @@ export type PromptAiSdkFn = (
     logger: Logger
     trackEvent: TrackEventFn
     n?: number
-  } & ParamsExcluding<typeof generateText, 'model' | 'messages'> &
-    ParamsExcluding<CheckLiveUserInputFn, 'clientSessionId'>,
+    signal: AbortSignal
+  } & ParamsExcluding<typeof generateText, 'model' | 'messages'>,
 ) => Promise<string>
 
 export type PromptAiSdkStructuredInput<T> = {
@@ -100,7 +99,8 @@ export type PromptAiSdkStructuredInput<T> = {
   sendAction: SendActionFn
   logger: Logger
   trackEvent: TrackEventFn
-} & ParamsExcluding<CheckLiveUserInputFn, 'clientSessionId'>
+  signal: AbortSignal
+}
 export type PromptAiSdkStructuredOutput<T> = Promise<T>
 export type PromptAiSdkStructuredFn = <T>(
   params: PromptAiSdkStructuredInput<T>,

common/src/types/contracts/llm.ts

@@ -4,7 +4,6 @@ import { execFileSync } from 'child_process'
 import fs from 'fs'
 import path from 'path'
 
-import { disableLiveUserInputCheck } from '@codebuff/agent-runtime/live-user-inputs'
 import { models } from '@codebuff/common/old-constants'
 import { userMessage } from '@codebuff/common/util/messages'
 import { promptAiSdkStructured } from '@codebuff/sdk'
@@ -399,7 +398,6 @@ async function screenCommitsWithGpt5(
     const prompt = `${COMMIT_SCREENING_PROMPT}\n\nCommit to evaluate:\n\n${commitInfo}`
 
     try {
-      disableLiveUserInputCheck()
       const response = await promptAiSdkStructured({
         messages: [userMessage(prompt)],
         schema: CommitSelectionSchema,
@@ -409,12 +407,11 @@ async function screenCommitsWithGpt5(
         userInputId,
         userId: undefined,
         sendAction: () => {},
-        liveUserInputRecord: {},
-        sessionConnections: {},
         logger: console,
         trackEvent: () => {},
         apiKey: 'unused-api-key',
         runId: 'unused-run-id',
+        signal: new AbortController().signal,
       })
 
       // Handle empty or invalid response

evals/buffbench/pick-commits.ts

@@ -64,8 +64,6 @@ export const EVALS_AGENT_RUNTIME_IMPL = Object.freeze<AgentRuntimeDeps>({
   },
 
   // Mutable State
-  liveUserInputRecord: {},
-  sessionConnections: {},
   databaseAgentCache: new Map<string, AgentTemplate | null>(),
 
   // Analytics