klas/t6_mem0
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Docs: use LlmConfig instead of QueryConfig (#626)

Browse Source
This commit is contained in:
cachho
2023-09-24 20:48:03 +02:00
committed by GitHub
parent 1db3e43adf
commit 6c71a1020d
4 changed files with 14 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/llm/test_generate_prompt.py
View File

@@ -12,7 +12,7 @@ class TestGeneratePrompt(unittest.TestCase):
def test_generate_prompt_with_template(self):
"""
Tests that the generate_prompt method correctly formats the prompt using
a custom template provided in the QueryConfig instance.
a custom template provided in the LlmConfig instance.
This test sets up a scenario with an input query and a list of contexts,
and a custom template, and then calls generate_prompt. It checks that the
@@ -58,7 +58,7 @@ class TestGeneratePrompt(unittest.TestCase):
def test_generate_prompt_with_history(self):
"""
Test the 'generate_prompt' method with QueryConfig containing a history attribute.
Test the 'generate_prompt' method with LlmConfig containing a history attribute.
"""
config = BaseLlmConfig()
config.template = Template("Context: $context | Query: $query | History: $history")