learningcircuit-local-deep-research-2517-2502

CONTRIBUTING.md
Dockerfile
LICENSE
MANIFEST.in
README.md
SECURITY.md
bearer.yml
community_benchmark_results
cookiecutter-docker
docker-compose.gpu.override.yml
docker-compose.unraid.yml
docker-compose.yml

docs
eslint.config.js
eslint.config.mjs
examples
package-lock.json
package.json
pdm.lock
pyproject.toml
scripts
src
tests
unraid-templates
vite.config.js
vulture_whitelist.py

"""
Factory for creating search strategies.
This module provides a centralized way to create search strategies
to avoid code duplication.
"""

from loguru import logger
from typing import Optional, Dict, Any, List
from langchain_core.language_models import BaseChatModel


def _get_setting(

    settings_snapshot: Optional[Dict], key: str, default: Any
) -> Any:
    """Get a setting value from the snapshot, handling nested dict structure."""
    if not settings_snapshot or key not in settings_snapshot:
        return default
    value = settings_snapshot[key]
    # Extract value from dict structure if needed
    if isinstance(value, dict) and "value" in value:
        return value["value"]
    return value


def create_strategy(
    strategy_name: str,
    model: BaseChatModel,
    search: Any,
    all_links_of_system: Optional[List[Dict]] = None,
    settings_snapshot: Optional[Dict] =
…[8630 chars omitted]…
  )

        return SourceBasedSearchStrategy(
            model=model,
            search=search,
            include_text_content=True,
            use_cross_engine_filter=True,
            all_links_of_system=all_links_of_system,
            use_atomic_facts=False,
            settings_snapshot=settings_snapshot,
            search_original_query=kwargs.get("search_original_query", True),
        )

8:class CitationHandler:
# citation_handler.py

from typing import Any, Dict, List, Optional, Union

from loguru import logger


