From 01ddb7db16e4b6e34a829ba0c383fb2ac686f678 Mon Sep 17 00:00:00 2001 From: DeepakAkkil Date: Fri, 6 Sep 2024 18:37:45 +0300 Subject: [PATCH] bug fix --- test/tests_processor.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/test/tests_processor.py b/test/tests_processor.py index 64b13b9..c27e10f 100644 --- a/test/tests_processor.py +++ b/test/tests_processor.py @@ -2,6 +2,7 @@ import json import os import time +from ae.core.agents_llm_config import AgentsLLMConfig from test.evaluators import evaluator_router from test.test_utils import get_formatted_current_timestamp from test.test_utils import load_config @@ -323,8 +324,9 @@ async def run_tests(ag: AutogenWrapper, browser_manager: PlaywrightManager, min_ results_dir = create_results_dir(test_file, test_results_id) test_results: list[dict[str, str | int | float | None]] = [] + llm_config = AgentsLLMConfig() if not ag: - ag = await AutogenWrapper.create() + ag = await AutogenWrapper.create(llm_config.get_planner_agent_config(), llm_config.get_browser_nav_agent_config()) if not browser_manager: browser_manager = browserManager.PlaywrightManager(headless=False)