suvereo-model-google 1.0.0__tar.gz

suvereo_model_google-1.0.0/PKG-INFO

@@ -0,0 +1,35 @@
+Metadata-Version: 2.4
+Name: suvereo-model-google
+Version: 1.0.0
+Summary: A plugin for Suvereo that provides a Google Gemini connector.
+Author: Damian Sowiński
+Author-email: damian.sowinski@upsidelab.io
+Requires-Python: >=3.10,<4.0
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
+Classifier: Programming Language :: Python :: 3.15
+Requires-Dist: google-genai (>=1.70.0,<2.0.0)
+Requires-Dist: langchain-core (>=1.2,<2.0)
+Requires-Dist: langchain-google-genai (>=4.2,<5.0)
+Requires-Dist: suvereo-common (>=1.0.0,<2)
+Description-Content-Type: text/markdown
+
+# Google models
+
+## This Suvereo plugin enables Google models for generating embeddings and LLMs
+
+### 1. Install the `suvereo-model-google` package using pip:
+
+```shell
+pip install suvereo-model-google
+```
+### 3. Set environment variable
+```
+GOOGLE_API_KEY=<your_gemini_api_key>
+```
+
+

suvereo_model_google-1.0.0/README.md

@@ -0,0 +1,14 @@
+# Google models
+
+## This Suvereo plugin enables Google models for generating embeddings and LLMs
+
+### 1. Install the `suvereo-model-google` package using pip:
+
+```shell
+pip install suvereo-model-google
+```
+### 3. Set environment variable
+```
+GOOGLE_API_KEY=<your_gemini_api_key>
+```
+

suvereo_model_google-1.0.0/pyproject.toml

@@ -0,0 +1,18 @@
+[tool.poetry]
+name = "suvereo-model-google"
+version = "1.0.0"
+description = "A plugin for Suvereo that provides a Google Gemini connector."
+authors = ["Damian Sowiński <damian.sowinski@upsidelab.io>"]
+readme = "README.md"
+
+[tool.poetry.dependencies]
+langchain-core = "^1.2"
+python = "^3.10"
+suvereo-common = ">=1.0.0,<2"
+google-genai = "^1.70.0"
+langchain-google-genai = "^4.2"
+
+
+[build-system]
+requires = ["poetry-core"]
+build-backend = "poetry.core.masonry.api"

suvereo_model_google-1.0.0/suvereo_model_google/__init__.py

@@ -0,0 +1,2 @@
+from .embedding import GoogleEmbeddingProvider as GoogleEmbeddingProvider
+from .language_model import GoogleLanguageModelProvider as GoogleLanguageModelProvider

suvereo_model_google-1.0.0/suvereo_model_google/embedding.py

@@ -0,0 +1,35 @@
+from google import genai
+from google.genai.types import EmbedContentConfig
+from suvereo_common.registry.embeddings import EmbeddingProvider
+from suvereo_common.utils import prioritize_items
+
+PRIORITIZED_MODELS = ["models/embedding-001", "models/text-embedding-004"]
+
+
+class GoogleEmbeddingProvider(EmbeddingProvider):
+    NAME = "Google"
+
+    def generate_embeddings(self, content: str) -> list[float]:
+        """
+        Generates and returns an embedding vector for the given content using OpenAI's embeddings API.
+
+        Args:
+            content (str): The input text for which the embedding vector is to be generated.
+        """
+        config = EmbedContentConfig(output_dimensionality=self._dimensions)
+        with genai.Client() as client:
+            google_embedding = client.models.embed_content(
+                model=self._model,
+                config=config,
+                contents=content,
+            )
+
+        return google_embedding.embeddings[0].values
+
+    @staticmethod
+    def available_models() -> list[str]:
+        with genai.Client() as client:
+            all_models = client.models.list()
+
+        embedding_models = [model.name for model in all_models if "embedContent" in model.supported_actions]
+        return prioritize_items(embedding_models, PRIORITIZED_MODELS)

suvereo_model_google-1.0.0/suvereo_model_google/language_model.py

@@ -0,0 +1,49 @@
+from google import genai
+from langchain_core.callbacks import BaseCallbackHandler
+from langchain_core.language_models import BaseLanguageModel
+from langchain_google_genai import ChatGoogleGenerativeAI
+from suvereo_common.registry.llm import LanguageModelProvider
+from suvereo_common.structures import BaseContent, LLMFile
+from suvereo_common.utils import prioritize_items
+
+PRIORITIZED_MODELS = ["models/gemini-2.0-flash", "models/gemini-1.5-flash"]
+
+
+class GoogleAIImageContent(BaseContent):
+    image_url: str
+
+
+class GoogleAIFileContent(BaseContent):
+    mime_type: str
+    data: str
+
+
+class GoogleLanguageModelProvider(LanguageModelProvider):
+    NAME = "Google"
+    STREAMING_AVAILABLE = False
+
+    def provide_language_model(self, callbacks: list[BaseCallbackHandler] | None = None) -> BaseLanguageModel:
+        return ChatGoogleGenerativeAI(model=self._model, callbacks=callbacks)
+
+    def model_name(self) -> str:
+        return self._model
+
+    @staticmethod
+    def available_models() -> list[str]:
+        with genai.Client() as client:
+            all_models = client.models.list()
+
+        gemini_models = [model.name for model in all_models if "generateContent" in model.supported_actions]
+        return prioritize_items(gemini_models, PRIORITIZED_MODELS)
+
+    @staticmethod
+    def prepare_image_object(file_object: LLMFile) -> GoogleAIImageContent:
+        image_url = f"data:{file_object.content_type};base64,{file_object.content}"
+        return GoogleAIImageContent(
+            type="image_url",
+            image_url=image_url,
+        )
+
+    @staticmethod
+    def prepare_file_object(file_object: LLMFile) -> GoogleAIFileContent:
+        return GoogleAIFileContent(type="media", data=file_object.content, mime_type=file_object.content_type)