[AUR-395, AUR-415] Adopt Example1 Injury pipeline; add .flow() for enabling bottom-up pipeline execution (#32)

* add example1/injury pipeline example
* add dotenv
* update various api

tests/test_post_processing.py

@@ -30,9 +30,11 @@ def test_is_batch(regex_extractor):
 
 def test_run_raw(regex_extractor):
     output = regex_extractor("This is a test. 123")
+    output = [each.text for each in output]
     assert output == ["123"]
 
 
 def test_run_batch_raw(regex_extractor):
     output = regex_extractor(["This is a test. 123", "456"])
+    output = [[each.text for each in batch] for batch in output]
     assert output == [["123"], ["456"]]

tests/test_prompt.py

@@ -2,7 +2,7 @@ import pytest
 
 from kotaemon.documents.base import Document
 from kotaemon.post_processing.extractor import RegexExtractor
-from kotaemon.prompt.base import BasePrompt
+from kotaemon.prompt.base import BasePromptComponent
 from kotaemon.prompt.template import PromptTemplate
 
 
@@ -14,7 +14,7 @@ def test_set_attributes():
     )
     comp.set_run(kwargs={"text": "This is a test. 1 2 3"}, temp=True)
 
-    prompt = BasePrompt(template=template, s="Alice", i=30, doc=doc, comp=comp)
+    prompt = BasePromptComponent(template=template, s="Alice", i=30, doc=doc, comp=comp)
     assert prompt.s == "Alice"
     assert prompt.i == 30
     assert prompt.doc == doc
@@ -23,23 +23,23 @@ def test_set_attributes():
 
 def test_check_redundant_kwargs():
     template = PromptTemplate("Hello, {name}!")
-    prompt = BasePrompt(template, name="Alice")
-    with pytest.raises(ValueError):
-        prompt._BasePrompt__check_redundant_kwargs(name="Alice", age=30)
+    prompt = BasePromptComponent(template, name="Alice")
+    with pytest.warns(UserWarning, match="Keys provided but not in template: {'age'}"):
+        prompt._BasePromptComponent__check_redundant_kwargs(name="Alice", age=30)
 
 
 def test_check_unset_placeholders():
     template = PromptTemplate("Hello, {name}! I'm {age} years old.")
-    prompt = BasePrompt(template, name="Alice")
+    prompt = BasePromptComponent(template, name="Alice")
     with pytest.raises(ValueError):
-        prompt._BasePrompt__check_unset_placeholders()
+        prompt._BasePromptComponent__check_unset_placeholders()
 
 
 def test_validate_value_type():
     template = PromptTemplate("Hello, {name}!")
-    prompt = BasePrompt(template)
+    prompt = BasePromptComponent(template)
     with pytest.raises(ValueError):
-        prompt._BasePrompt__validate_value_type(name={})
+        prompt._BasePromptComponent__validate_value_type(name={})
 
 
 def test_run():
@@ -50,18 +50,18 @@ def test_run():
     )
     comp.set_run(kwargs={"text": "This is a test. 1 2 3"}, temp=True)
 
-    prompt = BasePrompt(template=template, s="Alice", i=30, doc=doc, comp=comp)
+    prompt = BasePromptComponent(template=template, s="Alice", i=30, doc=doc, comp=comp)
 
     result = prompt()
 
     assert (
-        result
+        result.text
         == "str = Alice, int = 30, doc = Helloo, Alice!, comp = ['One', 'Two', 'Three']"
     )
 
 
 def test_set_method():
     template = PromptTemplate("Hello, {name}!")
-    prompt = BasePrompt(template)
+    prompt = BasePromptComponent(template)
     prompt.set(name="Alice")
     assert prompt.name == "Alice"