ccs-llmconnector 1.1.1__tar.gz → 1.1.2__tar.gz

{ccs_llmconnector-1.1.1/src/ccs_llmconnector.egg-info → ccs_llmconnector-1.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ccs-llmconnector
-Version: 1.1.1
+Version: 1.1.2
 Summary: Lightweight wrapper around different LLM provider Python SDK Responses APIs.
 Author: CCS
 License: MIT

{ccs_llmconnector-1.1.1 → ccs_llmconnector-1.1.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "ccs-llmconnector"
-version = "1.1.1"
+version = "1.1.2"
 description = "Lightweight wrapper around different LLM provider Python SDK Responses APIs."
 readme = "README.md"
 requires-python = ">=3.8"

{ccs_llmconnector-1.1.1 → ccs_llmconnector-1.1.2/src/ccs_llmconnector.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ccs-llmconnector
-Version: 1.1.1
+Version: 1.1.2
 Summary: Lightweight wrapper around different LLM provider Python SDK Responses APIs.
 Author: CCS
 License: MIT

{ccs_llmconnector-1.1.1 → ccs_llmconnector-1.1.2}/src/llmconnector/gemini_client.py

@@ -15,11 +15,29 @@ from google.genai import types
 from .types import ImageInput, MessageSequence, normalize_messages
 from .utils import clamp_retries, run_sync_in_thread, run_with_retries
 
-logger = logging.getLogger(__name__)
-
-
-class GeminiClient:
-    """Convenience wrapper around the Google Gemini SDK."""
+logger = logging.getLogger(__name__)
+
+
+_GEMINI_MIN_TIMEOUT_S = 10.0
+_GEMINI_MIN_TIMEOUT_MS = int(_GEMINI_MIN_TIMEOUT_S * 1000)
+
+
+def _normalize_gemini_timeout_ms(timeout_s: float) -> int:
+    """Convert a seconds timeout into the millisecond value expected by google-genai HttpOptions."""
+    # google-genai HttpOptions expects milliseconds, but our public API uses seconds.
+    effective_timeout_s = max(_GEMINI_MIN_TIMEOUT_S, timeout_s)
+    if effective_timeout_s != timeout_s:
+        logger.warning(
+            "Gemini timeout %ss is too short, clamping to %ss.",
+            timeout_s,
+            effective_timeout_s,
+        )
+    timeout_ms = int(effective_timeout_s * 1000)
+    return max(_GEMINI_MIN_TIMEOUT_MS, timeout_ms)
+
+
+class GeminiClient:
+    """Convenience wrapper around the Google Gemini SDK."""
 
     def generate_response(
         self,
@@ -102,15 +120,13 @@ class GeminiClient:
 
         retry_count = clamp_retries(max_retries)
 
-        def _build_client() -> genai.Client:
-            client_kwargs: dict[str, object] = {"api_key": api_key}
-            if timeout_s is not None:
-                # Gemini requires at least 10s timeout if set
-                effective_timeout = max(10.0, timeout_s)
-                if effective_timeout != timeout_s:
-                    logger.warning("Gemini timeout %ss is too short, clamping to %ss.", timeout_s, effective_timeout)
-                client_kwargs["http_options"] = types.HttpOptions(timeout=effective_timeout)
-            return genai.Client(**client_kwargs)
+        def _build_client() -> genai.Client:
+            client_kwargs: dict[str, object] = {"api_key": api_key}
+            if timeout_s is not None:
+                client_kwargs["http_options"] = types.HttpOptions(
+                    timeout=_normalize_gemini_timeout_ms(timeout_s)
+                )
+            return genai.Client(**client_kwargs)
 
         def _run_request() -> str:
             client = _build_client()
@@ -268,15 +284,13 @@ class GeminiClient:
 
         retry_count = clamp_retries(max_retries)
 
-        def _build_client() -> genai.Client:
-            client_kwargs: dict[str, object] = {"api_key": api_key}
-            if timeout_s is not None:
-                # Gemini requires at least 10s timeout if set
-                effective_timeout = max(10.0, timeout_s)
-                if effective_timeout != timeout_s:
-                    logger.warning("Gemini timeout %ss is too short, clamping to %ss.", timeout_s, effective_timeout)
-                client_kwargs["http_options"] = types.HttpOptions(timeout=effective_timeout)
-            return genai.Client(**client_kwargs)
+        def _build_client() -> genai.Client:
+            client_kwargs: dict[str, object] = {"api_key": api_key}
+            if timeout_s is not None:
+                client_kwargs["http_options"] = types.HttpOptions(
+                    timeout=_normalize_gemini_timeout_ms(timeout_s)
+                )
+            return genai.Client(**client_kwargs)
 
         def _run_request() -> bytes:
             client = _build_client()
@@ -362,15 +376,13 @@ class GeminiClient:
 
         retry_count = clamp_retries(max_retries)
 
-        def _build_client() -> genai.Client:
-            client_kwargs: dict[str, object] = {"api_key": api_key}
-            if timeout_s is not None:
-                # Gemini requires at least 10s timeout if set
-                effective_timeout = max(10.0, timeout_s)
-                if effective_timeout != timeout_s:
-                    logger.warning("Gemini timeout %ss is too short, clamping to %ss.", timeout_s, effective_timeout)
-                client_kwargs["http_options"] = types.HttpOptions(timeout=effective_timeout)
-            return genai.Client(**client_kwargs)
+        def _build_client() -> genai.Client:
+            client_kwargs: dict[str, object] = {"api_key": api_key}
+            if timeout_s is not None:
+                client_kwargs["http_options"] = types.HttpOptions(
+                    timeout=_normalize_gemini_timeout_ms(timeout_s)
+                )
+            return genai.Client(**client_kwargs)
 
         def _run_request() -> list[dict[str, Optional[str]]]:
             models: list[dict[str, Optional[str]]] = []
@@ -457,12 +469,18 @@ class GeminiClient:
         return _part_from_path(Path(image))
 
 
-def _part_from_path(path: Path) -> types.Part:
-    """Create an image part from a local filesystem path."""
-    expanded = path.expanduser()
-    data = expanded.read_bytes()
-    mime_type = mimetypes.guess_type(expanded.name)[0] or "application/octet-stream"
-    return types.Part.from_bytes(data=data, mime_type=mime_type)
+def _part_from_path(path: Path) -> types.Part:
+    """Create an image part from a local filesystem path."""
+    # Ensure common audio types are recognized across platforms (used for transcription as well).
+    mimetypes.add_type("audio/mp4", ".m4a")
+    mimetypes.add_type("audio/mpeg", ".mp3")
+    mimetypes.add_type("audio/wav", ".wav")
+    mimetypes.add_type("audio/aac", ".aac")
+
+    expanded = path.expanduser()
+    data = expanded.read_bytes()
+    mime_type = mimetypes.guess_type(expanded.name)[0] or "application/octet-stream"
+    return types.Part.from_bytes(data=data, mime_type=mime_type)
 
 
 def _part_from_url(url: str) -> types.Part: