diff --git a/pyclient/pydeephaven/_console_service.py b/pyclient/pydeephaven/_console_service.py index 42822d9aa92..a67a61ae758 100644 --- a/pyclient/pydeephaven/_console_service.py +++ b/pyclient/pydeephaven/_console_service.py @@ -20,7 +20,7 @@ def start_console(self): try: result_id = self.session.make_ticket() response = self._grpc_console_stub.StartConsole( - console_pb2.StartConsoleRequest(result_id=result_id, session_type='python'), + console_pb2.StartConsoleRequest(result_id=result_id, session_type=self.session._session_type), metadata=self.session.grpc_metadata) self.console_id = response.result_id except Exception as e: diff --git a/pyclient/pydeephaven/session.py b/pyclient/pydeephaven/session.py index 12429ab4e17..753dcdd3422 100644 --- a/pyclient/pydeephaven/session.py +++ b/pyclient/pydeephaven/session.py @@ -33,13 +33,14 @@ class Session: is_alive (bool): check if the session is still alive (may refresh the session) """ - def __init__(self, host: str = None, port: int = None, never_timeout: bool = True): + def __init__(self, host: str = None, port: int = None, never_timeout: bool = True, session_type: str = 'python'): """ Initialize a Session object that connects to the Deephaven server Args: host (str): the host name or IP address of the remote machine, default is 'localhost' port (int): the port number that Deephaven server is listening on, default is 10000 never_timeout (bool, optional): never allow the session to timeout, default is True + session_type (str, optional): the Deephaven session type. Defaults to 'python' Raises: DHError @@ -67,6 +68,7 @@ def __init__(self, host: str = None, port: int = None, never_timeout: bool = Tru self._tables = {} self._never_timeout = never_timeout self._keep_alive_timer = None + self._session_type = session_type self._connect()