Check if a dedicated session exists with a different memory configuration

Author: DarthMax

graphdatascience/session/dedicated_sessions.py

@@ -52,6 +52,7 @@ def get_or_create(
         # TODO configure session size (and check existing_session has same size)
         if existing_session:
             self._check_expiry_date(existing_session)
+            self._check_memory_configuration(existing_session, memory)
             session_id = existing_session.id
         else:
             create_details = self._create_session(session_name, dbid, db_connection.uri, password, memory)
@@ -118,7 +119,7 @@ def _create_session(
             name=session_name,
             dbid=dbid,
             pwd=pwd,
-            memory=memory.value,
+            memory=memory,
         )
         return create_details
 
@@ -139,6 +140,13 @@ def _check_expiry_date(self, session: SessionDetails) -> None:
             if until_expiry < timedelta(days=1):
                 raise Warning(f"Session `{session.name}` is expiring in less than a day.")
 
+    def _check_memory_configuration(self, existing_session: SessionDetails, requested_memory: SessionMemory) -> None:
+        if existing_session.memory != requested_memory:
+            raise RuntimeError(
+                f"Session `{existing_session.name}` exists with a different memory configuration. "
+                f"Current: {existing_session.memory.value}, Requested: {requested_memory.value}."
+            )
+
     @classmethod
     def _fail_ambiguous_session(cls, session_name: str, sessions: List[SessionDetails]) -> None:
         candidates = [i.id for i in sessions]