fix: Simplify constructor code in ToolboxClient.

googleapis · Nov 12, 2024 · f0cf9ec · f0cf9ec
1 parent 0dc6548
commit f0cf9ec
Showing 1 changed file with 2 additions and 7 deletions.
diff --git a/sdks/langchain/src/toolbox_langchain_sdk/client.py b/sdks/langchain/src/toolbox_langchain_sdk/client.py
@@ -19,13 +19,8 @@ def __init__(self, url: str, session: Optional[ClientSession] = None):
                 Default: None
         """
         self._url: str = url
-
-        if session:
-            self._should_close_session: bool = False
-            self._session: ClientSession = session
-        else:
-            self._should_close_session: bool = True
-            self._session: ClientSession = ClientSession()
+        self._should_close_session: bool = session != None
+        self._session: ClientSession = session or ClientSession()
 
     async def close(self) -> None:
         """