swarmauri_provider_gemini 0.1.0.dev3__py3-none-any.whl

swarmauri_provider_gemini/GeminiAuthConfig.py

@@ -0,0 +1,27 @@
+from collections.abc import Mapping
+from typing import Literal
+
+from pydantic import Field, SecretStr
+from swarmauri_base.model_providers.ProviderAuthBase import ProviderAuthBase
+from swarmauri_base.ComponentBase import ComponentBase
+
+
+@ComponentBase.register_type(ProviderAuthBase, "GeminiAuthConfig")
+class GeminiAuthConfig(ProviderAuthBase):
+    api_key: SecretStr | None = Field(default=None, exclude=True)
+    vertex_project: str | None = None
+    vertex_location: str | None = None
+    type: Literal["GeminiAuthConfig"] = "GeminiAuthConfig"
+
+    def apply(self, headers: Mapping[str, str]) -> dict[str, str]:
+        authorized = dict(headers)
+        if self.api_key is not None:
+            authorized["x-goog-api-key"] = self.api_key.get_secret_value()
+        return authorized
+
+    def apply_query(self, query: Mapping[str, str]) -> dict[str, str]:
+        authorized = dict(query)
+        if self.api_key is None:
+            raise ValueError("GeminiProvider requires an api_key")
+        authorized["key"] = self.api_key.get_secret_value()
+        return authorized

swarmauri_provider_gemini/GeminiEndpoints.py

@@ -0,0 +1,28 @@
+from typing import Literal
+
+from swarmauri_base.model_providers.ProviderEndpointBase import (
+    ProviderEndpointBase,
+)
+from swarmauri_base.ComponentBase import ComponentBase
+
+
+@ComponentBase.register_type(ProviderEndpointBase, "GeminiEndpoints")
+class GeminiEndpoints(ProviderEndpointBase):
+    base_url: str = "https://generativelanguage.googleapis.com/v1beta"
+    websocket_url: str = (
+        "wss://generativelanguage.googleapis.com/ws/"
+        "google.ai.generativelanguage.v1beta.GenerativeService.BidiGenerateContent"
+    )
+    type: Literal["GeminiEndpoints"] = "GeminiEndpoints"
+
+    def resolve(self, capability: str) -> str:
+        endpoints = {
+            "api": self.base_url,
+            "live.connect": self.websocket_url,
+        }
+        try:
+            return endpoints[capability]
+        except KeyError as error:
+            raise ValueError(
+                f"Gemini does not support capability {capability!r}"
+            ) from error

swarmauri_provider_gemini/GeminiErrorMapper.py

@@ -0,0 +1,16 @@
+from typing import Any, Literal
+
+from swarmauri_base.model_providers.ProviderErrorMapperBase import (
+    ProviderErrorMapperBase,
+)
+from swarmauri_base.ComponentBase import ComponentBase
+
+
+@ComponentBase.register_type(ProviderErrorMapperBase, "GeminiErrorMapper")
+class GeminiErrorMapper(ProviderErrorMapperBase):
+    type: Literal["GeminiErrorMapper"] = "GeminiErrorMapper"
+
+    def map_error(self, error: BaseException | Any) -> BaseException:
+        if isinstance(error, BaseException):
+            return RuntimeError(f"Gemini provider error: {error}")
+        return RuntimeError(f"Gemini provider error: {error}")

swarmauri_provider_gemini/GeminiModelCatalog.py

@@ -0,0 +1,17 @@
+from typing import Literal
+
+from swarmauri_base.model_providers.ModelCatalogBase import ModelCatalogBase
+from swarmauri_base.ComponentBase import ComponentBase
+
+
+@ComponentBase.register_type(ModelCatalogBase, "GeminiModelCatalog")
+class GeminiModelCatalog(ModelCatalogBase):
+    models: dict = {
+        "gemini-3.1-flash-live-preview": {
+            "name": "gemini-3.1-flash-live-preview",
+            "input_modalities": ["text", "audio", "image", "video"],
+            "output_modalities": ["audio", "text"],
+            "features": ["realtime", "tools", "transcription"],
+        }
+    }
+    type: Literal["GeminiModelCatalog"] = "GeminiModelCatalog"

swarmauri_provider_gemini/GeminiProvider.py

