From 5fd1be143b6cb48e93d6e9127f4673bd6070e25e 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 d99f91c8..1d2f6401 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