Skip to content

Commit 7bf90c8

Browse files
committed
increase timeouts for gemini api
1 parent 987ea93 commit 7bf90c8

File tree

2 files changed

+5
-4
lines changed

2 files changed

+5
-4
lines changed

community/gemini/src/services.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -41,8 +41,8 @@ async def main():
4141
task_queue="gemini",
4242
functions=[gemini_generate_content, gemini_function_call, gemini_multi_function_call, gemini_multi_function_call_advanced],
4343
options=ServiceOptions(
44-
rate_limit=5,
45-
max_concurrent_function_runs=3
44+
rate_limit=0.1,
45+
max_concurrent_function_runs=1
4646
)
4747
)
4848
)

community/gemini/src/workflows/multi_function_call_advanced.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -36,8 +36,9 @@ async def run(self, input: MultiFunctionCallAdvancedInputParams):
3636
user_content=current_content,
3737
chat_history=self.chat_history
3838
),
39-
start_to_close_timeout=timedelta(seconds=120),
40-
retry_policy=RetryPolicy(maximum_attempts=2),
39+
schedule_to_close_timeout=timedelta(seconds=860),
40+
start_to_close_timeout=timedelta(seconds=860),
41+
retry_policy=RetryPolicy(maximum_attempts=3),
4142
task_queue="gemini"
4243
)
4344

0 commit comments

Comments
 (0)