@@ -0,0 +1,46 @@
+from typing import Literal
+
+from pydantic import Field, model_validator
+from swarmauri_base.ComponentBase import ComponentBase
+from swarmauri_base.model_providers.ModelProviderBase import ModelProviderBase
+from swarmauri_base.model_providers.ProviderConnectionConfig import (
+    ProviderConnectionConfig,
+)
+
+from .GeminiAuthConfig import GeminiAuthConfig
+from .GeminiEndpoints import GeminiEndpoints
+from .GeminiErrorMapper import GeminiErrorMapper
+from .GeminiModelCatalog import GeminiModelCatalog
+
+
+@ComponentBase.register_type(ModelProviderBase, "GeminiProvider")
+class GeminiProvider(ModelProviderBase):
+    provider_name: Literal["gemini"] = "gemini"
+    endpoints: GeminiEndpoints = Field(default_factory=GeminiEndpoints)
+    auth: GeminiAuthConfig = Field(
+        default_factory=GeminiAuthConfig, exclude=True
+    )
+    catalog: GeminiModelCatalog = Field(default_factory=GeminiModelCatalog)
+    error_mapper: GeminiErrorMapper = Field(default_factory=GeminiErrorMapper)
+    type: Literal["GeminiProvider"] = "GeminiProvider"
+
+    @model_validator(mode="after")
+    def configure(self) -> "GeminiProvider":
+        self.base_url = self.base_url or self.endpoints.resolve("api")
+        self.endpoints.base_url = self.base_url
+        if self.api_key is not None:
+            self.auth.api_key = self.api_key
+        elif self.auth.api_key is not None:
+            self.api_key = self.auth.api_key
+        return self
+
+    def prepare_connection(
+        self, capability: str, model: str | None = None
+    ) -> ProviderConnectionConfig:
+        if capability != "live.connect":
+            return super().prepare_connection(capability, model)
+
+        config = self.connection_config(capability, model)
+        return config.model_copy(
+            update={"query": dict(self.auth.apply_query(config.query))}
+        )

swarmauri_provider_gemini/GeminiToolSchemaConverter.py

@@ -0,0 +1,27 @@
+from typing import Any
+
+
+class GeminiToolSchemaConverter:
+    def convert(self, tool: Any) -> dict[str, Any]:
+        parameters = getattr(tool, "parameters", [])
+        properties = {
+            parameter.name: {
+                "type": getattr(parameter, "type", "string"),
+                "description": getattr(parameter, "description", None),
+            }
+            for parameter in parameters
+        }
+        required = [
+            parameter.name
+            for parameter in parameters
+            if getattr(parameter, "required", False)
+        ]
+        return {
+            "name": tool.name,
+            "description": getattr(tool, "description", None),
+            "parameters": {
+                "type": "object",
+                "properties": properties,
+                "required": required,
+            },
+        }

swarmauri_provider_gemini/__init__.py

@@ -0,0 +1,15 @@
+from .GeminiAuthConfig import GeminiAuthConfig
+from .GeminiEndpoints import GeminiEndpoints
+from .GeminiErrorMapper import GeminiErrorMapper
+from .GeminiModelCatalog import GeminiModelCatalog
+from .GeminiProvider import GeminiProvider
+from .GeminiToolSchemaConverter import GeminiToolSchemaConverter
+
+__all__ = [
+    "GeminiAuthConfig",
+    "GeminiEndpoints",
+    "GeminiErrorMapper",
+    "GeminiModelCatalog",
+    "GeminiProvider",
+    "GeminiToolSchemaConverter",
+]

swarmauri_provider_gemini-0.1.0.dev3.dist-info/METADATA

