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, ):