Facing issue while setting and getting session variables. about taskweaver HOT 5 CLOSED

INFO:httpx:HTTP Request: POST https://api.openai.com/v1/chat/completions "HTTP/1.1 200 OK"
INFO:taskweaver.logging:LLM output: {"response": [{"type": "init_plan", "content": "1. Fetch metadata of the datasource"}, {"type": "plan", "content": "1. Fetch metadata of the datasource"}, {"type": "current_plan_step", "content": "1. Fetch metadata of the datasource"}, {"type": "send_to", "content": "User"}, {"type": "message", "content": "The execution of fetching metadata failed due to a missing datasource ID. Please provide the correct datasource ID to proceed."}]}
INFO:taskweaver.logging:Planner talk to User: The execution of fetching metadata failed due to a missing datasource ID. Please provide the correct datasource ID to proceed.
INFO:metadata.consumers:AI response for message: Fetch the metadata of the datasource, response: {'id': 'round-20240421-113341-64dc3d28', 'user_query': 'Fetch the metadata of the datasource', 'state': 'finished', 'post_list': [{'id': 'post-20240421-113341-0c3a0f31', 'message': 'Fetch the metadata of the datasource', 'send_from': 'User', 'send_to': 'Planner', 'attachment_list': []}, {'id': 'post-20240421-113341-146f69b7', 'message': 'Please fetch the metadata of the datasource', 'send_from': 'Planner', 'send_to': 'CodeInterpreter', 'attachment_list': [{'id': 'atta-20240421-113343-cb5024fb', 'type': 'init_plan', 'content': '1. Fetch metadata of the datasource', 'extra': None}, {'id': 'atta-20240421-113343-ebed90f6', 'type': 'plan', 'content': '1. Fetch metadata of the datasource', 'extra': None}, {'id': 'atta-20240421-113344-f1b3a65c', 'type': 'current_plan_step', 'content': '1. Fetch metadata of the datasource', 'extra': None}, {'id': 'atta-20240421-113344-a334c861', 'type': 'board', 'content': 'I have drawn up a plan: \n1. Fetch metadata of the datasource\n\nPlease proceed with this step of this plan: Please fetch the metadata of the datasource', 'extra': None}]}, {'id': 'post-20240421-113344-fc6934ab', 'message': 'The following python code has been executed:\n```python\n# Fetch the metadata of the datasource\nmetadata, description = metadata_fetch()\nmetadata\n```\n\n\nThe execution of the generated python code above has failed\n\nDuring execution, the following messages were logged:\nTraceback (most recent call last):\n\n  Cell In[1], line 2\n    metadata, description = metadata_fetch()\n\n  File /tmp/tmpr5_yjs9f/metadata_fetch.py:27 in __call__\n\nValueError: No datasource ID found in session. stop the execution and ask user to update session properly\n\n<taskweaver.ces.runtime.context.ExecutorPluginContext object at 0x7f9f0d2b4a90>\n\n', 'send_from': 'CodeInterpreter', 'send_to': 'CodeInterpreter', 'attachment_list': [{'id': 'atta-20240421-113347-987930bb', 'type': 'python', 'content': '# Fetch the metadata of the datasource\nmetadata, description = metadata_fetch()\nmetadata', 'extra': None}, {'id': 'atta-20240421-113347-5c2a5777', 'type': 'verification', 'content': 'NONE', 'extra': None}, {'id': 'atta-20240421-113347-aeff8e03', 'type': 'code_error', 'content': 'No code verification is performed.', 'extra': None}, {'id': 'atta-20240421-113356-cef7f8f8', 'type': 'execution_status', 'content': 'FAILURE', 'extra': None}, {'id': 'atta-20240421-113356-de672ecb', 'type': 'execution_result', 'content': 'The execution of the generated python code above has failed\n\nDuring execution, the following messages were logged:\nTraceback (most recent call last):\n\n  Cell In[1], line 2\n    metadata, description = metadata_fetch()\n\n  File /tmp/tmpr5_yjs9f/metadata_fetch.py:27 in __call__\n\nValueError: No datasource ID found in session. stop the execution and ask user to update session properly\n\n<taskweaver.ces.runtime.context.ExecutorPluginContext object at 0x7f9f0d2b4a90>\n\n', 'extra': None}, {'id': 'atta-20240421-113356-d9732e08', 'type': 'artifact_paths', 'content': [], 'extra': None}, {'id': 'atta-20240421-113356-e2265e13', 'type': 'revise_message', 'content': 'The execution of the previous generated code has failed. If you think you can fix the problem by rewriting the code, please generate code and run it again.\nOtherwise, please explain the problem to me.', 'extra': None}]}, {'id': 'post-20240421-113356-3e022c28', 'message': "The following python code has been executed:\n```python\n# Set the datasource ID properly\n\n# Assuming the datasource ID is 123\nmetadata, description = metadata_fetch(datasource_id=123)\nmetadata\n```\n\n\nThe execution of the generated python code above has failed\n\nDuring execution, the following messages were logged:\nTraceback (most recent call last):\n\n  Cell In[2], line 4\n    metadata, description = metadata_fetch(datasource_id=123)\n\nTypeError: MetadataFetchPlugin.__call__() got an unexpected keyword argument 'datasource_id'\n\n", 'send_from': 'CodeInterpreter', 'send_to': 'CodeInterpreter', 'attachment_list': [{'id': 'atta-20240421-113359-a1fc8ed2', 'type': 'thought', 'content': 'ProgramApe needs to ensure that the datasource ID is properly set before fetching the metadata.', 'extra': None}, {'id': 'atta-20240421-113359-abd9804c', 'type': 'python', 'content': '# Set the datasource ID properly\n\n# Assuming the datasource ID is 123\nmetadata, description = metadata_fetch(datasource_id=123)\nmetadata', 'extra': None}, {'id': 'atta-20240421-113400-5d675af5', 'type': 'verification', 'content': 'NONE', 'extra': None}, {'id': 'atta-20240421-113400-564b4c6a', 'type': 'code_error', 'content': 'No code verification is performed.', 'extra': None}, {'id': 'atta-20240421-113401-3954b9c6', 'type': 'execution_status', 'content': 'FAILURE', 'extra': None}, {'id': 'atta-20240421-113401-2e20cdec', 'type': 'execution_result', 'content': "The execution of the generated python code above has failed\n\nDuring execution, the following messages were logged:\nTraceback (most recent call last):\n\n  Cell In[2], line 4\n    metadata, description = metadata_fetch(datasource_id=123)\n\nTypeError: MetadataFetchPlugin.__call__() got an unexpected keyword argument 'datasource_id'\n\n", 'extra': None}, {'id': 'atta-20240421-113401-b15460ff', 'type': 'artifact_paths', 'content': [], 'extra': None}, {'id': 'atta-20240421-113401-fdb486b3', 'type': 'revise_message', 'content': 'The execution of the previous generated code has failed. If you think you can fix the problem by rewriting the code, please generate code and run it again.\nOtherwise, please explain the problem to me.', 'extra': None}]}, {'id': 'post-20240421-113401-0079243a', 'message': 'The following python code has been executed:\n```python\n# Set the datasource ID properly\n\n# Assuming the datasource ID is 123\nmetadata = {}\ndescription = ""\nmetadata, description = metadata_fetch()\nmetadata\n```\n\n\nThe execution of the generated python code above has failed\n\nDuring execution, the following messages were logged:\nTraceback (most recent call last):\n\n  Cell In[3], line 6\n    metadata, description = metadata_fetch()\n\n  File /tmp/tmpr5_yjs9f/metadata_fetch.py:27 in __call__\n\nValueError: No datasource ID found in session. stop the execution and ask user to update session properly\n\n<taskweaver.ces.runtime.context.ExecutorPluginContext object at 0x7f9f0d2b4a90>\n\n', 'send_from': 'CodeInterpreter', 'send_to': 'CodeInterpreter', 'attachment_list': [{'id': 'atta-20240421-113404-7473d031', 'type': 'python', 'content': '# Set the datasource ID properly\n\n# Assuming the datasource ID is 123\nmetadata = {}\ndescription = ""\nmetadata, description = metadata_fetch()\nmetadata', 'extra': None}, {'id': 'atta-20240421-113406-446396e7', 'type': 'verification', 'content': 'NONE', 'extra': None}, {'id': 'atta-20240421-113406-c433b440', 'type': 'code_error', 'content': 'No code verification is performed.', 'extra': None}, {'id': 'atta-20240421-113407-fcc63d58', 'type': 'execution_status', 'content': 'FAILURE', 'extra': None}, {'id': 'atta-20240421-113407-0c801b17', 'type': 'execution_result', 'content': 'The execution of the generated python code above has failed\n\nDuring execution, the following messages were logged:\nTraceback (most recent call last):\n\n  Cell In[3], line 6\n    metadata, description = metadata_fetch()\n\n  File /tmp/tmpr5_yjs9f/metadata_fetch.py:27 in __call__\n\nValueError: No datasource ID found in session. stop the execution and ask user to update session properly\n\n<taskweaver.ces.runtime.context.ExecutorPluginContext object at 0x7f9f0d2b4a90>\n\n', 'extra': None}, {'id': 'atta-20240421-113407-950ed620', 'type': 'artifact_paths', 'content': [], 'extra': None}, {'id': 'atta-20240421-113407-7dffe03c', 'type': 'revise_message', 'content': 'The execution of the previous generated code has failed. If you think you can fix the problem by rewriting the code, please generate code and run it again.\nOtherwise, please explain the problem to me.', 'extra': None}]}, {'id': 'post-20240421-113407-3a95a544', 'message': 'The following python code has been executed:\n```python\n# Set the datasource ID properly\n\n# Assuming the datasource ID is 123\nmetadata = {}\ndescription = ""\nmetadata, description = metadata_fetch()\nmetadata\n```\n\n\nThe execution of the generated python code above has failed\n\nDuring execution, the following messages were logged:\nTraceback (most recent call last):\n\n  Cell In[4], line 6\n    metadata, description = metadata_fetch()\n\n  File /tmp/tmpr5_yjs9f/metadata_fetch.py:27 in __call__\n\nValueError: No datasource ID found in session. stop the execution and ask user to update session properly\n\n<taskweaver.ces.runtime.context.ExecutorPluginContext object at 0x7f9f0d2b4a90>\n\n', 'send_from': 'CodeInterpreter', 'send_to': 'Planner', 'attachment_list': [{'id': 'atta-20240421-113410-17d661b8', 'type': 'python', 'content': '# Set the datasource ID properly\n\n# Assuming the datasource ID is 123\nmetadata = {}\ndescription = ""\nmetadata, description = metadata_fetch()\nmetadata', 'extra': None}, {'id': 'atta-20240421-113411-b8c79a8d', 'type': 'verification', 'content': 'NONE', 'extra': None}, {'id': 'atta-20240421-113411-7f6c9f2a', 'type': 'code_error', 'content': 'No code verification is performed.', 'extra': None}, {'id': 'atta-20240421-113412-0cda3d00', 'type': 'execution_status', 'content': 'FAILURE', 'extra': None}, {'id': 'atta-20240421-113412-879dc071', 'type': 'execution_result', 'content': 'The execution of the generated python code above has failed\n\nDuring execution, the following messages were logged:\nTraceback (most recent call last):\n\n  Cell In[4], line 6\n    metadata, description = metadata_fetch()\n\n  File /tmp/tmpr5_yjs9f/metadata_fetch.py:27 in __call__\n\nValueError: No datasource ID found in session. stop the execution and ask user to update session properly\n\n<taskweaver.ces.runtime.context.ExecutorPluginContext object at 0x7f9f0d2b4a90>\n\n', 'extra': None}, {'id': 'atta-20240421-113412-9fd22251', 'type': 'artifact_paths', 'content': [], 'extra': None}]}, {'id': 'post-20240421-113412-dfa2b63b', 'message': 'The execution of fetching metadata failed due to a missing datasource ID. Please provide the correct datasource ID to proceed.', 'send_from': 'Planner', 'send_to': 'User', 'attachment_list': [{'id': 'atta-20240421-113413-4ff3eff8', 'type': 'init_plan', 'content': '1. Fetch metadata of the datasource', 'extra': None}, {'id': 'atta-20240421-113414-316410a1', 'type': 'plan', 'content': '1. Fetch metadata of the datasource', 'extra': None}, {'id': 'atta-20240421-113414-2945921c', 'type': 'current_plan_step', 'content': '1. Fetch metadata of the datasource', 'extra': None}, {'id': 'atta-20240421-113415-befdcd1c', 'type': 'board', 'content': 'I have drawn up a plan: \n1. Fetch metadata of the datasource\n\nPlease proceed with this step of this plan: The execution of fetching metadata failed due to a missing datasource ID. Please provide the correct datasource ID to proceed.', 'extra': None}]}]}
> /app/metadata/consumers.py(91)receive()
-> logger.info(f"Message processed and response sent for session_id={self.session_id}")
(Pdb) session_var
*** NameError: name 'session_var' is not defined
(Pdb) self.session
*** AttributeError: 'ChatAIConsumer' object has no attribute 'session'
(Pdb) self.ai_session
*** AttributeError: 'ChatAIConsumer' object has no attribute 'ai_session'
(Pdb) l
 86  	            message = text_data_json.get("message")
 87  	            session = user_sessions.get(self.session_id)
 88  	            if session and session.ai_client:
 89  	                # Use the session's AI client to handle the message and get a response
 90  	                ai_response = await self.handle_ai_response(message, session.ai_client)
 91  ->	                logger.info(f"Message processed and response sent for session_id={self.session_id}")
 92  	                await self.send(text_data=json.dumps({"message": ai_response}))
 93  	            else:
 94  	                # Session not found or message received before authentication
 95  	                await self.send(text_data=json.dumps({"error": "Unauthorized"}))
 96  	                logger.warning(f"Unauthorized access attempt or session not found for session_id={self.session_id}")
(Pdb) session.ai_client
<taskweaver.session.session.Session object at 0x7ebd24673f10>
(Pdb) session.ai_client.session_var
{'datasource_id': '34'}
(Pdb) 

def _get_session(
        self,
        session_id: str,
        session_dir: Optional[str] = None,
    ) -> EnvSession:
        if session_id not in self.session_dict:
            new_session = EnvSession(session_id)
            new_session.session_dir = (
                session_dir if session_dir is not None else self._get_default_session_dir(session_id)
            )
            os.makedirs(new_session.session_dir, exist_ok=True)
            self.session_dict[session_id] = new_session

        return self.session_dict.get(session_id, None)