Update main.py

main.py

@@ -50,14 +50,15 @@ def execute_python(code: str, config: RunnableConfig):
         code: Valid Python code with correct indentation and syntax including necessary imports.
     """
 
-    #print(config)
+    thread_config = config.get("configurable", {})
+    session_token = thread_config.get("thread_id", "test")
 
     headers = {
         'accept': 'application/json',
         'Content-Type': 'application/json'
     }
     data = {
-        "session_token": config.configurable.get("thread_id", "test"),
+        "session_token": session_token,
         "code": code
     }
     response = requests.post(