fix: gemini timeout error (#4955)

2024-06-06 10:19:03 +08:00 · 2024-06-06 10:19:03 +08:00 · 5cdb95be1f
commit 5cdb95be1f
parent 7fa735a43b
1 changed files with 1 additions and 0 deletions
--- 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: