diff --git a/patchwork/common/client/llm/google_.py b/patchwork/common/client/llm/google_.py
index 62b48e5ec..8b3abe2aa 100644
--- a/patchwork/common/client/llm/google_.py
+++ b/patchwork/common/client/llm/google_.py
@@ -264,7 +264,7 @@ def chat_completion(
         max_tokens: Optional[int] | NotGiven = NOT_GIVEN,
         n: Optional[int] | NotGiven = NOT_GIVEN,
         presence_penalty: Optional[float] | NotGiven = NOT_GIVEN,
-        response_format: str | completion_create_params.ResponseFormat | NotGiven = NOT_GIVEN,
+        response_format: dict | completion_create_params.ResponseFormat | NotGiven = NOT_GIVEN,
         stop: Union[Optional[str], List[str]] | NotGiven = NOT_GIVEN,
         temperature: Optional[float] | NotGiven = NOT_GIVEN,
         tools: Iterable[ChatCompletionToolParam] | NotGiven = NOT_GIVEN,
diff --git a/patchwork/common/client/llm/protocol.py b/patchwork/common/client/llm/protocol.py
index 3e47b38c9..21c5acfed 100644
--- a/patchwork/common/client/llm/protocol.py
+++ b/patchwork/common/client/llm/protocol.py
@@ -138,7 +138,7 @@ def chat_completion(
         max_tokens: Optional[int] | NotGiven = NOT_GIVEN,
         n: Optional[int] | NotGiven = NOT_GIVEN,
         presence_penalty: Optional[float] | NotGiven = NOT_GIVEN,
-        response_format: str | completion_create_params.ResponseFormat | NotGiven = NOT_GIVEN,
+        response_format: dict | completion_create_params.ResponseFormat | NotGiven = NOT_GIVEN,
         stop: Union[Optional[str], List[str]] | NotGiven = NOT_GIVEN,
         temperature: Optional[float] | NotGiven = NOT_GIVEN,
         tools: Iterable[ChatCompletionToolParam] | NotGiven = NOT_GIVEN,