#16 update example to OOP Framework

business-science · Jan 9, 2025 · bbaf682 · bbaf682
1 parent 0c6b7dd
commit bbaf682
Show file tree

Hide file tree

Showing 2 changed files with 362 additions and 183 deletions.
diff --git a/ai_data_science_team/agents/data_cleaning_agent.py b/ai_data_science_team/agents/data_cleaning_agent.py
@@ -103,6 +103,10 @@ class DataCleaningAgent(BaseAgent):
         Retrieves the generated Python function used for cleaning the data.
     get_recommended_cleaning_steps()
         Retrieves the agent's recommended cleaning steps.
+    get_response()
+        Returns the response from the agent as a dictionary.
+    show()
+        Displays the agent's mermaid diagram.
 
     Examples
     --------