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

[docs]: Revamp embedchain docs (#799)

Browse Source
This commit is contained in:
Deshraj Yadav
2023-10-13 15:38:15 -07:00
committed by GitHub
parent a86d7f52e9
commit 4a8c50f886
68 changed files with 1175 additions and 673 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 LlmConfig instance.
a custom template provided in the BaseLlmConfig 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 LlmConfig containing a history attribute.
Test the 'generate_prompt' method with BaseLlmConfig containing a history attribute.
"""
config = BaseLlmConfig()
config.template = Template("Context: $context | Query: $query | History: $history")