Skip to content

Commit ee13268

Browse files
committed
fix: test
1 parent baba621 commit ee13268

1 file changed

Lines changed: 12 additions & 1 deletion

File tree

libs/core/kiln_ai/adapters/model_adapters/test_litellm_adapter.py

Lines changed: 12 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1307,7 +1307,18 @@ async def test_dict_input_converted_to_json(tmp_path, config):
13071307

13081308

13091309
@pytest.mark.asyncio
1310-
async def test_run_with_prior_trace_uses_multiturn_formatter(config, mock_task):
1310+
async def test_run_with_prior_trace_uses_multiturn_formatter(mock_task):
1311+
config = LiteLlmConfig(
1312+
base_url="https://api.test.com",
1313+
run_config_properties=KilnAgentRunConfigProperties(
1314+
model_name="test-model",
1315+
model_provider_name="openai_compatible",
1316+
prompt_id="simple_prompt_builder",
1317+
structured_output_mode="json_schema",
1318+
),
1319+
default_headers={"X-Test": "test"},
1320+
additional_body_options={"api_key": "test_key"},
1321+
)
13111322
prior_trace = [
13121323
{"role": "user", "content": "hi"},
13131324
{"role": "assistant", "content": "hello"},

0 commit comments

Comments
 (0)