bindai-provider-google 0.1.0__py3-none-any.whl

bindai_provider_google/__init__.py

@@ -0,0 +1,10 @@
+from bindai_core import ProviderFactory
+
+from .provider import GoogleProvider
+
+
+def register():
+    ProviderFactory.register("google", GoogleProvider)
+
+
+__all__ = ["GoogleProvider", "register"]

bindai_provider_google/client.py

@@ -0,0 +1,18 @@
+from __future__ import annotations
+
+from bindai_core import ProviderConfiguration
+from google import genai
+
+
+class GoogleClient:
+    def __init__(self, configuration: ProviderConfiguration) -> None:
+        self._configuration = configuration
+        self._client: genai.Client | None = None
+
+    @property
+    def client(self) -> genai.Client:
+        if self._client is None:
+            self._client = genai.Client(
+                api_key=self._configuration.api_key,
+            )
+        return self._client

bindai_provider_google/mapper.py

@@ -0,0 +1,49 @@
+from __future__ import annotations
+
+from bindai_core import Message, MessageRole
+
+
+class GoogleMapper:
+    """
+    Converts BindAI messages into Google Gemini SDK-compatible content.
+    """
+
+    @staticmethod
+    def messages(
+        messages: list[Message],
+    ) -> tuple[str | None, list[dict]]:
+        system: str | None = None
+        result: list[dict] = []
+
+        for message in messages:
+            if message.role == MessageRole.SYSTEM:
+                system = message.content
+                continue
+
+            if message.role == MessageRole.USER:
+                result.append(
+                    {
+                        "role": "user",
+                        "parts": [{"text": message.content}],
+                    }
+                )
+                continue
+
+            if message.role == MessageRole.ASSISTANT:
+                result.append(
+                    {
+                        "role": "model",
+                        "parts": [{"text": message.content}],
+                    }
+                )
+                continue
+
+            if message.role == MessageRole.TOOL:
+                result.append(
+                    {
+                        "role": "user",
+                        "parts": [{"text": message.content}],
+                    }
+                )
+
+        return system, result

bindai_provider_google/provider.py

@@ -0,0 +1,129 @@
+from __future__ import annotations
+
+from bindai_core import (
+    ModelProvider,
+    ModelRequest,
+    ModelResponse,
+    ProviderCapabilities,
+    ProviderConfiguration,
+    StreamChunk,
+    TokenUsage,
+)
+
+from .client import GoogleClient
+from .mapper import GoogleMapper
+
+
+class GoogleProvider(ModelProvider):
+    """
+    Google Gemini implementation of ModelProvider.
+    """
+
+    def __init__(
+        self,
+        configuration: ProviderConfiguration,
+    ):
+        super().__init__(configuration)
+        self._client = GoogleClient(configuration)
+
+    @property
+    def name(self) -> str:
+        return "google"
+
+    @property
+    def capabilities(self) -> ProviderCapabilities:
+        return ProviderCapabilities(
+            chat=True,
+            streaming=True,
+            vision=True,
+            embeddings=False,
+            tool_calling=True,
+            structured_output=True,
+            reasoning=True,
+        )
+
+    def _build_kwargs(
+        self,
+        request: ModelRequest,
+    ) -> dict:
+        system, messages = GoogleMapper.messages(
+            request.messages,
+        )
+
+        kwargs: dict[str, object] = {
+            "model": self.configuration.model,
+            "contents": messages,
+        }
+
+        config: dict[str, object] = {}
+
+        if system is not None:
+            config["system_instruction"] = system
+
+        if request.temperature is not None:
+            config["temperature"] = request.temperature
+
+        if request.max_tokens is not None:
+            config["max_output_tokens"] = request.max_tokens
+
+        if request.top_p is not None:
+            config["top_p"] = request.top_p
+
+        if request.stop is not None:
+            config["stop_sequences"] = request.stop
+
+        if config:
+            kwargs["config"] = config
+
+        return kwargs
+
+    def generate(
+        self,
+        request: ModelRequest,
+    ) -> ModelResponse:
+        response = self._client.client.models.generate_content(
+            **self._build_kwargs(request),
+        )
+
+        usage_metadata = response.usage_metadata
+
+        usage = TokenUsage(
+            prompt_tokens=(usage_metadata.prompt_token_count or 0) if usage_metadata else 0,
+            completion_tokens=(usage_metadata.candidates_token_count or 0) if usage_metadata else 0,
+            total_tokens=(usage_metadata.total_token_count or 0) if usage_metadata else 0,
+        )
+
+        content = response.text or ""
+
+        finish_reason = None
+
+        if response.candidates:
+            finish_reason = str(response.candidates[0].finish_reason)
+
+        return ModelResponse(
+            content=content,
+            usage=usage,
+            finish_reason=finish_reason,
+            model=response.model_version or self.configuration.model,
+        )
+
+    def stream(
+        self,
+        request: ModelRequest,
+    ):
+        response = self._client.client.models.generate_content_stream(
+            **self._build_kwargs(request),
+        )
+
+        for chunk in response:
+            text = chunk.text or ""
+
+            if text:
+                yield StreamChunk(
+                    delta=text,
+                )
+
+        yield StreamChunk(
+            delta="",
+            finished=True,
+        )

