From 911dd19ac1520054b48a1ddcf04393ee45f5e427 Mon Sep 17 00:00:00 2001
From: Georgi Atsev <georgi@deepinfra.com>
Date: Thu, 19 Mar 2026 16:02:29 +0200
Subject: [PATCH] Fix gemini caps for deepinfra mock in test

---
 src/agents/provider-capabilities.test.ts | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/agents/provider-capabilities.test.ts b/src/agents/provider-capabilities.test.ts
index cd560471122..fb5648490a7 100644
--- a/src/agents/provider-capabilities.test.ts
+++ b/src/agents/provider-capabilities.test.ts
@@ -17,6 +17,12 @@ const resolveProviderCapabilitiesWithPluginMock = vi.fn((params: { provider: str
         geminiThoughtSignatureSanitization: true,
         geminiThoughtSignatureModelHints: ["gemini"],
       };
+    case "deepinfra":
+      return {
+        openAiCompatTurnValidation: false,
+        geminiThoughtSignatureSanitization: true,
+        geminiThoughtSignatureModelHints: ["gemini"],
+      };
     case "openai-codex":
       return {
         providerFamily: "openai",