fix: type hint for property last_prompt_id

pandasai/__init__.py

@@ -35,6 +35,7 @@
 """
 import warnings
 from typing import List, Optional, Union, Dict, Type
+import uuid
 import importlib.metadata
 
 import pandas as pd
@@ -236,7 +237,7 @@ def logs(self) -> List[dict[str, str]]:
         return [] if self._dl is None else self._dl.logs
 
     @property
-    def last_prompt_id(self) -> str:
+    def last_prompt_id(self) -> uuid.UUID:
         """Return the id of the last prompt that was run."""
         return None if self._dl is None else self._dl.last_prompt_id
 

pandasai/smart_dataframe/__init__.py

@@ -19,6 +19,7 @@
 """
 
 import hashlib
+import uuid
 from io import StringIO
 
 import pandas as pd
@@ -538,7 +539,7 @@ def last_prompt(self):
         return self.lake.last_prompt
 
     @property
-    def last_prompt_id(self) -> str:
+    def last_prompt_id(self) -> uuid.UUID:
         return self.lake.last_prompt_id
 
     @property

pandasai/smart_datalake/__init__.py

@@ -582,7 +582,7 @@ def last_prompt(self):
         return self._llm.last_prompt
 
     @property
-    def last_prompt_id(self) -> str:
+    def last_prompt_id(self) -> uuid.UUID:
         """Return the id of the last prompt that was run."""
         if self._last_prompt_id is None:
             raise ValueError("Pandas AI has not been run yet.")