implement welcome.parse_abstracts()

mind_palace/welcome.py

@@ -1,2 +1,22 @@
+from typing import List
+
+from docs import Section
+
+
+def parse_abstracts(nodes) -> List[str]:
+    """Returns a list of abstract texts from a list of nodes."""
+    return [
+        node.get_text()
+        for node in nodes
+        if node.metadata["section"] == str(Section.ABSTRACT)
+    ]
+
+
+def summarize(texts):
+    # TODO:
+    return
+
+
 def get_welcome_message(nodes):
+    # TODO:
     return "Ask me a question about these PDFs"

tests/context.py

@@ -5,7 +5,8 @@
     0, os.path.abspath(os.path.join(os.path.dirname(__file__), "../mind_palace/"))
 )
 
-import extract
 import docs
+import extract
 import index
 import measure
+import welcome

tests/unit/test_welcome.py

@@ -0,0 +1,17 @@
+from tests.context import docs
+from tests.context import welcome as w
+
+
+def test_parse_abstracts():
+    nodes = [
+        docs.create_text_node(
+            text="this is abstract", node_id=1, section=docs.Section.ABSTRACT
+        ),
+        docs.create_text_node(
+            text="this is not abstract", node_id=2, section=docs.Section.BODY
+        ),
+    ]
+    abstracts = w.parse_abstracts(nodes)
+    assert len(abstracts) == 1
+    assert isinstance(abstracts, list)
+    assert all(isinstance(abstract, str) for abstract in abstracts)