@@ -0,0 +1,51 @@
+Metadata-Version: 2.4
+Name: swarmauri_provider_gemini
+Version: 0.1.0.dev3
+Summary: Aggregated Google Gemini provider resources for Swarmauri models.
+License: Copyright 2024 Jacob Stewart
+         
+         Licensed under the Apache License, Version 2.0 (the "License");
+         you may not use this file except in compliance with the License.
+         You may obtain a copy of the License at
+         
+             http://www.apache.org/licenses/LICENSE-2.0
+         
+         Unless required by applicable law or agreed to in writing, software
+         distributed under the License is distributed on an "AS IS" BASIS,
+         WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+         See the License for the specific language governing permissions and
+         limitations under the License.
+License-File: LICENSE
+Keywords: swarmauri,provider,gemini,google,model catalog
+Author: Jacob Stewart
+Author-email: jacob@swarmauri.com
+Requires-Python: >=3.10,<3.13
+Classifier: Development Status :: 1 - Planning
+Classifier: License :: OSI Approved :: Apache Software License
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Requires-Dist: swarmauri_base (>=0.11.0.dev5)
+Requires-Dist: swarmauri_core (>=0.11.0.dev5)
+Description-Content-Type: text/markdown
+
+![Swarmauri](https://github.com/swarmauri/swarmauri-sdk/blob/3d4d1cfa949399d7019ae9d8f296afba773dfb7f/assets/swarmauri.brand.theme.svg)
+
+[![PyPI](https://img.shields.io/pypi/v/swarmauri_provider_gemini.svg)](https://pypi.org/project/swarmauri_provider_gemini/) [![Python](https://img.shields.io/pypi/pyversions/swarmauri_provider_gemini.svg)](https://pypi.org/project/swarmauri_provider_gemini/) [![License](https://img.shields.io/badge/license-Apache--2.0-blue)](LICENSE) [![Downloads](https://static.pepy.tech/badge/swarmauri_provider_gemini)](https://pepy.tech/project/swarmauri_provider_gemini) [![Hits](https://hits.sh/github.com/swarmauri/swarmauri-sdk.svg)](https://hits.sh/github.com/swarmauri/swarmauri-sdk/)
+
+# swarmauri_provider_gemini
+
+## Features
+
+- Gemini credentials and endpoint configuration.
+- Shared model catalog, error mapping, and tool schema conversion.
+- Reusable by LLM, VLM, and LiveModel packages.
+
+## Installation
+
+`uv add swarmauri_provider_gemini` or `pip install swarmauri_provider_gemini`.
+
+## Usage
+
+Construct `GeminiProvider(api_key=...)` and inject it into a Gemini model adapter.
+

swarmauri_provider_gemini-0.1.0.dev3.dist-info/RECORD

@@ -0,0 +1,12 @@
+swarmauri_provider_gemini/GeminiAuthConfig.py,sha256=2cmLeQZn_o5E0xwDlbRd0cXHbyrl5czXAnk1FYAlzaY,1079
+swarmauri_provider_gemini/GeminiEndpoints.py,sha256=27iy1IC3FUxuy0poCVits7PR1nFF9V77fMHANXpxXDs,967
+swarmauri_provider_gemini/GeminiErrorMapper.py,sha256=VpqNUK7fdL4IVVjJWZOkx_3CnwvaH7eTCkPGYz6Mw0I,622
+swarmauri_provider_gemini/GeminiModelCatalog.py,sha256=Vky-4ES8HEXnYZG6Ydwq3z770u4CS8T2qm56x_pyzpU,656
+swarmauri_provider_gemini/GeminiProvider.py,sha256=kchnC_02NDiLPF2LLELmW8WaKAOAv5nPyK6xKTkt1pU,1830
+swarmauri_provider_gemini/GeminiToolSchemaConverter.py,sha256=xNJJGVQHwmSDCJ8HV0Gp95US4yhmjpPR4TuhHGD3-yw,843
+swarmauri_provider_gemini/__init__.py,sha256=gOA_6DxNED_UOMzW2Yn1DTZ5kxvtLt2R6jtGk994uSU,468
+swarmauri_provider_gemini-0.1.0.dev3.dist-info/METADATA,sha256=FJLMMLuvjf1OCtA_f7ex1rx2aTLR_Or1b9aHAdeFXws,2498
+swarmauri_provider_gemini-0.1.0.dev3.dist-info/WHEEL,sha256=L-WvLdSBvJWHXv5zrQxDvNRnIgoBzLUpcbuiw19KCuQ,88
+swarmauri_provider_gemini-0.1.0.dev3.dist-info/entry_points.txt,sha256=W9m6Q2uUqwEj1mAclPRUU2U5e-ckukip1IYH_8fTjcg,85
+swarmauri_provider_gemini-0.1.0.dev3.dist-info/licenses/LICENSE,sha256=kkwAgrDtfQ7c-Ozql2xlztDDjHt0YwGmRG7b_oUU9Jg,554
+swarmauri_provider_gemini-0.1.0.dev3.dist-info/RECORD,,

swarmauri_provider_gemini-0.1.0.dev3.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: poetry-core 2.5.0
+Root-Is-Purelib: true
+Tag: py3-none-any

swarmauri_provider_gemini-0.1.0.dev3.dist-info/entry_points.txt

@@ -0,0 +1,3 @@
+[swarmauri.model_providers]
+GeminiProvider=swarmauri_provider_gemini:GeminiProvider
+

swarmauri_provider_gemini-0.1.0.dev3.dist-info/licenses/LICENSE

@@ -0,0 +1,13 @@
+Copyright 2024 Jacob Stewart
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.