ccs-llmconnector 1.1.1__py3-none-any.whl → 1.1.2__py3-none-any.whl

{ccs_llmconnector-1.1.1.dist-info → ccs_llmconnector-1.1.2.dist-info}/METADATA

@@ -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.dist-info → ccs_llmconnector-1.1.2.dist-info}/RECORD

@@ -1,16 +1,16 @@
-ccs_llmconnector-1.1.1.dist-info/licenses/LICENSE,sha256=rPcz2YmBB9VUWZTLJcRO_B4jKDpqmGRYi2eSI-unysg,1083
+ccs_llmconnector-1.1.2.dist-info/licenses/LICENSE,sha256=rPcz2YmBB9VUWZTLJcRO_B4jKDpqmGRYi2eSI-unysg,1083
 llmconnector/__init__.py,sha256=w68d7BCkhjjqMaOG9VDkPoYR4PNbv8NJqGmAV_Vyn6g,1358
 llmconnector/anthropic_client.py,sha256=hGFZlUQ4yAs_H8StxWzHAYmqwgNT6lGI4yW8UmKhtm8,13260
 llmconnector/client.py,sha256=2rhDpcBKKbQ1kGp7-Bk8ci_VVe9DEEvBf9nH4hKwg3w,13849
 llmconnector/client_cli.py,sha256=ojLPNJ14lFycsLWtEMtdGpV033v2-23qPMuWqFlnySA,11463
-llmconnector/gemini_client.py,sha256=_XPy6C5WQeuP_GY2eb3dy4DMFX9Oj8DRhb9u7XA5TfI,19155
+llmconnector/gemini_client.py,sha256=6J_dhPGEjh2n30uM8Jo1gUocyC_v7HnqZrXYXA5BVIA,19407
 llmconnector/grok_client.py,sha256=uicZI28-vw7go677zZB5MTkw5aqWp2UP_GaguNUU99o,11254
 llmconnector/openai_client.py,sha256=wvrEFzltg8PctM2DcNZ_7qmRPpF-CB8WQek6iYoRvR4,11051
 llmconnector/py.typed,sha256=frcCV1k9oG9oKj3dpUqdJg1PxRT2RSN_XKdLCPjaYaY,2
 llmconnector/types.py,sha256=JhXKjKncdUB2yPd-bqjOUnl-4Wgs6GIleDMChmwPW8w,1287
 llmconnector/utils.py,sha256=Tw6wQLh7Cjk6igLJ-MnYGSSPJBUGl5-7R1xB0Aj7o8Y,2047
-ccs_llmconnector-1.1.1.dist-info/METADATA,sha256=k4esAN7SCw583miw_IqV9sehyt0GGcgGylpLLPAL35s,17001
-ccs_llmconnector-1.1.1.dist-info/WHEEL,sha256=qELbo2s1Yzl39ZmrAibXA2jjPLUYfnVhUNTlyF1rq0Y,92
-ccs_llmconnector-1.1.1.dist-info/entry_points.txt,sha256=eFvLY3nHAG_QhaKlemhhK7echfezW0KiMdSNMZOStLc,60
-ccs_llmconnector-1.1.1.dist-info/top_level.txt,sha256=Doer7TAUsN8UXQfPHPNsuBXVNCz2uV-Q0v4t4fwv_MM,13
-ccs_llmconnector-1.1.1.dist-info/RECORD,,
+ccs_llmconnector-1.1.2.dist-info/METADATA,sha256=LEWid1Hh1gdCA3rcF5KskOpc_Em8najLRjZdA1ep8kI,17001
+ccs_llmconnector-1.1.2.dist-info/WHEEL,sha256=qELbo2s1Yzl39ZmrAibXA2jjPLUYfnVhUNTlyF1rq0Y,92
+ccs_llmconnector-1.1.2.dist-info/entry_points.txt,sha256=eFvLY3nHAG_QhaKlemhhK7echfezW0KiMdSNMZOStLc,60
+ccs_llmconnector-1.1.2.dist-info/top_level.txt,sha256=Doer7TAUsN8UXQfPHPNsuBXVNCz2uV-Q0v4t4fwv_MM,13
+ccs_llmconnector-1.1.2.dist-info/RECORD,,

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: