fix agent-runtime tests

Author: charleslien

packages/agent-runtime/src/__tests__/main-prompt.test.ts

@@ -195,48 +195,6 @@ describe('mainPrompt', () => {
     },
   }
 
-  it('should handle direct terminal command', async () => {
-    const sessionState = getInitialSessionState(mockFileContext)
-    const action = {
-      type: 'prompt' as const,
-      prompt: 'ls -la',
-      sessionState,
-      fingerprintId: 'test',
-      costMode: 'max' as const,
-      promptId: 'test',
-      toolResults: [],
-    }
-
-    const { sessionState: newSessionState, output } = await mainPrompt({
-      ...mainPromptBaseParams,
-      action,
-    })
-
-    // Verify that requestToolCall was called with the terminal command
-    const requestToolCallSpy = mainPromptBaseParams.requestToolCall
-    expect(requestToolCallSpy).toHaveBeenCalledTimes(1)
-    expect(requestToolCallSpy).toHaveBeenCalledWith({
-      userInputId: expect.any(String), // userInputId
-      toolName: 'run_terminal_command',
-      input: expect.objectContaining({
-        command: 'ls -la',
-        mode: 'user',
-        process_type: 'SYNC',
-        timeout_seconds: -1,
-      }),
-    })
-
-    // Verify that the output contains the expected structure
-    expect(output.type).toBeDefined()
-
-    // Verify that a tool result was added to message history
-    const toolResultMessages =
-      newSessionState.mainAgentState.messageHistory.filter(
-        (m) => m.role === 'tool',
-      )
-    expect(toolResultMessages.length).toBeGreaterThan(0)
-  })
-
   it('should handle write_file tool call', async () => {
     // Mock LLM to return a write_file tool call using getToolCallString
     const mockResponse =

packages/agent-runtime/src/__tests__/prompt-caching-subagents.test.ts

@@ -400,7 +400,7 @@ describe('Prompt Caching for Subagents with inheritParentSystemPrompt', () => {
     // Verify both agents use the same system prompt
     expect(parentMessages[0].role).toBe('system')
     expect(childMessages[0].role).toBe('system')
-    expect(childMessages[0].content).toBe(parentMessages[0].content)
+    expect(childMessages[0].content).toEqual(parentMessages[0].content)
 
     // This matching system prompt enables prompt caching:
     // Both agents will have the same system message at the start,

packages/agent-runtime/src/__tests__/spawn-agents-message-history.test.ts

@@ -167,24 +167,27 @@ describe('Spawn Agents Message History', () => {
       (msg: any) => msg.role === 'system',
     )
     expect(systemMessages).toHaveLength(1)
-    expect(systemMessages[0].content).toBe(
-      'This is the parent system prompt that should be excluded',
-    )
+    expect(systemMessages[0].content).toEqual([
+      {
+        type: 'text',
+        text: 'This is the parent system prompt that should be excluded',
+      },
+    ])
 
     // Verify user and assistant messages are included
     expect(
       capturedSubAgentState.messageHistory.find(
-        (msg: any) => msg.content === 'Hello',
+        (msg: any) => msg.content[0]?.text === 'Hello',
       ),
     ).toBeTruthy()
     expect(
       capturedSubAgentState.messageHistory.find(
-        (msg: any) => msg.content === 'Hi there!',
+        (msg: any) => msg.content[0]?.text === 'Hi there!',
       ),
     ).toBeTruthy()
     expect(
       capturedSubAgentState.messageHistory.find(
-        (msg: any) => msg.content === 'How are you?',
+        (msg: any) => msg.content[0]?.text === 'How are you?',
       ),
     ).toBeTruthy()
   })