From 18f68e3c28d8c2ab07c0c8e98f2577b66c449e2f Mon Sep 17 00:00:00 2001 From: Langfun Authors Date: Thu, 1 May 2025 16:48:43 -0700 Subject: [PATCH] Internal change PiperOrigin-RevId: 753777730 --- langfun/core/langfunc_test.py | 2 +- langfun/core/language_model.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/langfun/core/langfunc_test.py b/langfun/core/langfunc_test.py index b6dd88e9..e04e7950 100644 --- a/langfun/core/langfunc_test.py +++ b/langfun/core/langfunc_test.py @@ -108,7 +108,7 @@ def test_call(self): ' max_tokens=None, n=1, top_k=40, top_p=None, stop=None,' ' random_seed=None, logprobs=False, top_logprobs=None,' ' max_thinking_tokens=None, reasoning_effort=None), cache=None,' - ' max_concurrency=None, timeout=120.0, max_attempts=5,' + ' max_concurrency=None, timeout=120, max_attempts=5,' ' retry_interval=(5, 60), exponential_backoff=True,' ' max_retry_interval=300, debug=False))', ) diff --git a/langfun/core/language_model.py b/langfun/core/language_model.py index b0eb705c..63835886 100644 --- a/langfun/core/language_model.py +++ b/langfun/core/language_model.py @@ -707,8 +707,8 @@ class LanguageModel(component.Component): ] = None timeout: Annotated[ - float | None, 'Timeout in seconds. If None, there is no timeout.' - ] = 120.0 + int | None, 'Timeout in seconds. If None, there is no timeout.' + ] = 120 max_attempts: Annotated[ int,