From 218b599c6663108913eed4db5fe5cc9e2c934389 Mon Sep 17 00:00:00 2001 From: Javier Martinez Date: Tue, 14 Jul 2026 16:44:28 +0200 Subject: [PATCH] fix: tests --- tests/server/chat/anthropic/test_anthropic_client.py | 5 ++--- tests/server/chat/anthropic/test_langchain_anthropic.py | 6 +++--- tests/server/chat/test_chat_knowledge_revamp.py | 6 +++--- tests/server/chat/test_chat_routes.py | 7 +++---- tests/server/chat/test_chat_routes_skills_integration.py | 2 +- 5 files changed, 12 insertions(+), 14 deletions(-) diff --git a/tests/server/chat/anthropic/test_anthropic_client.py b/tests/server/chat/anthropic/test_anthropic_client.py index d4c4406f..5e59a52b 100644 --- a/tests/server/chat/anthropic/test_anthropic_client.py +++ b/tests/server/chat/anthropic/test_anthropic_client.py @@ -119,9 +119,8 @@ def setup_mock_llm( ) llm_component = injector.get(LLMComponent) - llm_component.llm = mock_llm_instance - llm_component.get_llm.return_value = mock_llm_instance - injector.bind_mock(LLMComponent, mock_llm_instance) + llm_component.get_llm = Mock(return_value=mock_llm_instance) + injector.bind_mock(LLMComponent, llm_component) def create_mock_http_client( diff --git a/tests/server/chat/anthropic/test_langchain_anthropic.py b/tests/server/chat/anthropic/test_langchain_anthropic.py index fdffae7f..0117992a 100644 --- a/tests/server/chat/anthropic/test_langchain_anthropic.py +++ b/tests/server/chat/anthropic/test_langchain_anthropic.py @@ -1,6 +1,7 @@ import uuid from collections.abc import Iterator from typing import Any, get_args +from unittest.mock import Mock import anthropic import httpx @@ -139,9 +140,8 @@ def setup_mock_llm( ) llm_component = injector.get(LLMComponent) - llm_component.llm = mock_llm_instance - llm_component.get_llm.return_value = mock_llm_instance - injector.bind_mock(LLMComponent, mock_llm_instance) + llm_component.get_llm = Mock(return_value=mock_llm_instance) + injector.bind_mock(LLMComponent, llm_component) def create_mock_http_client( diff --git a/tests/server/chat/test_chat_knowledge_revamp.py b/tests/server/chat/test_chat_knowledge_revamp.py index aae0fe1d..5d90b099 100644 --- a/tests/server/chat/test_chat_knowledge_revamp.py +++ b/tests/server/chat/test_chat_knowledge_revamp.py @@ -1,5 +1,6 @@ import uuid from typing import Any +from unittest.mock import Mock import pytest from httpx import AsyncClient @@ -72,9 +73,8 @@ async def mock_llm_with_capture( mock_llm_instance.astream_chat_with_tools = coro llm_component = injector.get(LLMComponent) - llm_component.llm = mock_llm_instance - llm_component.get_llm.return_value = mock_llm_instance - injector.bind_mock(LLMComponent, mock_llm_instance) + llm_component.get_llm = Mock(return_value=mock_llm_instance) + injector.bind_mock(LLMComponent, llm_component) def create_tool_definition( diff --git a/tests/server/chat/test_chat_routes.py b/tests/server/chat/test_chat_routes.py index 3037dec9..a8e488c0 100644 --- a/tests/server/chat/test_chat_routes.py +++ b/tests/server/chat/test_chat_routes.py @@ -3,7 +3,7 @@ import json import uuid from collections.abc import AsyncGenerator from typing import Any -from unittest.mock import AsyncMock +from unittest.mock import AsyncMock, Mock import pytest from httpx import AsyncClient @@ -107,9 +107,8 @@ async def mock_llm( mock_llm = get_mock_function_calling_llm(deltas) llm_component = injector.get(LLMComponent) - llm_component.llm = mock_llm - llm_component.get_llm.return_value = mock_llm - injector.bind_mock(LLMComponent, mock_llm) + llm_component.get_llm = Mock(return_value=mock_llm) + injector.bind_mock(LLMComponent, llm_component) @pytest.mark.anyio diff --git a/tests/server/chat/test_chat_routes_skills_integration.py b/tests/server/chat/test_chat_routes_skills_integration.py index 6ba7acf6..4b98ad92 100644 --- a/tests/server/chat/test_chat_routes_skills_integration.py +++ b/tests/server/chat/test_chat_routes_skills_integration.py @@ -73,7 +73,7 @@ async def mock_llm_with_capture( mock_llm.astream_chat_with_tools = coro llm_component = injector.get(LLMComponent) llm_component.llm = mock_llm - llm_component.get_llm.return_value = mock_llm + llm_component.get_llm.side_effect = lambda *args, **kwargs: mock_llm injector.bind_mock(LLMComponent, mock_llm)