bindai_provider_google-0.1.0.dist-info/METADATA

@@ -0,0 +1,43 @@
+Metadata-Version: 2.4
+Name: bindai-provider-google
+Version: 0.1.0
+Summary: Google Gemini provider for BindAI
+Requires-Python: >=3.11
+Description-Content-Type: text/markdown
+Requires-Dist: google-genai>=2.22.0
+Requires-Dist: bindai-core
+
+# bindai-provider-google
+
+Google Gemini provider for the BindAI framework.
+
+## Installation
+
+```bash
+pip install "bindai[google]"
+```
+
+Or install the provider directly:
+
+```bash
+pip install bindai-provider-google
+```
+
+## Configuration
+
+Set your Google Gemini API key:
+
+```env
+GEMINI_API_KEY=your-api-key
+```
+
+Then configure your BindAI project:
+
+```toml
+provider = "google"
+model = "gemini-2.5-flash"
+```
+
+## License
+
+MIT

bindai_provider_google-0.1.0.dist-info/RECORD

@@ -0,0 +1,9 @@
+bindai_provider_google/__init__.py,sha256=42z6pKEAq4-qIdgJM9pOQhHq4ssAinsbBY5tHwX3a1I,194
+bindai_provider_google/client.py,sha256=OqysX1OQK5tXrYWS5kClQTkvpBU_locm9k5zzEnjXhA,514
+bindai_provider_google/mapper.py,sha256=GHFFhb9k2gjsFgMIgVVSDT9nve6GD1zyBXIRArKaYsY,1340
+bindai_provider_google/provider.py,sha256=B45aA8z4Y2zbmonE2CRaa3l8NS_iTE2EM6NAW4i2DkI,3235
+bindai_provider_google-0.1.0.dist-info/METADATA,sha256=WfDjsXWjXd7AXjxpehqnGu_6m8rJbwHsfwNR0zpAV8g,702
+bindai_provider_google-0.1.0.dist-info/WHEEL,sha256=YVMoNqKzERt-wjUZwJ33xBGAwnFl-4cqbYkTtWa4itE,91
+bindai_provider_google-0.1.0.dist-info/entry_points.txt,sha256=vggMe0Hffzlhs_ApaY3e2uppbfQEhVrtlm13RcibF68,60
+bindai_provider_google-0.1.0.dist-info/top_level.txt,sha256=uRmYXUwfRxfAY09Ou_BZ9ICUs0Wqf33J3HvOPMAC7WE,23
+bindai_provider_google-0.1.0.dist-info/RECORD,,

bindai_provider_google-0.1.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (84.0.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

bindai_provider_google-0.1.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[bindai.providers]
+google = bindai_provider_google:register

bindai_provider_google-0.1.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+bindai_provider_google