Add creds to the constructors

GoogleCloudPlatform · Jul 18, 2024 · 6f4c1a1 · 6f4c1a1
1 parent 1242ae0
commit 6f4c1a1
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/src/dfcx_scrapi/core/sessions.py b/src/dfcx_scrapi/core/sessions.py
@@ -41,13 +41,15 @@ def __init__(
         self,
         creds_path: str = None,
         creds_dict: Dict = None,
+        creds=None,
         scope=False,
         agent_id: str = None,
         session_id: str = None,
         tools_map: Dict[str, str] = None,
     ):
         super().__init__(
-            creds_path=creds_path, creds_dict=creds_dict, scope=scope
+            creds_path=creds_path, creds_dict=creds_dict,
+            creds=creds, scope=scope
         )
 
         self.session_id = session_id

diff --git a/src/dfcx_scrapi/core/tools.py b/src/dfcx_scrapi/core/tools.py
@@ -28,12 +28,14 @@ def __init__(
         self,
         creds_path: str = None,
         creds_dict: Dict = None,
+        creds=None,
         scope=False,
         agent_id: str = None,
         tool_id: str = None
     ):
         super().__init__(
-            creds_path=creds_path, creds_dict=creds_dict, scope=scope
+            creds_path=creds_path, creds_dict=creds_dict,
+            creds=creds, scope=scope
         )
 
         self.agent_id = agent_id