hyperforge-google 1.0.0.post35__tar.gz → 1.0.0.post45__tar.gz

{hyperforge_google-1.0.0.post35 → hyperforge_google-1.0.0.post45}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hyperforge_google
-Version: 1.0.0.post35
+Version: 1.0.0.post45
 Summary: Google Search Hyperforge agent
 Author-email: Nuclia <nucliadb@nuclia.com>
 License-Expression: Apache-2.0

{hyperforge_google-1.0.0.post35 → hyperforge_google-1.0.0.post45}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "hyperforge_google"
-version = "1.0.0.post35"
+version = "1.0.0.post45"
 license = "Apache-2.0"
 description = "Google Search Hyperforge agent"
 authors = [{ name = "Nuclia", email = "nucliadb@nuclia.com" }]

{hyperforge_google-1.0.0.post35 → hyperforge_google-1.0.0.post45}/src/hyperforge_google/driver.py

@@ -1,4 +1,7 @@
+import json
+
 from google.genai import Client
+from google.oauth2 import service_account
 from hyperforge.configure import driver
 from hyperforge.driver import Driver
 from typing_extensions import Self
@@ -8,8 +11,8 @@ from hyperforge_google.config import GoogleDriverConfig
 
 @driver(
     id="google",
-    title="Google Driver",
-    description="Driver for interacting with the Google API.",
+    title="Google Source",
+    description="Source for interacting with the Google API.",
     config_schema=GoogleDriverConfig,
 )
 class GoogleDriver(Driver):
@@ -17,11 +20,15 @@ class GoogleDriver(Driver):
 
     @classmethod
     async def init(cls, driver: GoogleDriverConfig) -> Self:
+
         creds = None
         if driver.config.vertexai:
-            from google.auth import load_credentials_from_file
-
-            creds, _ = load_credentials_from_file(driver.config.credentials)
+            if driver.config.credentials:
+                info = json.loads(driver.config.credentials)
+                creds = service_account.Credentials.from_service_account_info(
+                    info,
+                    scopes=["https://www.googleapis.com/auth/cloud-platform"],
+                )
         client = Client(
             vertexai=driver.config.vertexai,
             credentials=creds,

{hyperforge_google-1.0.0.post35 → hyperforge_google-1.0.0.post45}/src/hyperforge_google.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hyperforge_google
-Version: 1.0.0.post35
+Version: 1.0.0.post45
 Summary: Google Search Hyperforge agent
 Author-email: Nuclia <nucliadb@nuclia.com>
 License-Expression: Apache-2.0