Edit tests to adjust to changes + web agent validation to false

WaitThatShouldntWork · Sep 5, 2024 · 9693524 · 9693524
1 parent d2d15fb
commit 9693524
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/backend/src/agents/web_agent.py b/backend/src/agents/web_agent.py
@@ -35,7 +35,7 @@ async def web_general_search_core(search_query, llm, model) -> str:
             if await is_valid_answer(summary, search_query):
                 response = {
                     "content": summary,
-                    "ignore_validation": "true"
+                    "ignore_validation": "false"
                 }
                 return json.dumps(response, indent=4)
         return "No relevant information found on the internet for the given query."

diff --git a/backend/tests/agents/datastore_agent_test.py b/backend/tests/agents/datastore_agent_test.py
@@ -34,7 +34,7 @@ async def test_generate_query_success(mock_load_prompt, mock_publish_log_info,
     result = await generate_cypher_query_core(question_intent, operation, question_params, aggregation, sort_order,
                                               timeframe, llm, model)
 
-    assert result == "Mocked response from the database"
+    assert result == '{\n    "content": "Mocked response from the database",\n    "ignore_validation": "false"\n}'
     mock_load_prompt.assert_called()
     llm.chat.assert_called_once_with(
         model,