fix(data profile): add try-exception in profile

Author: StCarmen

alias/src/alias/agent/agents/data_source/_data_profiler_factory.py

@@ -49,11 +49,16 @@ def generate_profile(self) -> Dict[str, Any]:
         Returns:
             Dictionary containing the complete data profile
         """
-        self.data = self._read_data()
-        content = self._generate_content(self.prompt, self.data)
-        # content = self.prompt.format(data=self.data)
-        res = self._call_model(content)
-        self.profile = self._wrap_data_response(res)
+        try:
+            self.profile = self.generate_profile()
+            self.data = self._read_data()
+            content = self._generate_content(self.prompt, self.data)
+            # content = self.prompt.format(data=self.data)
+            res = self._call_model(content)
+            self.profile = self._wrap_data_response(res)
+        except Exception as e:
+            print(f"Error generating profile: {e}")
+            self.profile = {}
         return self.profile
 
     @staticmethod
@@ -212,6 +217,7 @@ def _call_model(
                     messages=messages,
                     api_key=self.api_key,
                 )
+            # TODO: handle failed call of model
             response = response.output["choices"][0]["message"]["content"]
             # Clean and parse the JSON response from the LLM
             response = (
@@ -223,10 +229,8 @@ def _call_model(
             )
             return response
 
-        except Exception:
-            import traceback
-
-            print(traceback.format_exc())
+        except Exception as e:
+            print(f"Error calling the model {self.model} with {e}")
             # Consider returning None or an empty dict on failure
             return {}
 

alias/src/alias/agent/agents/data_source/data_source.py

@@ -58,8 +58,10 @@ def __init__(
         self.endpoint = endpoint
         self.name = name
 
-        source_access_type = SOURCE_TYPE_TO_ACCESS_TYPE.get(source_type, \
-            SourceAccessType.DIRECT)
+        source_access_type = SOURCE_TYPE_TO_ACCESS_TYPE.get(
+            source_type,
+            SourceAccessType.DIRECT,
+        )
 
         self.source_access_type = source_access_type
         self.source_type = source_type