hyperforge-google 1.0.0.post38__tar.gz → 1.0.0.post49__tar.gz

{hyperforge_google-1.0.0.post38 → hyperforge_google-1.0.0.post49}/PKG-INFO

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

{hyperforge_google-1.0.0.post38 → hyperforge_google-1.0.0.post49}/pyproject.toml

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

{hyperforge_google-1.0.0.post38 → hyperforge_google-1.0.0.post49}/src/hyperforge_google/driver.py

@@ -1,3 +1,6 @@
+import json
+
+from google.auth import load_credentials_from_dict  # type: ignore
 from google.genai import Client
 from hyperforge.configure import driver
 from hyperforge.driver import Driver
@@ -17,17 +20,25 @@ 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)
+        credentials = None
+        location = None
+        if driver.config.vertexai:
+            if driver.config.credentials:
+                info = json.loads(driver.config.credentials)
+                credentials, project = load_credentials_from_dict(
+                    info,
+                    scopes=["https://www.googleapis.com/auth/cloud-platform"],
+                )
+        else:
+            project = driver.config.location
+            location = driver.config.location
         client = Client(
             vertexai=driver.config.vertexai,
-            credentials=creds,
+            credentials=credentials,
             api_key=driver.config.api_key,
-            project=driver.config.project,
-            location=driver.config.location,
+            project=project,
+            location=location,
         )
         return cls(
             client=client,

{hyperforge_google-1.0.0.post38 → hyperforge_google-1.0.0.post49}/src/hyperforge_google.egg-info/PKG-INFO

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