Improve codebase and add missing features

outline_agent.py

@@ -15,9 +15,16 @@ class OutlineAgent:
 
     def run(self, query: str) -> tuple[str, dict]:
         with get_openai_callback() as cb:
-            accountability = self._chain.invoke({"query": query}).content.strip()
+            outline = self._chain.invoke({"query": query}).content.strip()
 
             tokens = cb.total_tokens
             cost = cb.total_cost
 
-        return accountability, {"tokens": tokens, "cost": cost}
+        themes = []
+        keys = ["title", "query", "description"]
+
+        for theme in outline.split("\n\n")[1:]:
+            items = theme.strip().split("\n")
+            themes.append({keys[i]: item[item.index(":") + 1 :].strip().replace('"', "") for i, item in enumerate(items)})
+
+        return themes, {"tokens": tokens, "cost": cost}

prompts/outline_agent.py

@@ -3,18 +3,19 @@ In your role as an accomplished mastermind of outlining subjects, your task is t
 
 Here is an example structure to follow:
 
-Title  
-Theme 1  
-Web search query based on theme 1 and the subject context  
-Theme description  
+Title
 
-Theme 2  
-Web search query based on theme 2 and the subject context  
-Theme description  
+Theme 1
+Web search query based on theme 1 and the subject context
+Theme description
 
-Theme 3  
-Web search query based on theme 3 and the subject context  
-Theme description  
+Theme 2
+Web search query based on theme 2 and the subject context
+Theme description
+
+Theme 3
+Web search query based on theme 3 and the subject context
+Theme description
 
 ...