From 506669af3fc44869c5ab819e2d3b43a51b3c88a8 Mon Sep 17 00:00:00 2001 From: Martin Bielik Date: Tue, 17 Dec 2024 23:13:01 +0100 Subject: special role --- tests/context_test.py | 6 +++++- tests/resources/roles.ini | 3 +++ 2 files changed, 8 insertions(+), 1 deletion(-) (limited to 'tests') diff --git a/tests/context_test.py b/tests/context_test.py index 1cc0330..a1d9624 100644 --- a/tests/context_test.py +++ b/tests/context_test.py @@ -32,8 +32,12 @@ def test_default_config(): 'user_selection': 'Hello world!', 'command_type': 'chat', }) + expected_options = { + **default_config['options'], + 'token_file_path': '/custom/path/ai.token', + } expected_context = { - 'config': default_config, + 'config': { **default_config, 'options': expected_options }, 'prompt': 'translate to Slovak:\nHello world!', } assert expected_context == actual_context diff --git a/tests/resources/roles.ini b/tests/resources/roles.ini index b677462..00903f7 100644 --- a/tests/resources/roles.ini +++ b/tests/resources/roles.ini @@ -1,3 +1,6 @@ +[default] +options.token_file_path = /custom/path/ai.token + [test-role-simple] prompt = simple role prompt options.model = o1-preview -- cgit v1.2.3