From 6f4c1a19b515416e0ddf09b2cb9d9f40445ad83e Mon Sep 17 00:00:00 2001
From: Milad <63479762+MRyderOC@users.noreply.github.com>
Date: Thu, 11 Jul 2024 15:09:05 -0500
Subject: [PATCH] Add creds to the constructors

---
 src/dfcx_scrapi/core/sessions.py | 4 +++-
 src/dfcx_scrapi/core/tools.py    | 4 +++-
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/src/dfcx_scrapi/core/sessions.py b/src/dfcx_scrapi/core/sessions.py
index a36ac55f..c9146f8f 100644
--- 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
index b24e26d6..78b122f6 100644
--- 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