From 5da6d45cc480ca8d6c97dce6e6576a21a9e7029a Mon Sep 17 00:00:00 2001 From: shanshan wang Date: Sun, 21 Jun 2026 09:32:23 +0800 Subject: [PATCH] updated gemini model name --- src/tensorlake_docai/ocr/gemini.py | 2 +- src/tensorlake_docai/providers/model_provider_utils.py | 6 +++--- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/src/tensorlake_docai/ocr/gemini.py b/src/tensorlake_docai/ocr/gemini.py index d7ce599..37a8b61 100644 --- a/src/tensorlake_docai/ocr/gemini.py +++ b/src/tensorlake_docai/ocr/gemini.py @@ -19,7 +19,7 @@ from tensorlake_docai.pipeline.simple_page_creator import SimplePageCreator from tensorlake_docai.extraction.form_extraction_utils import convert_form_json_to_markdown -GEMINI_MODEL = "gemini-3.1-flash-lite-preview" +GEMINI_MODEL = "gemini-3.1-flash-lite" PROVIDER_NAME = "GEMINI" PROMPT = ( diff --git a/src/tensorlake_docai/providers/model_provider_utils.py b/src/tensorlake_docai/providers/model_provider_utils.py index 42186a0..0cfebfb 100644 --- a/src/tensorlake_docai/providers/model_provider_utils.py +++ b/src/tensorlake_docai/providers/model_provider_utils.py @@ -24,14 +24,14 @@ def get_gemini_async_client_and_model( - api_key: Optional[str] = None, model_name: str = "gemini-3.1-flash-lite-preview" + api_key: Optional[str] = None, model_name: str = "gemini-3.1-flash-lite" ): """ Return (async Gemini client (.aio), model_name). Args: api_key: Optional API key. If None, uses "GEMINI_API_KEY" from env. - model_name: Model name to use. Defaults to "gemini-3.1-flash-lite-preview". + model_name: Model name to use. Defaults to "gemini-3.1-flash-lite". Returns: Tuple of (client.aio, model_name) @@ -145,7 +145,7 @@ async def _make_gemini_call( job_type=None, timeout: Optional[int] = None, pdf_bytes=None, - model_name: str = "gemini-3.1-flash-lite-preview", + model_name: str = "gemini-3.1-flash-lite", system_instruction=None, config_overrides=None, ):