Public/kotaemon
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source 
* add example1/injury pipeline example
* add dotenv
* update various api
This commit is contained in:
ian_Cin
2023-10-02 16:24:56 +07:00
committed by GitHub
parent 3cceec63ef
commit d83c22aa4e
16 changed files with 114 additions and 69 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
tests/test_prompt.py
View File

@@ -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"