test(b17): add 4 coverage tests for mimo.ts edge cases

Author: Zoo (VP)

src/api/providers/__tests__/mimo.spec.ts

@@ -657,6 +657,149 @@ describe("MimoHandler", () => {
 			expect(mockCreate).toHaveBeenCalledTimes(1)
 		})
 
+		it("should not retry when rejection is a non-Error value (parallel_tool_calls path)", async () => {
+			// A non-Error rejection (e.g. a string) must not trigger the
+			// parallel_tool_calls fallback. isParallelToolCallsRejected
+			// returns false for non-Error values.
+			mockCreate.mockRejectedValueOnce("network failure")
+
+			const messages: Anthropic.Messages.MessageParam[] = [
+				{ role: "user", content: [{ type: "text", text: "Hello" }] },
+			]
+
+			await expect(async () => {
+				const stream = handler.createMessage("System prompt", messages, {
+					taskId: "test-task",
+				parallelToolCalls: false,
+				})
+				for await (const _chunk of stream) {
+					// drain
+				}
+			}).rejects.toThrow()
+
+			expect(mockCreate).toHaveBeenCalledTimes(1)
+		})
+
+		it("should not retry strict-schema fallback for non-400 errors", async () => {
+			// A 500 error must NOT trigger the strict-schema fallback.
+			// isStrictToolSchemaRejected returns false when status !== 400.
+			const rejectionError = Object.assign(
+				new Error("500 - Internal server error"),
+				{ status: 500 },
+			)
+			mockCreate.mockRejectedValueOnce(rejectionError)
+
+			const tools: OpenAI.Chat.ChatCompletionTool[] = [
+				{
+					type: "function",
+					function: { name: "read_file", description: "Read", parameters: {} },
+				},
+			]
+
+			const messages: Anthropic.Messages.MessageParam[] = [
+				{ role: "user", content: [{ type: "text", text: "Hello" }] },
+			]
+
+			await expect(async () => {
+				const stream = handler.createMessage("System prompt", messages, { taskId: "test-task", tools })
+				for await (const _chunk of stream) {
+					// drain
+				}
+			}).rejects.toThrow()
+
+			expect(mockCreate).toHaveBeenCalledTimes(1)
+		})
+
+		it("should not retry strict-schema fallback for non-Error rejections", async () => {
+			// A non-Error rejection (e.g. a string) must not trigger the
+			// strict-schema fallback. isStrictToolSchemaRejected returns
+			// false for non-Error values.
+			mockCreate.mockRejectedValueOnce("bad gateway")
+
+			const tools: OpenAI.Chat.ChatCompletionTool[] = [
+				{
+					type: "function",
+					function: { name: "read_file", description: "Read", parameters: {} },
+				},
+			]
+
+			const messages: Anthropic.Messages.MessageParam[] = [
+				{ role: "user", content: [{ type: "text", text: "Hello" }] },
+			]
+
+			await expect(async () => {
+				const stream = handler.createMessage("System prompt", messages, { taskId: "test-task", tools })
+				for await (const _chunk of stream) {
+					// drain
+				}
+			}).rejects.toThrow()
+
+			expect(mockCreate).toHaveBeenCalledTimes(1)
+		})
+
+		it("should pass non-function tools through unchanged during strict-schema retry", async () => {
+			// When the endpoint rejects strict tool schemas, the retry
+			// strips strict from function tools but passes non-function
+			// tools (e.g. type "code_interpreter") through unchanged.
+			const rejectionError = Object.assign(new Error("400 - Unknown parameter: tools[0].function.strict"), {
+				status: 400,
+			})
+			mockCreate.mockRejectedValueOnce(rejectionError)
+
+			// Second call (retry) succeeds
+			mockCreate.mockImplementationOnce(async () => ({
+				[Symbol.asyncIterator]: async function* () {
+					yield {
+						choices: [{ delta: { content: "Retried" }, index: 0 }],
+						usage: null,
+					}
+					yield {
+						choices: [{ delta: {}, index: 0, finish_reason: "stop" }],
+						usage: { prompt_tokens: 1, completion_tokens: 1, total_tokens: 2 },
+					}
+				},
+			}))
+
+			const tools: OpenAI.Chat.ChatCompletionTool[] = [
+				{
+					type: "function",
+					function: {
+						name: "read_file",
+						description: "Read",
+						parameters: { type: "object", properties: {} },
+						strict: true,
+					},
+				},
+				// Non-function tool — should pass through stripStrictFromTools unchanged
+				{
+					type: "code_interpreter" as OpenAI.Chat.ChatCompletionTool["type"],
+					code_interpreter: { name: "code_interpreter" },
+				} as unknown as OpenAI.Chat.ChatCompletionTool,
+			]
+
+			const messages: Anthropic.Messages.MessageParam[] = [
+				{ role: "user", content: [{ type: "text", text: "Hello" }] },
+			]
+
+			const stream = handler.createMessage("System prompt", messages, { taskId: "test-task", tools })
+			for await (const _chunk of stream) {
+				// drain
+			}
+
+			// The retry call should have been made
+			expect(mockCreate).toHaveBeenCalledTimes(2)
+
+			// The retry call's tools should have the function tool with strict removed
+			// and the non-function tool preserved unchanged
+			const retryCallParams = mockCreate.mock.calls[1][0]
+			expect(retryCallParams.tools).toBeDefined()
+			expect(retryCallParams.tools).toHaveLength(2)
+			// Function tool should have strict removed
+			expect(retryCallParams.tools[0].function).not.toHaveProperty("strict")
+			// Non-function tool should be preserved
+			expect(retryCallParams.tools[1].type).toBe("code_interpreter")
+		})
+
 		it("should send stream_options with include_usage", async () => {
 			const messages: Anthropic.Messages.MessageParam[] = [
 				{ role: "user", content: [{ type: "text", text: "Hello" }] },