From afcf1e0494b4dae694db5c90d9cdd2142ab5e5f1 Mon Sep 17 00:00:00 2001 From: EugeneP Date: Thu, 30 Jan 2025 18:39:38 +0100 Subject: [PATCH] TLK-2709 - Github Tool Implementation review fixes --- src/backend/tests/unit/tools/test_lang_chain.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/backend/tests/unit/tools/test_lang_chain.py b/src/backend/tests/unit/tools/test_lang_chain.py index dc00bac23a..31ede1ce27 100644 --- a/src/backend/tests/unit/tools/test_lang_chain.py +++ b/src/backend/tests/unit/tools/test_lang_chain.py @@ -72,7 +72,7 @@ async def test_wiki_retriever_no_docs() -> None: ): result = await retriever.call({"query": query}, ctx) - assert result == ToolError(type=ToolErrorCode.OTHER, success=False, text='No results found.', details='No results found for the given params.') + assert result == [ToolError(type=ToolErrorCode.OTHER, success=False, text='No results found.', details='No results found for the given params.').model_dump()] @@ -156,4 +156,4 @@ async def test_vector_db_retriever_no_docs() -> None: mock_db.as_retriever().get_relevant_documents.return_value = mock_docs result = await retriever.call({"query": query}, ctx) - assert result == ToolError(type=ToolErrorCode.OTHER, success=False, text='No results found.', details='No results found for the given params.') + assert result == [ToolError(type=ToolErrorCode.OTHER, success=False, text='No results found.', details='No results found for the given params.').model_dump()]