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

Change template to private attribute and simplify imports (#101)

Browse Source 
---------

Co-authored-by: ian <ian@cinnamon.is>
This commit is contained in:
Duc Nguyen (john)
2023-12-08 18:10:34 +07:00
committed by GitHub
parent 1f927d3391
commit da0ac1d69f
13 changed files with 31 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tests/test_prompt.py
View File

@@ -22,21 +22,21 @@ def test_set_attributes():
def test_check_redundant_kwargs():
template = PromptTemplate("Hello, {name}!")
prompt = BasePromptComponent(template, name="Alice")
prompt = BasePromptComponent(template=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 = BasePromptComponent(template, name="Alice")
prompt = BasePromptComponent(template=template, name="Alice")
with pytest.raises(ValueError):
prompt._BasePromptComponent__check_unset_placeholders()
def test_validate_value_type():
template = PromptTemplate("Hello, {name}!")
prompt = BasePromptComponent(template)
prompt = BasePromptComponent(template=template)
with pytest.raises(ValueError):
prompt._BasePromptComponent__validate_value_type(name={})
@@ -58,6 +58,6 @@ def test_run():
def test_set_method():
template = PromptTemplate("Hello, {name}!")
prompt = BasePromptComponent(template)
prompt = BasePromptComponent(template=template)
prompt.set(name="Alice")
assert prompt.name == "Alice"