class CitationHandler:
    """
    Configurable citation handler that delegates to specific implementations.
    Maintains backward compatibility while allowing strategy-specific handlers.

    """

    def __init__(
        self, llm, handler_type: Optional[str] = None, settings_snapshot=None
    ):
        self.llm = llm
        self.settings_snapshot = settings_snapshot or {}

        # Determine which handler to use
        if handler_type is None:
            # Try to get from settings snapshot, default to standard
            if "citation.handler_type" in self.settings_snapshot:
                value = self.settings_snapshot["citation.handler_type"]
                handler_type = (
                    value["value"]
                    if isinstance(val
…[1612 chars omitted]…
andler,
            )

            logger.info(
                "Using PrecisionExtractionHandler for precise answer extraction"
            )
            return PrecisionExtractionHandler(
                self.llm, settings_snapshot=self.settings_snapshot
            )

        else:
            logger.warning(
                f"Unknown citation handler type: {handler_type}, falling back to standard"

=== direct_search_strategy
33:    def __init__(
=== source_based_strategy
64:    def __init__(
=== standard_strategy
21:    def __init__(
=== parallel_search_strategy
26:    def __init__(
=== rapid_search_strategy
25:    def __init__(
=== iterdrag_strategy
25:    def __init__(

=== focused_iteration_strategy
53:    def __init__(

=== direct_search_strategy
    def __init__(
        self,
        search,
        model,
        citation_handler=None,
        include_text_content: bool = True,
        use_cross_engine_filter: bool = True,
        filter_reorder: bool = True,
        filter_reindex: bool = True,
        cross_engine_max_results: int = None,
        all_links_of_system=None,

    ):
        """Initialize with minimal components for efficiency."""
        super().__init__(all_links_of_system=all_links_of_system)
=== source_based_strategy
    def __init__(
        self,
        search,
        model,
        citation_handler=None,
        include_text_content: bool = True,
        use_cross_engine_filter: bool = True,
        filter_reorder: bool = True,
        filter_reindex: bool = True,
        cross_engine_max_results: int = None,
        all_links_of_system=None,
        use_atomic_facts: bool =
…[5270 chars omitted]…
, 200=main behavior)
        prompt_knowledge_truncate: int = 1500,  # Truncate knowledge in LLM prompt (None=unlimited, 1500=main behavior)
        previous_searches_limit: int = 10,  # Limit previous searches shown to LLM (None=unlimited, 10=main behavior)
    ):
        """Initialize with components optimized for focused iteration."""
        super().__init__(all_links_of_system, settings_snapshot)

=== direct_search_strategy
62-        )
63-
64-        # Use provided citation_handler or create one
65:        self.citation_handler = citation_handler or CitationHandler(self.model)
66-        self.findings_repository = FindingsRepository(self.model)
67-
68-    def analyze_topic(self, query: str) -> Dict:
=== source_based_strategy
127-            self.search.include_full_content = include_text_content
128-
129-        # Use provided citation_handler or create default

130:        self.citation_handler = citation_handler or CitationHandler(self.model)
131-
132-        # Initialize question generator (atomic facts variant is experimental)
133-        if use_atomic_facts:
=== standard_strategy
56-        # Note: questions_by_iteration is already initialized by parent class
57-
58-        # Use provided citation_handler or create one
59:        self.citation_handler = citation_handler or
…[1183 chars omitted]…
compositionQuestionGenerator(self.model)
=== focused_iteration_strategy
120-        handler_type = (
121-            "forced_answer" if use_browsecomp_optimization else "standard"
122-        )
123:        self.citation_handler = citation_handler or CitationHandler(
124-            self.model, handler_type=handler_type
125-        )
126-        self.findings_repository = FindingsRepository(self.model)

__init__.py
answer_decoding
candidate_exploration
candidates
constraint_checking
constraints
evidence
filters
findings
knowledge
query_generation
questions

repositories
search_optimization
source_management
strategies
tools
33:    def __init__(
34-        self,
35-        search,
36-        model,
37-        citation_handler=None,
38-        include_text_content: bool = True,
39-        use_cross_engine_filter: bool = True,
40-        filter_reorder: bool = True,
41-        filter_reindex: bool = True,
42-        cross_engine_max_results: int = None,
43-        all_links_of_system=None,
44-    ):
45-        """Initialize with minimal components for efficiency."""
46-        super().__init__(all_links_of_system=all_links_of_system)
47-        self.search = search
48-        self.model = model

answer_decoding/browsecomp_answer_decoder.py:29:    def __init__(self):
answer_decoding/browsecomp_answer_decoder.py-30-        self.encoding_schemes = [
answer_decoding/browsecomp_answer_decoder.py-31-            "base64",
answer_decoding/browsecomp_answer_decoder.py-32-            "hex",
answer_decoding/browsecomp_answer_decoder.py-33-            "url_encoding",
answer_decoding/browsecomp_answer_decoder.py-34-            "rot13",
answer_decoding/browsecomp_answer_decoder.py-35-            "caesar_cipher",
answer_decoding/browsecomp_answer_decoder.py-36-        ]
answer_decoding/browsecomp_answer_decoder.py-37-
answer_decoding/browsecomp_answer_decoder.py-38-        # Patterns that suggest encoded content
answer_decoding/browsecomp_answer_decoder.py-39-        self.encoded_patterns = [
answer_decoding/browsecomp_answer_decoder.py-40-            r"^[A-Za-z0-9+/]+=*$",  # Base64 patter

…[8630 chars omitted]…
py-35-        self.name = name
tools/base_tool.py-36-        self.description = description
tools/base_tool.py-37-        self.parameters: Dict[str, Dict[str, Any]] = {}
tools/base_tool.py-38-
tools/base_tool.py-39-    @abstractmethod
tools/base_tool.py-40-    def execute(self, **kwargs) -> Any:
tools/base_tool.py-41-        """
tools/base_tool.py-42-        Execute the tool with the given parameters.

10:from typing import Dict
12:from loguru import logger
14:from ...citation_handler import CitationHandler
17:from ..filters.cross_engine_filter import CrossEngineFilter
18:from ..findings.repository import FindingsRepository
19:from .base_strategy import BaseSearchStrategy
adaptive_decomposition_strategy.py:43:class AdaptiveDecompositionStrategy(BaseSearchStrategy):
base_strategy.py:12:class BaseSearchStrategy(ABC):
browsecomp_entity_strategy.py:97:class BrowseCompEntityStrategy(BaseSearchStrategy):
browsecomp_optimized_strategy.py:35:class BrowseCompOptimizedStrategy(BaseSearchStrategy):
constrained_search_strategy.py:24:class ConstrainedSearchStrategy(EvidenceBasedStrategy):
direct_search_strategy.py:22:class DirectSearchStrategy(BaseSearchStrategy):

entity_aware_source_strategy.py:13:class EntityAwareSourceStrategy(SourceBasedSearchStrategy):
evidence_based_strategy.py:24:class EvidenceBasedStrategy(BaseSearchStrategy):
focused_iteration_strategy.py:40:class FocusedIterationStrategy(BaseSearchStrategy):
improved_evidence_based_strategy.py:41:class ImprovedEvidenceBasedStrategy(BaseSearchStrategy):

"""
Base class for all search strategies.
Defines the common interface and shared functionality for different search approaches.
"""

from abc import ABC, abstractmethod
from typing import Any, Callable, Optional

from loguru import logger


class BaseSearchStrategy(ABC):

    """Abstract base class for all search strategies."""

    def __init__(
        self,
        all_links_of_system=None,
        settings_snapshot=None,
        questions_by_iteration=None,
        search_original_query: bool = True,
    ):
        """Initialize the base strategy with common attributes.

        Args:
            all_links_of_system: List to store all discovered links
            settings_snapshot: Settings snapshot for configuration
            questions_by_iteration: Dictionary of questions by iteration
            search_original_query: Whether to include the original query in the first iterati
…[1907 chars omitted]…
opagated.  A dedicated
        ``set_termination_check`` method could be introduced later if the
        callback-based approach becomes a maintenance burden.

        The callback in ``research_service.py`` recognises the
        ``"termination_check"`` phase and returns immediately after the
        flag check -- no UI logging or socket emission occurs.
        """
        if self.progress_callback:

28:    def _get_output_instruction_prefix(self) -> str:
29-        """
30-        Get formatted output instructions from settings if present.
31-
32-        This allows users to customize output language, tone, style, and formatting
33-        for research answers and reports. Instructions are prepended to prompts
34-        sent to the LLM.
35-
36-        Returns:
37-            str: Formatted instruction prefix if custom instructions are set,
38-                 empty string otherwise.
39-

40-        Examples:
41-            - "Respond in Spanish with formal academic tone"
42-            - "Use simple language suitable for beginners"
43-            - "Be concise with bullet points"
44-        """
45-        output_instructions = self.get_setting(
46-            "general.output_instructions", ""
47-        ).strip()
48-

"""Tests for database/thread_local_session.py."""

import threading
from unittest.mock import Mock, call, patch

from sqlalchemy.exc import OperationalError, PendingRollbackError


class TestThreadLocalSessionManager:
    """Tests for ThreadLocalSessionManager class."""

    def test_init_creates_thread_local_storage(self):

        """Test that initialization creates thread-local storage."""
        from local_deep_research.database.thread_local_session import (
            ThreadLocalSessionManager,
        )

        manager = ThreadLocalSessionManager()
        assert hasattr(manager, "_local")
        assert isinstance(manager._local, threading.local)

    def test_init_creates_credentials_tracking(self):
        """Test that initialization creates credentials tracking dict."""
        from local_deep_research.database.thread_local_session import (
            ThreadLocalSessionMa
…[5012 chars omitted]…
manager_extended.py:234:    def test_thread_cleanup_by_thread_id(self, mock_data_dir):
tests/database/test_database_manager_extended.py:257:    def test_thread_cleanup_all(self, mock_data_dir):
tests/database/test_encrypted_db_manager.py:1016:    def test_concurrent_thread_cleanup(self, mock_data_dir, tmp_path):
tests/integration/test_error_recovery.py:464:    def test_thread_cleanup_on_failure(self):

245:                ) as executor:
247:                        executor.submit(search_question, q)

questions_per_iteration=int(questions_per_iter)
                        if questions_per_iter is not None
                        else 3,
                        questions_by_iteration=self.questions_by_iteration,
                    )

                    # Use only the new questions for this iteration's searches
                    all_questions = questions

                    # Store in questions_by_iteration
                    self.questions_by_iteration[iteration] = questions
                    logger.info(

                        f"Generated questions for iteration {iteration}: {questions}"
                    )

                # Skip if no questions (all_questions includes original query in iteration 1)
                if not all_questions:
                    logger.warning(
                        f"No questions generated for iteration {iteration},
…[5082 chars omitted]…
rategies/modular_strategy.py:7:from concurrent.futures import ThreadPoolExecutor, as_completed
src/local_deep_research/advanced_search_system/strategies/modular_strategy.py:522:                # Execute batch searches in parallel using ThreadPoolExecutor
src/local_deep_research/advanced_search_system/strategies/modular_strategy.py:524:                with ThreadPoolExecutor(max_workers=8) as executor:

all ok

done

313:    elif strategy_name_lower == "iterdrag":
314:        from .advanced_search_system.strategies.iterdrag_strategy import (
326:    elif strategy_name_lower == "parallel":
327:        from .advanced_search_system.strategies.parallel_search_strategy import (
341:    elif strategy_name_lower == "rapid":
342:        from .advanced_search_system.strategies.rapid_search_strategy import (
491:        "parallel-constrained",
492:        "parallel_constrained",
494:        from .advanced_search_system.strategies.parallel_constrained_strategy import (
509:            parallel_workers=kwargs.get("parallel_workers", 100),
533:            parallel_workers=kwargs.get("parallel_workers", 100),
556:            parallel_workers=kwargs.get("parallel_workers", 100),

583:            parallel_workers=kwargs.get("parallel_workers", 100),
591:            use_direct_property_search=kwargs.get(
592:                "use_direct_property_search", True
619:            parallel_workers=kwargs.get("parallel_workers", 100),
627:            use_direct_property_search=kwargs.get(
628:                "use_direct_property_search", True
660:            parallel_workers=kwargs.get("parallel_workers", 10),
668:            use_direct_property_search=kwargs.get(

)

    # IterDRAG strategy
    elif strategy_name_lower == "iterdrag":
        from .advanced_search_system.strategies.iterdrag_strategy import (
            IterDRAGStrategy,
        )

        return IterDRAGStrategy(
            model=model,
            search=search,
            all_links_of_system=all_links_of_system,

            settings_snapshot=settings_snapshot,
        )

    # Parallel strategy
    elif strategy_name_lower == "parallel":
        from .advanced_search_system.strategies.parallel_search_strategy import (
            ParallelSearchStrategy,
        )

        return ParallelSearchStrategy(
            model=model,
            search=search,
            include_text_content=kwargs.get("include_text_content", True),
            use_cross_engine_filter=kwargs.get("use_cross_engine_filter", True),
            all_links_of_system=all_links_of_system,
       
…[364 chars omitted]…
nks_of_system,
            settings_snapshot=settings_snapshot,
        )

    # Recursive decomposition strategy
    elif strategy_name_lower in ["recursive", "recursive-decomposition"]:
        from .advanced_search_system.strategies.recursive_decomposition_strategy import (
            RecursiveDecompositionStrategy,
        )

        return RecursiveDecompositionStrategy(
            model=model,

source-based 'User-Specified Output Style: Respond in Spanish\n\n'
Traceback (most recent call last):
  File "<stdin>", line 7, in <module>
  File "/repo/src/local_deep_research/search_system_factory.py", line 859, in create_strategy
    return StandardSearchStrategy(
           ^^^^^^^^^^^^^^^^^^^^^^^
  File "/repo/src/local_deep_research/advanced_search_system/strategies/standard_strategy.py", line 39, in __init__
    get_setting_from_snapshot(
  File "/repo/src/local_deep_research/config/thread_settings.py", line 154, in get_setting_from_snapshot
    raise NoSettingsContextError(
local_deep_research.config.thread_settings.NoSettingsContextError: No settings context available in thread for key 'search.iterations'. All settings must be passed via settings_snapshot.

super().__init__(
            all_links_of_system=all_links_of_system,
            settings_snapshot=settings_snapshot,
        )
        self.search = search
        self.model = model

        # Get iterations setting
        self.max_iterations = int(
            get_setting_from_snapshot(
                "search.iterations", settings_snapshot=settings_snapshot
            )

        )

        self.questions_per_iteration = int(
            get_setting_from_snapshot(
                "search.questions_per_iteration",
                settings_snapshot=settings_snapshot,
            )
        )
        self.context_limit = int(
            get_setting_from_snapshot(
                "general.knowledge_accumulation_context_limit",
                settings_snapshot=settings_snapshot,
            )
        )
        # Note: questions_by_iteration is already initialized by parent c
…[780 chars omitted]…
    )


def get_llm_setting_from_snapshot(
    key, default=None, username=None, settings_snapshot=None
):
    """Get setting from snapshot with fallback LLM check enabled.

    Convenience wrapper used by LLM provider modules that always need
    ``check_fallback_llm=True``.
    """
    return get_setting_from_snapshot(
        key, default, username, settings_snapshot, check_fallback_llm=True
    )

source-based 'User-Specified Output Style: Respond in Spanish\n\n'
Traceback (most recent call last):
  File "<stdin>", line 8, in <module>
  File "/repo/src/local_deep_research/search_system_factory.py", line 859, in create_strategy
    return StandardSearchStrategy(
           ^^^^^^^^^^^^^^^^^^^^^^^
  File "/repo/src/local_deep_research/advanced_search_system/strategies/standard_strategy.py", line 51, in __init__
    get_setting_from_snapshot(
  File "/repo/src/local_deep_research/config/thread_settings.py", line 154, in get_setting_from_snapshot
    raise NoSettingsContextError(
local_deep_research.config.thread_settings.NoSettingsContextError: No settings context available in thread for key 'general.knowledge_accumulation_context_limit'. All settings must be passed via settings_snapshot.

source-based 'User-Specified Output Style: Respond in Spanish\n\n'
standard 'User-Specified Output Style: Respond in Spanish\n\n'
parallel 'User-Specified Output Style: Respond in Spanish\n\n'
rapid 'User-Specified Output Style: Respond in Spanish\n\n'
iterdrag 'User-Specified Output Style: Respond in Spanish\n\n'
focused-iteration 'User-Specified Output Style: Respond in Spanish\n\n'
OK

1
5:import concurrent.futures
6:from typing import Dict
8:from loguru import logger
10:from ...citation_handler import CitationHandler
13:from ...utilities.search_utilities import extract_links_from_search_results
14:from ..filters.cross_engine_filter import CrossEngineFilter
15:from ..findings.repository import FindingsRepository
16:from ..questions.standard_question import StandardQuestionGenerator
17:from .base_strategy import BaseSearchStrategy

0
22:import concurrent.futures
23:from typing import Dict, List
25:from loguru import logger
27:from ...citation_handler import CitationHandler
30:from ...utilities.thread_context import (
34:from ..candidate_exploration import ProgressiveExplorer
35:from ..findings.repository import FindingsRepository
36:from ..questions import BrowseCompQuestionGenerator
37:from .base_strategy import BaseSearchStrategy

logger.warning("No queries provided for parallel search")
            return []

        all_results = []

        def search_question(q):
            try:
                # Get the current research context to pass explicitly

                current_context = get_search_context()
                result = self.search.run(q, research_context=current_context)
                return {"question": q, "results": result or []}

            except Exception:
                logger.exception(f"Error searching '{q}'")
                return {"question": q, "results": [], "error": "Search failed"}

        # Create context-preserving wrapper for the search function
        context_aware_search = preserve_research_context(search_question)

        # Run searches in parallel
        with concurrent.futures.ThreadPoolExecutor(
            max_workers=len(queries)
        ) as executor:
 
…[845 chars omitted]…
aware_search_with_progress, q)
                for q in queries
            ]

            total_results_found = 0
            for future in concurrent.futures.as_completed(futures):
                result_dict = future.result()
                results = result_dict.get("results", [])
                all_results.extend(results)
                total_results_found += result_dict.get("result_count", 0)

ok
377:        def search_question(q):
420:        def search_question_with_progress(q):

@thread_cleanup
        def search_question(q):
            try:
                # Get the current research context to pass explicitly


        @thread_cleanup
        def search_question_with_progress(q):
            nonlocal completed_searches
            try:
                # Report starting this search
                self._update_progress(

            self.search.include_full_content = include_text_content

        # Use provided citation_handler or create one
        self.citation_handler = citation_handler or CitationHandler(
            self.model, settings_snapshot=settings_snapshot
        )

        # Initialize components
        self.question_generator = StandardQuestionGenerator(self.model)
        self.findings_repository = FindingsRepository(self.model)

    def analyze_topic(self, query: str) -> Dict:
        """
        Analyze a topic using parallel search, supporting multiple iterations.

232-                # Function for thread pool
233-                @thread_cleanup
234:                def search_question(q):

/usr/local/lib/python3.12/site-packages/_pytest/config/__init__.py:1464: PytestConfigWarning: Unknown config option: timeout
  
    self._warn_or_fail_if_strict(f"Unknown config option: {key}\n")

../usr/local/lib/python3.12/site-packages/_pytest/config/__init__.py:1464
  /usr/local/lib/python3.12/site-packages/_pytest/config/__init__.py:1464: PytestConfigWarning: Unknown config option: timeout_method
  
    self._warn_or_fail_if_strict(f"Unknown config option: {key}\n")

tests/database/test_thread_local_session.py::TestThreadLocalSessionManager::test_init_creates_thread_local_storage
  /repo/src/local_deep_research/web_search_engines/engines/local_embedding_manager.py:7: DeprecationWarning: `langchain-community` is being sunset and is no longer actively maintained. See https://github.com/langchain-ai/langchain-community/issues/674 for details and migration guidance toward standalone integration packages.
    from langchain_community.embeddings import (

-- Docs: https://docs.pytest.org/en/stable/how-to/capture-warnings.html
159 passed, 3 warnings in 26.87s

-- Docs: https://docs.pytest.org/en/stable/how-to/capture-warnings.html
1269 passed, 8 skipped, 4 warnings in 213.74s (0:03:33)

Traceback (most recent call last):
  File "<stdin>", line 6, in <module>
  File "/usr/local/lib/python3.12/unittest/mock.py", line 961, in assert_called_once_with
    return self.assert_called_with(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/usr/local/lib/python3.12/unittest/mock.py", line 949, in assert_called_with
    raise AssertionError(_error_message()) from cause
AssertionError: expected call not found.
Expected: CitationHandler(<MagicMock id='130186507096080'>, settings_snapshot=None)
  Actual: CitationHandler(<MagicMock id='130186507096080'>, settings_snapshot={'search.iterations': 1, 'search.questions_per_iteration': 1, 'general.knowledge_accumulation_context_limit': 2})
__init__.py
answer_decoding

candidate_exploration
candidates
conftest.py
constraint_checking
constraints
evidence
filters
findings

Traceback (most recent call last):
  File "<stdin>", line 4, in <module>
  File "/repo/src/local_deep_research/advanced_search_system/strategies/standard_strategy.py", line 39, in __init__
    get_setting_from_snapshot(
  File "/repo/src/local_deep_research/config/thread_settings.py", line 154, in get_setting_from_snapshot
    raise NoSettingsContextError(
local_deep_research.config.thread_settings.NoSettingsContextError: No settings context available in thread for key 'search.iterations'. All settings must be passed via settings_snapshot.

Traceback (most recent call last):
  File "<stdin>", line 7, in <module>
  File "/repo/src/local_deep_research/advanced_search_system/strategies/standard_strategy.py", line 38, in __init__
    self.max_iterations = int(
                          ^^^^
TypeError: int() argument must be a string, a bytes-like object or a real number, not 'NoneType'

ok

__init__.py
conftest.py
test_base_strategy.py
test_browsecomp_entity_strategy.py
test_browsecomp_optimized_strategy.py
test_constrained_search_extended.py
test_constrained_search_strategy.py
test_constraint_parallel_strategy.py
test_decomposition_strategies.py
test_dual_confidence_strategy.py
test_evidence_based_strategy.py
test_evidence_based_v2.py

test_evidence_based_v2_extended.py
test_focused_iteration_strategy.py
test_iterative_reasoning_strategy.py
test_iterative_refinement_strategy.py
test_iterdrag_strategy.py
test_llm_driven_modular_strategy.py
test_modular_strategy.py
test_news_strategy_pure_logic.py
test_parallel_strategies.py
test_rapid_search_strategy_extended.py
test_smart_query_strategy.py
test_standard_strategy.py
test_strategy_helpers_pure_logic.py
test_strategy_pure_logic.py
test_topic_organization_extended.py
test_topic_organization_strategy.py
ERROR tests/advanced_search_system/questions/test_news_question_generator.py
!!!!!!!!!!!!!!!!!!!!!!!!!! stopping after 1 failures !!!!!!!!!!!!!!!!!!!!!!!!!!!
3 warnings, 1 error in 5.81s

=========================== short test summary info ============================
ERROR tests/advanced_search_system/questions/test_news_question_generator.py
ERROR tests/advanced_search_system/strategies/test_news_strategy_pure_logic.py
!!!!!!!!!!!!!!!!!!! Interrupted: 2 errors during collection !!!!!!!!!!!!!!!!!!!!
3 warnings, 2 errors in 8.65s
Saved working directory and index state WIP on master: b18f5f9 base
ERROR tests/advanced_search_system/questions/test_news_question_generator.py
!!!!!!!!!!!!!!!!!!!! Interrupted: 1 error during collection !!!!!!!!!!!!!!!!!!!!
2 warnings, 1 error in 1.05s
On branch master
Changes not staged for commit:
  (use "git add <file>..." to update what will be committed)

  (use "git restore <file>..." to discard changes in working directory)
	modified:   src/local_deep_research/advanced_search_system/strategies/direct_search_strategy.py
	modified:   src
…[358 chars omitted]…
h_strategy.py
	modified:   src/local_deep_research/advanced_search_system/strategies/source_based_strategy.py
	modified:   src/local_deep_research/advanced_search_system/strategies/standard_strategy.py
	modified:   src/local_deep_research/database/thread_local_session.py

no changes added to commit (use "git add" and/or "git commit -a")
Dropped refs/stash@{0} (bacf3c8c00358a2fddfccf1a3a14ade9b7cb00b0)

FAILED tests/advanced_search_system/strategies/test_modular_strategy.py::TestAsyncMethods::test_generate_intelligent_combinations
FAILED tests/advanced_search_system/strategies/test_modular_strategy.py::TestAsyncMethods::test_quick_confidence_check
6 failed, 2876 passed, 3 skipped, 5 warnings in 38.89s

FAILED tests/advanced_search_system/strategies/test_browsecomp_entity_strategy.py::TestBrowseCompEntityStrategySearch::test_search_returns_tuple
FAILED tests/advanced_search_system/strategies/test_browsecomp_entity_strategy.py::TestBrowseCompEntityStrategySearch::test_search_calls_progress_callback
FAILED tests/advanced_search_system/strategies/test_iterdrag_strategy.py::TestIterDRAGStrategyInit::test_init_creates_components
FAILED tests/advanced_search_system/strategies/test_modular_strategy.py::TestAsyncMethods::test_decompose_constraints_intelligently
FAILED tests/advanced_search_system/strategies/test_modular_strategy.py::TestAsyncMethods::test_generate_intelligent_combinations
FAILED tests/advanced_search_system/strategies/test_modular_strategy.py::TestAsyncMethods::test_quick_confidence_check
FAILED tests/advanced_search_system/strategies/test_modular_strategy.py::TestAsyncMethods::test_generate_intelligent_combinations
FAILED tests/advanced_search_system/strategies/test_modular_strategy.py::TestAsyncMethods::test_quick_confidence_check
3 failed, 35 passed, 4 warnings in 21.61s

2026-10-01 20:05:37.480 | DEBUG    | src.local_deep_research.llm.providers.auto_discovery:discover_providers:116 -   Found class: OpenAICompatibleProvider, bases: (<class 'local_deep_research.llm.providers.base.BaseLLMProvider'>,)
2026-10-01 20:05:37.480 | DEBUG    | src.local_deep_research.llm.providers.auto_discovery:discover_providers:116 -   Found class: OpenAIProvider, bases: (<class 'local_deep_research.llm.providers.openai_base.OpenAICompatibleProvider'>,)
2026-10-01 20:05:37.481 | DEBUG    | src.local_deep_research.llm.providers.auto_discovery:discover_providers:102 - Processing module: openrouter from /repo/src/local_deep_research/llm/providers/implementations/openrouter.py
2026-10-01 20:05:37.481 | DEBUG    | src.local_deep_research.llm.providers.auto_discovery:discover_providers:111 - Inspecting module openrouter for Provider classes
2026-10-01 20:05:37.481 | DEBUG    | src
…[2444 chars omitted]…
e toward standalone integration packages.
    from langchain_community.embeddings import (

-- Docs: https://docs.pytest.org/en/stable/how-to/capture-warnings.html
=========================== short test summary info ============================
FAILED tests/advanced_search_system/strategies/test_iterdrag_strategy.py::TestIterDRAGStrategyInit::test_init_creates_components

1 failed, 3 warnings in 20.23s

mock_search = Mock()
        mock_model = Mock()
    
        IterDRAGStrategy(search=mock_search, model=mock_model)
    
>       mock_citation.assert_called_once_with(mock_model)

tests/advanced_search_system/strategies/test_iterdrag_strategy.py:109: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
/usr/local/lib/python3.12/unittest/mock.py:961: in assert_called_once_with
    return self.assert_called_with(*args, **kwargs)
--

_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <MagicMock name='CitationHandler' id='138954893423392'>
args = (<Mock id='138954892007744'>,), kwargs = {}

    def assert_called_with(self, /, *args, **kwargs):
        """assert that the last call was made with the specified arguments.
    
        Raises an AssertionError if the args and keyword args passed in are
--
        if sel
…[795 chars omitted]…
model=mock_model)
108-
109-        mock_citation.assert_called_once_with(mock_model)
110-        mock_question.assert_called_once_with(mock_model)
111-        mock_knowledge.assert_called_once_with(mock_model)
112-        mock_findings.assert_called_once_with(mock_model)
113-
114-    @patch(
115-        "local_deep_research.advanced_search_system.strategies.iterdrag_strategy.CitationHandler"
116-    )

FAILED tests/advanced_search_system/strategies/test_browsecomp_entity_strategy.py::TestBrowseCompEntityStrategySearch::test_search_returns_tuple
FAILED tests/advanced_search_system/strategies/test_browsecomp_entity_strategy.py::TestBrowseCompEntityStrategySearch::test_search_calls_progress_callback
2 failed, 40 passed, 3 warnings in 19.84s

.../strategies/direct_search_strategy.py           | 10 ++-
 .../strategies/focused_iteration_strategy.py       |  7 +-
 .../strategies/iterdrag_strategy.py                |  4 +-
 .../strategies/parallel_search_strategy.py         |  6 +-
 .../strategies/rapid_search_strategy.py            |  4 +-
 .../strategies/source_based_strategy.py            |  4 +-
 .../strategies/standard_strategy.py                |  4 +-
 .../database/thread_local_session.py               | 86 +++++++++++++++++-----
 8 files changed, 98 insertions(+), 27 deletions(-)
-- Docs: https://docs.pytest.org/en/stable/how-to/capture-warnings.html
22 passed, 3 warnings in 17.45s

diff --git a/src/local_deep_research/database/thread_local_session.py b/src/local_deep_research/database/thread_local_session.py
index 8ced415..4411376 100644
--- a/src/local_deep_research/database/thread_local_session.py
+++ b/src/local_deep_research/database/thread_local_session.py
@@ -187,36 +187,84 @@ def cleanup_current_thread():
     thread_session_manager.cleanup_thread()
 
 
-def thread_cleanup(func):
-    """Decorator ensuring all thread-local resources are cleaned up when func exits.
+class thread_cleanup:
+    """Ensure all thread-local resources are cleaned up when a function exits.

+
+    Supports four usage shapes:
+
+    1. Bare decorator::
+
+           @thread_cleanup
+           def worker():
+               ...
+
+    2. Decorator factory::
+
+           @thread_cleanup()
+           def worker():
+               ...
+
+    3. Context manager::
+
+           with th
…[2087 chars omitted]…
== 1 and callable(args[0]) and not kwargs:
+            return thread_cleanup(args[0])
 
-    return wrapper
+        # Used as a decorator factory or context manager: thread_cleanup()
+        return self
+
+    def __enter__(self):
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        self._run_cleanup()
+        return False
 
 
 # Context manager for automatic cleanup

FAILED tests/news/subscription_manager/test_scheduler.py::TestCheckUserOverdueSubscriptions::test_overdue_schedules_immediate_jobs
7 failed, 146 passed, 3 warnings in 23.25s
-- Docs: https://docs.pytest.org/en/stable/how-to/capture-warnings.html
395 deselected, 2 warnings in 4.82s

-- Docs: https://docs.pytest.org/en/stable/how-to/capture-warnings.html
153 passed, 3 warnings in 21.80s

___ TestCheckUserOverdueSubscriptions.test_overdue_schedules_immediate_jobs ____

self = <subscription_manager.test_scheduler.TestCheckUserOverdueSubscriptions object at 0x7a15246c1520>
scheduler = <local_deep_research.news.subscription_manager.scheduler.NewsScheduler object at 0x7a135ab5eb10>
mock_background_scheduler = <MagicMock name='BackgroundScheduler()' id='134223544837440'>

    def test_overdue_schedules_immediate_jobs(
        self, scheduler, mock_background_scheduler
    ):
        """_check_user_overdue_subscriptions schedules overdue subscriptions."""
        scheduler.user_sessions["testuser"] = {
            "password": "testpass",

            "scheduled_jobs": set(),
            "last_activity": datetime.now(UTC),
        }
    
        mock_subscription = MagicMock()
        mock_subscription.id = 1
        mock_subscription.name = "Overdue"
        mock_subscription
…[557 chars omitted]…
ions("testuser")

tests/news/subscription_manager/test_scheduler.py:662: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
src/local_deep_research/database/thread_local_session.py:253: in __call__
--
FAILED tests/news/subscription_manager/test_scheduler.py::TestCheckUserOverdueSubscriptions::test_overdue_schedules_immediate_jobs
7 failed, 45 passed, 3 warnings in 22.49s

logger.info(
                    f"[DOC_SCHEDULER] Successfully verified job {job_id} exists, next run: {job.next_run_time}"
                )
            else:
                logger.error(
                    f"[DOC_SCHEDULER] Failed to verify job {job_id} exists!"
                )

        except Exception:
            logger.exception(
                f"Error scheduling document processing for {username}"
            )

    @thread_cleanup
    def _process_user_documents(self, username: str):
        """Process documents for a user."""
        logger.info(f"[DOC_SCHEDULER] Processing documents for user {username}")
        start_time = datetime.now(UTC)

        try:
            session_info = self.user_sessions.get(username)
            if not session_info:
                logger.warning(
                    f"[DOC_SCHEDULER] No session info found for user {username}"
                )
                return

            password = session_info["password"]
            logger.debug(
                f"[DOC_SCHEDULER] Starting document processing for {username}"
            )

src/local_deep_research/database/thread_local_session.py:253: in __call__
    return self._run(self.func, args, kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <local_deep_research.database.thread_local_session.thread_cleanup object at 0x715281869610>
func = <function NewsScheduler._check_user_overdue_subscriptions at 0x7152817af420>
args = ('testuser',), kwargs = {}

    def _run(self, func, args, kwargs):
        try:
>           return func(*args, **kwargs)

                   ^^^^^^^^^^^^^^^^^^^^^
E           TypeError: NewsScheduler._check_user_overdue_subscriptions() missing 1 required positional argument: 'username'

-- Docs: https://docs.pytest.org/en/stable/how-to/capture-warnings.html
153 passed, 3 warnings in 19.25s

ALL OK

-- Docs: https://docs.pytest.org/en/stable/how-to/capture-warnings.html
1478 passed, 8 skipped, 4 warnings in 231.30s (0:03:51)
-- Docs: https://docs.pytest.org/en/stable/how-to/capture-warnings.html
115 passed, 3 warnings in 17.32s

diff --git a/src/local_deep_research/advanced_search_system/strategies/direct_search_strategy.py b/src/local_deep_research/advanced_search_system/strategies/direct_search_strategy.py
index 71da2c9..e793413 100644
--- a/src/local_deep_research/advanced_search_system/strategies/direct_search_strategy.py
+++ b/src/local_deep_research/advanced_search_system/strategies/direct_search_strategy.py
@@ -41,9 +41,13 @@ class DirectSearchStrategy(BaseSearchStrategy):
         filter_reindex: bool = True,
         cross_engine_max_results: int = None,
         all_links_of_system=None,
+        settings_snapshot=None,
     ):
         """Initialize with minimal components for efficiency."""
-        super().__init__(all_links_of_system=all_links_of_system)

+        super().__init__(
+            all_links_of_system=all_links_of_system,
+            settings_snapshot=settings_snapshot,
+        )
 
…[4727 chars omitted]…
rch_system/strategies/rapid_search_strategy.py b/src/local_deep_research/advanced_search_system/strategies/rapid_search_strategy.py
index ee55d8d..4a9810b 100644
--- a/src/local_deep_research/advanced_search_system/strategies/rapid_search_strategy.py
+++ b/src/local_deep_research/advanced_search_system/strategies/rapid_search_strategy.py
@@ -41,7 +41,9 @@ class RapidSearchStrategy(BaseSearchStrategy):

submitted.