Fix issue with class folder path for collab notebook

VRSEN · Apr 29, 2024 · 8505647 · 8505647
1 parent 4ebf8a8
commit 8505647
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/agency_swarm/agents/agent.py b/agency_swarm/agents/agent.py
@@ -620,7 +620,10 @@ def get_class_folder_path(self):
             return os.path.abspath(os.path.dirname(self.__module__.__file__))
         except AttributeError:
             # If that fails, fall back to inspect
-            class_file = inspect.getfile(self.__class__)
+            try:
+                class_file = inspect.getfile(self.__class__)
+            except TypeError or OSError or AttributeError:
+                return "./"
             return os.path.abspath(os.path.realpath(os.path.dirname(class_file)))
 
     def add_shared_instructions(self, instructions: str):