better error handling

src/rag.py

@@ -85,22 +85,24 @@ class SlowestFunctionPromptFormatter(PerfGuruPromptFormatter):
                 if error_fn:
                     error_fn("Profile type must be provided if a profile file is provided.")
                 return None
-            profile = self._read_profile(profile_path, profile_type)
-            slowest = profile.gf.dataframe.nlargest(self.k, 'time')
-            function_names = [slowest['name'].values[i] for i in range(self.k) if i < len(slowest['name'].values)]
-            execution_times = [slowest['time'].values[i] for i in range(self.k) if i < len(slowest['name'].values)]
-            # print(profile_content)
-            hot_path = profile.gf.hot_path()
-            hot_path_functions = []
-
-            for node in hot_path:
-                if "name" in node.frame.attrs: 
-                    hot_path_functions.append(node.frame["name"])
-            hot_path_functions = hot_path_functions[:self.k]
-
-            profile_content = (f"The slowest functions are {function_names} and they took {execution_times} seconds, respectively." + 
-            f" Also, these functions were in the hot path: {hot_path_functions}.")
-            print(profile_content)
+            
+            try:
+                profile = self._read_profile(profile_path, profile_type)
+                slowest = profile.gf.dataframe.nlargest(self.k, 'time')
+                function_names = [slowest['name'].values[i] for i in range(self.k) if i < len(slowest['name'].values)]
+                execution_times = [slowest['time'].values[i] for i in range(self.k) if i < len(slowest['name'].values)]
+                hot_path = profile.gf.hot_path()
+                hot_path_functions = []
+
+                for node in hot_path:
+                    if "name" in node.frame.attrs: 
+                        hot_path_functions.append(node.frame["name"])
+                hot_path_functions = hot_path_functions[:self.k]
+
+                profile_content = (f"The slowest functions are {function_names} and they took {execution_times} seconds, respectively." + 
+                f" Also, these functions were in the hot path: {hot_path_functions}.")
+            except:
+                profile_content = ""
 
         else:
             profile_content = ""
@@ -127,28 +129,30 @@ class SlowestFunctionParsedPromptFormatter(PerfGuruPromptFormatter):
                     error_fn("Profile type must be provided if a profile file is provided.")
                 return None
             
-            k = 1
-            profile = self._read_profile(profile_path, profile_type)
-            slowest = profile.gf.dataframe.nlargest(k, 'time')
-            function_name = slowest['name'].values[0] if len(slowest['name'].values) > 0 else None
-            line_number = slowest['line'].values[0] if len(slowest['line'].values) > 0 else None
-            code = None
-            
-            if line_number:
-                filename = ""
-                code_file_contents = self._read_code_files(code_paths)
-                for code_path, content in code_file_contents.items():
-                    filename = basename(code_path)
-                    code, _ = get_function_at_line(filename, str(line_number))
-                    if code:
-                        break
+            try:
+                k = 1
+                profile = self._read_profile(profile_path, profile_type)
+                slowest = profile.gf.dataframe.nlargest(k, 'time')
+                function_name = slowest['name'].values[0] if len(slowest['name'].values) > 0 else None
+                line_number = slowest['line'].values[0] if len(slowest['line'].values) > 0 else None
+                code = None
                 
-                if code:
-                    concatenated_code = f"{fname}:\n{code}\n\n"
-                    print("Only function code:", concatenated_code)
+                if line_number:
+                    filename = ""
+                    code_file_contents = self._read_code_files(code_paths)
+                    for code_path, content in code_file_contents.items():
+                        filename = basename(code_path)
+                        code, _ = get_function_at_line(filename, str(line_number))
+                        if code:
+                            break
+                    
+                    if code:
+                        concatenated_code = f"{fname}:\n{code}\n\n"
+                        print("Only function code:", concatenated_code)
 
-                profile_content = (f"The slowest function is {function_name}.")
-                print(profile_content)
+                    profile_content = (f"The slowest function is {function_name}.")
+            except:
+                profile_content = ""
         
         if concatenated_code == "":
             code_file_contents = self._read_code_files(code_paths)