From 5cdb95be1facd82b97549639afca3475871fdc81 Mon Sep 17 00:00:00 2001
From: Joe <79627742+ZhouhaoJiang@users.noreply.github.com>
Date: Thu, 6 Jun 2024 10:19:03 +0800
Subject: [PATCH] fix: gemini timeout error (#4955)

---
 api/core/model_runtime/model_providers/google/llm/llm.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/api/core/model_runtime/model_providers/google/llm/llm.py b/api/core/model_runtime/model_providers/google/llm/llm.py
index 27912b13cc..5a674fdeee 100644
--- a/api/core/model_runtime/model_providers/google/llm/llm.py
+++ b/api/core/model_runtime/model_providers/google/llm/llm.py
@@ -204,6 +204,7 @@ class GoogleLargeLanguageModel(LargeLanguageModel):
             stream=stream,
             safety_settings=safety_settings,
             tools=self._convert_tools_to_glm_tool(tools) if tools else None,
+            request_options={"timeout": 600}
         )
 
         if stream: