django-explain-errors 0.3.1__tar.gz → 0.4.0__tar.gz

{django_explain_errors-0.3.1/django_explain_errors.egg-info → django_explain_errors-0.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: django-explain-errors
-Version: 0.3.1
+Version: 0.4.0
 Summary: Django middleware that captures errors and exceptions, sends them to OpenAI for a detailed explanation, and prints the explanation to stdout when debug mode is enabled. Supports both sync and async views.
 Home-page: https://github.com/topunix/django-explain-errors
 Author: topunix
@@ -27,7 +27,7 @@ Requires-Python: >=3.9
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: Django>=4.2
-Requires-Dist: openai>=1.0
+Requires-Dist: openai<4.0,>=1.0
 Requires-Dist: python-dotenv>=1.0
 Requires-Dist: asgiref>=3.6
 Provides-Extra: rag
@@ -99,7 +99,7 @@ pip install django-explain-errors
 
 2. **Trigger an error in your Django application**:
 
-   The middleware will capture the error, send it to OpenAI for explanation, and print the explanation to stdout. When an exception is caught, it returns a JSON `500` response containing the error message and the explanation.
+   The middleware will capture the error, send it to OpenAI for explanation, and print the explanation to stdout. When an exception is caught, it returns a JSON `500` response containing the error message and the explanation. Set `EXPLAIN_ERRORS_PRESERVE_DEBUG_PAGE = True` to keep the stdout explanation while letting Django render its standard debug page instead.
 
 ## Async Support
 
@@ -120,6 +120,22 @@ No additional settings are needed. Place the middleware last in `MIDDLEWARE` for
 | `OPENAI_MAX_TOKENS` | No | Maximum tokens in the explanation. Defaults to `150`. |
 | `OPENAI_TIMEOUT` | No | Request timeout in seconds for the OpenAI client. Defaults to `10`. |
 | `OPENAI_MAX_TRACEBACK_CHARS` | No | Traceback is trimmed to its last N characters before being sent. Defaults to `3000`. |
+| `EXPLAIN_ERRORS_PRESERVE_DEBUG_PAGE` | No | When True, the middleware prints the explanation to stdout and re-raises the exception so Django renders its standard debug page instead of a JSON 500. Defaults to False. |
+| `OPENAI_BASE_URL` (env or settings) | No | Base URL for any OpenAI-compatible API (for example Ollama at `http://localhost:11434/v1`). When set, a missing API key is replaced with a placeholder since local servers do not require one. |
+
+## Using local models (Ollama)
+
+Point `OPENAI_BASE_URL` at any OpenAI-compatible server to run explanations
+against a local model instead of the OpenAI API:
+
+```python
+OPENAI_BASE_URL = "http://localhost:11434/v1"
+OPENAI_MODEL = "llama3.1"
+EXPLAIN_ERRORS_RAG_EMBED_MODEL = "nomic-embed-text"
+```
+
+If you use the RAG layer, rebuild the index after changing the embedding
+model or provider. Stored vectors are model-specific.
 
 ## Codebase-aware explanations (RAG)
 

{django_explain_errors-0.3.1 → django_explain_errors-0.4.0}/README.md

@@ -52,7 +52,7 @@ pip install django-explain-errors
 
 2. **Trigger an error in your Django application**:
 
-   The middleware will capture the error, send it to OpenAI for explanation, and print the explanation to stdout. When an exception is caught, it returns a JSON `500` response containing the error message and the explanation.
+   The middleware will capture the error, send it to OpenAI for explanation, and print the explanation to stdout. When an exception is caught, it returns a JSON `500` response containing the error message and the explanation. Set `EXPLAIN_ERRORS_PRESERVE_DEBUG_PAGE = True` to keep the stdout explanation while letting Django render its standard debug page instead.
 
 ## Async Support
 
@@ -73,6 +73,22 @@ No additional settings are needed. Place the middleware last in `MIDDLEWARE` for
 | `OPENAI_MAX_TOKENS` | No | Maximum tokens in the explanation. Defaults to `150`. |
 | `OPENAI_TIMEOUT` | No | Request timeout in seconds for the OpenAI client. Defaults to `10`. |
 | `OPENAI_MAX_TRACEBACK_CHARS` | No | Traceback is trimmed to its last N characters before being sent. Defaults to `3000`. |
+| `EXPLAIN_ERRORS_PRESERVE_DEBUG_PAGE` | No | When True, the middleware prints the explanation to stdout and re-raises the exception so Django renders its standard debug page instead of a JSON 500. Defaults to False. |
+| `OPENAI_BASE_URL` (env or settings) | No | Base URL for any OpenAI-compatible API (for example Ollama at `http://localhost:11434/v1`). When set, a missing API key is replaced with a placeholder since local servers do not require one. |
+
+## Using local models (Ollama)
+
+Point `OPENAI_BASE_URL` at any OpenAI-compatible server to run explanations
+against a local model instead of the OpenAI API:
+
+```python
+OPENAI_BASE_URL = "http://localhost:11434/v1"
+OPENAI_MODEL = "llama3.1"
+EXPLAIN_ERRORS_RAG_EMBED_MODEL = "nomic-embed-text"
+```
+
+If you use the RAG layer, rebuild the index after changing the embedding
+model or provider. Stored vectors are model-specific.
 
 ## Codebase-aware explanations (RAG)
 

{django_explain_errors-0.3.1 → django_explain_errors-0.4.0/django_explain_errors.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: django-explain-errors
-Version: 0.3.1
+Version: 0.4.0
 Summary: Django middleware that captures errors and exceptions, sends them to OpenAI for a detailed explanation, and prints the explanation to stdout when debug mode is enabled. Supports both sync and async views.
 Home-page: https://github.com/topunix/django-explain-errors
 Author: topunix
@@ -27,7 +27,7 @@ Requires-Python: >=3.9
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: Django>=4.2
-Requires-Dist: openai>=1.0
+Requires-Dist: openai<4.0,>=1.0
 Requires-Dist: python-dotenv>=1.0
 Requires-Dist: asgiref>=3.6
 Provides-Extra: rag
@@ -99,7 +99,7 @@ pip install django-explain-errors
 
 2. **Trigger an error in your Django application**:
 
-   The middleware will capture the error, send it to OpenAI for explanation, and print the explanation to stdout. When an exception is caught, it returns a JSON `500` response containing the error message and the explanation.
+   The middleware will capture the error, send it to OpenAI for explanation, and print the explanation to stdout. When an exception is caught, it returns a JSON `500` response containing the error message and the explanation. Set `EXPLAIN_ERRORS_PRESERVE_DEBUG_PAGE = True` to keep the stdout explanation while letting Django render its standard debug page instead.
 
 ## Async Support
 
@@ -120,6 +120,22 @@ No additional settings are needed. Place the middleware last in `MIDDLEWARE` for
 | `OPENAI_MAX_TOKENS` | No | Maximum tokens in the explanation. Defaults to `150`. |
 | `OPENAI_TIMEOUT` | No | Request timeout in seconds for the OpenAI client. Defaults to `10`. |
 | `OPENAI_MAX_TRACEBACK_CHARS` | No | Traceback is trimmed to its last N characters before being sent. Defaults to `3000`. |
+| `EXPLAIN_ERRORS_PRESERVE_DEBUG_PAGE` | No | When True, the middleware prints the explanation to stdout and re-raises the exception so Django renders its standard debug page instead of a JSON 500. Defaults to False. |
+| `OPENAI_BASE_URL` (env or settings) | No | Base URL for any OpenAI-compatible API (for example Ollama at `http://localhost:11434/v1`). When set, a missing API key is replaced with a placeholder since local servers do not require one. |
+
+## Using local models (Ollama)
+
+Point `OPENAI_BASE_URL` at any OpenAI-compatible server to run explanations
+against a local model instead of the OpenAI API:
+
+```python
+OPENAI_BASE_URL = "http://localhost:11434/v1"
+OPENAI_MODEL = "llama3.1"
+EXPLAIN_ERRORS_RAG_EMBED_MODEL = "nomic-embed-text"
+```
+
+If you use the RAG layer, rebuild the index after changing the embedding
+model or provider. Stored vectors are model-specific.
 
 ## Codebase-aware explanations (RAG)
 

{django_explain_errors-0.3.1 → django_explain_errors-0.4.0}/django_explain_errors.egg-info/SOURCES.txt

@@ -8,6 +8,7 @@ django_explain_errors.egg-info/dependency_links.txt
 django_explain_errors.egg-info/requires.txt
 django_explain_errors.egg-info/top_level.txt
 explain_errors/__init__.py
+explain_errors/client.py
 explain_errors/middleware.py
 explain_errors/sanitize.py
 explain_errors/throttle.py
@@ -19,6 +20,7 @@ explain_errors/rag/indexer.py
 explain_errors/rag/retriever.py
 explain_errors/rag/store.py
 tests/test_build_error_index.py
+tests/test_client.py
 tests/test_middleware.py
 tests/test_rag.py
 tests/test_sanitize.py

{django_explain_errors-0.3.1 → django_explain_errors-0.4.0}/django_explain_errors.egg-info/requires.txt

@@ -1,5 +1,5 @@
 Django>=4.2
-openai>=1.0
+openai<4.0,>=1.0
 python-dotenv>=1.0
 asgiref>=3.6
 

django_explain_errors-0.4.0/explain_errors/client.py

@@ -0,0 +1,37 @@
+import os
+
+from django.conf import settings
+from dotenv import load_dotenv, find_dotenv
+
+PLACEHOLDER_API_KEY = "sk-no-key-required"
+
+
+def get_openai_client(timeout=None):
+    """Build the OpenAI client used by the middleware and the
+    RAG indexer. Honors OPENAI_BASE_URL for OpenAI-compatible
+    servers (Ollama, LM Studio, gateways)."""
+    load_dotenv(find_dotenv(usecwd=True))
+    from openai import OpenAI
+
+    base_url = os.getenv(
+        "OPENAI_BASE_URL", getattr(settings, "OPENAI_BASE_URL", None)
+    )
+    api_key = os.getenv(
+        "OPENAI_API_KEY", getattr(settings, "OPENAI_API_KEY", None)
+    )
+    if not api_key:
+        if base_url:
+            # Local OpenAI-compatible servers ignore the key but
+            # the SDK requires one.
+            api_key = PLACEHOLDER_API_KEY
+        else:
+            raise ValueError(
+                "OpenAI API key not found. Please set the "
+                "OPENAI_API_KEY environment variable."
+            )
+    kwargs = {"api_key": api_key}
+    if base_url:
+        kwargs["base_url"] = base_url
+    if timeout is not None:
+        kwargs["timeout"] = timeout
+    return OpenAI(**kwargs)

{django_explain_errors-0.3.1 → django_explain_errors-0.4.0}/explain_errors/middleware.py

@@ -1,14 +1,12 @@
-import os
 import asyncio
 import logging
 import traceback
 
-from openai import OpenAI
-from dotenv import load_dotenv, find_dotenv
 from django.conf import settings
 from django.http import JsonResponse
 from asgiref.sync import sync_to_async
 
+from .client import get_openai_client
 from .sanitize import sanitize_traceback
 from .throttle import SlidingWindowThrottle
 from .rag.retriever import format_chunks_for_prompt, retrieve_chunks
@@ -37,24 +35,12 @@ class ExplainErrorsMiddleware:
             window_seconds = getattr(settings, "EXPLAIN_ERRORS_WINDOW_SECONDS", 60)
             self.throttle = SlidingWindowThrottle(max_calls, window_seconds)
 
-            # Load environment variables from .env file
-            load_dotenv(find_dotenv(usecwd=True))
-            # Get the OpenAI API key from environment variable (or settings)
-            openai_api_key = os.getenv(
-                "OPENAI_API_KEY", getattr(settings, "OPENAI_API_KEY", None)
-            )
-            if not openai_api_key:
-                raise ValueError(
-                    "OpenAI API key not found. Please set the OPENAI_API_KEY "
-                    "environment variable."
-                )
-
             # Configurable via settings, with sensible defaults.
             self.model = getattr(settings, "OPENAI_MODEL", "gpt-4o-mini")
             self.max_tokens = getattr(settings, "OPENAI_MAX_TOKENS", 150)
             timeout = getattr(settings, "OPENAI_TIMEOUT", 10)
 
-            self.openai_client = OpenAI(api_key=openai_api_key, timeout=timeout)
+            self.openai_client = get_openai_client(timeout=timeout)
 
     def __call__(self, request):
         # Delegate to the async path when wrapped around an async view chain.
@@ -67,7 +53,10 @@ class ExplainErrorsMiddleware:
         try:
             response = self.get_response(request)
         except Exception as exception:
-            return self.process_exception(request, exception)
+            response = self.process_exception(request, exception)
+            if response is None:
+                raise
+            return response
         return response
 
     # --------- Async path ----------
@@ -77,7 +66,10 @@ class ExplainErrorsMiddleware:
         except Exception as exception:
             # process_exception performs blocking OpenAI I/O, so run it in a
             # thread to keep the event loop free.
-            return await sync_to_async(self.process_exception)(request, exception)
+            response = await sync_to_async(self.process_exception)(request, exception)
+            if response is None:
+                raise
+            return response
         return response
 
     def process_exception(self, request, exception):
@@ -138,6 +130,9 @@ class ExplainErrorsMiddleware:
                 # a 500 so the request lifecycle completes cleanly.
                 print("Failed to get an explanation from OpenAI:", e)
 
+        if getattr(settings, "EXPLAIN_ERRORS_PRESERVE_DEBUG_PAGE", False):
+            return None
+
         return JsonResponse(
             {"error": "An error occurred.", "message": explanation}, status=500
         )

{django_explain_errors-0.3.1 → django_explain_errors-0.4.0}/explain_errors/rag/indexer.py

@@ -3,8 +3,8 @@ import ast
 import os
 
 from django.conf import settings
-from dotenv import load_dotenv, find_dotenv
 
+from ..client import get_openai_client
 from ..sanitize import sanitize_traceback
 from .store import VectorStore
 
@@ -62,19 +62,6 @@ def get_embed_model():
     return getattr(settings, "EXPLAIN_ERRORS_RAG_EMBED_MODEL", "text-embedding-3-small")
 
 
-def get_openai_client():
-    load_dotenv(find_dotenv(usecwd=True))
-    from openai import OpenAI
-
-    api_key = os.getenv("OPENAI_API_KEY", getattr(settings, "OPENAI_API_KEY", None))
-    if not api_key:
-        raise ValueError(
-            "OpenAI API key not found. Please set the OPENAI_API_KEY "
-            "environment variable."
-        )
-    return OpenAI(api_key=api_key)
-
-
 def discover_files(include_dirs, exclude_dirs):
     """Walk include_dirs, pruning excluded directory names, keep chunkable files."""
     exclude_set = set(exclude_dirs)

{django_explain_errors-0.3.1 → django_explain_errors-0.4.0}/setup.py

@@ -10,7 +10,7 @@ with open('README.md', encoding='utf-8') as f:
 
 setup(
     name='django-explain-errors',
-    version='0.3.1',
+    version='0.4.0',
     packages=find_packages(exclude=['tests', 'tests.*']),
     description='Django middleware that captures errors and exceptions, sends them to OpenAI for a detailed explanation, and prints the explanation to stdout when debug mode is enabled. Supports both sync and async views.',
     long_description_content_type='text/markdown',
@@ -18,7 +18,7 @@ setup(
     python_requires='>=3.9',
     install_requires=[
         'Django>=4.2',
-        'openai>=1.0',
+        'openai>=1.0,<4.0',
         'python-dotenv>=1.0',
         'asgiref>=3.6',
     ],

django_explain_errors-0.4.0/tests/test_client.py

@@ -0,0 +1,114 @@
+import os
+from unittest.mock import patch
+
+from django.test import SimpleTestCase, override_settings
+
+from explain_errors.client import PLACEHOLDER_API_KEY, get_openai_client
+from explain_errors.middleware import ExplainErrorsMiddleware
+
+
+class GetOpenAIClientTest(SimpleTestCase):
+
+    def _clear_env(self):
+        """Pop the env vars the factory reads; patch.dict restores the
+        original environment on exit regardless of these mutations."""
+        os.environ.pop("OPENAI_API_KEY", None)
+        os.environ.pop("OPENAI_BASE_URL", None)
+
+    @override_settings(OPENAI_API_KEY="test-key")
+    def test_default_no_base_url_kwarg(self):
+        with patch.dict(os.environ, {}, clear=False):
+            self._clear_env()
+            with patch("openai.OpenAI") as mock_cls:
+                get_openai_client()
+
+        mock_cls.assert_called_once()
+        kwargs = mock_cls.call_args.kwargs
+        self.assertEqual(kwargs["api_key"], "test-key")
+        self.assertNotIn("base_url", kwargs)
+
+    @override_settings(
+        OPENAI_API_KEY="test-key",
+        OPENAI_BASE_URL="http://localhost:11434/v1",
+    )
+    def test_base_url_from_settings_passed_to_client(self):
+        with patch.dict(os.environ, {}, clear=False):
+            self._clear_env()
+            with patch("openai.OpenAI") as mock_cls:
+                get_openai_client()
+
+        kwargs = mock_cls.call_args.kwargs
+        self.assertEqual(kwargs["base_url"], "http://localhost:11434/v1")
+
+    @override_settings(
+        OPENAI_API_KEY="test-key",
+        OPENAI_BASE_URL="http://settings-host/v1",
+    )
+    def test_base_url_env_var_takes_precedence_over_settings(self):
+        with patch.dict(os.environ, {"OPENAI_BASE_URL": "http://env-host/v1"}):
+            with patch("openai.OpenAI") as mock_cls:
+                get_openai_client()
+
+        kwargs = mock_cls.call_args.kwargs
+        self.assertEqual(kwargs["base_url"], "http://env-host/v1")
+
+    @override_settings(OPENAI_BASE_URL="http://localhost:11434/v1")
+    def test_missing_key_with_base_url_uses_placeholder(self):
+        with patch.dict(os.environ, {}, clear=False):
+            self._clear_env()
+            with patch("openai.OpenAI") as mock_cls:
+                get_openai_client()
+
+        kwargs = mock_cls.call_args.kwargs
+        self.assertEqual(kwargs["api_key"], PLACEHOLDER_API_KEY)
+
+    def test_missing_key_without_base_url_raises(self):
+        with patch.dict(os.environ, {}, clear=False):
+            self._clear_env()
+            with self.assertRaises(ValueError) as ctx:
+                get_openai_client()
+
+        self.assertEqual(
+            str(ctx.exception),
+            "OpenAI API key not found. Please set the "
+            "OPENAI_API_KEY environment variable.",
+        )
+
+    @override_settings(OPENAI_API_KEY="test-key")
+    def test_timeout_forwarded_when_provided(self):
+        with patch.dict(os.environ, {}, clear=False):
+            self._clear_env()
+            with patch("openai.OpenAI") as mock_cls:
+                get_openai_client(timeout=15)
+
+        self.assertEqual(mock_cls.call_args.kwargs["timeout"], 15)
+
+
+class MiddlewareUsesFactoryTest(SimpleTestCase):
+
+    @override_settings(DEBUG=True, OPENAI_API_KEY="test-key", OPENAI_TIMEOUT=7)
+    def test_middleware_uses_factory(self):
+        with patch("explain_errors.middleware.get_openai_client") as mock_factory:
+            ExplainErrorsMiddleware(lambda r: None)
+
+        mock_factory.assert_called_once_with(timeout=7)
+
+
+class IndexerUsesFactoryTest(SimpleTestCase):
+
+    @override_settings(
+        OPENAI_API_KEY="test-key",
+        OPENAI_BASE_URL="http://localhost:11434/v1",
+    )
+    def test_indexer_uses_factory(self):
+        from explain_errors.rag.indexer import get_openai_client as indexer_get_client
+
+        with patch.dict(os.environ, {}, clear=False):
+            os.environ.pop("OPENAI_API_KEY", None)
+            os.environ.pop("OPENAI_BASE_URL", None)
+            with patch("openai.OpenAI") as mock_cls:
+                indexer_get_client()
+
+        kwargs = mock_cls.call_args.kwargs
+        self.assertEqual(kwargs["api_key"], "test-key")
+        self.assertEqual(kwargs["base_url"], "http://localhost:11434/v1")

{django_explain_errors-0.3.1 → django_explain_errors-0.4.0}/tests/test_middleware.py

@@ -8,8 +8,14 @@ from explain_errors.middleware import ExplainErrorsMiddleware
 
 
 def _mock_openai():
-    """Return a patch context for OpenAI plus a configured fake client."""
-    patcher = patch("explain_errors.middleware.OpenAI")
+    """Return a patch context for OpenAI plus a configured fake client.
+
+    Patches openai.OpenAI at its source (rather than
+    explain_errors.middleware.OpenAI) because the client is now built by
+    explain_errors.client.get_openai_client(), which imports OpenAI
+    locally inside its function body.
+    """
+    patcher = patch("openai.OpenAI")
     mock_cls = patcher.start()
     client = MagicMock()
     client.chat.completions.create.return_value = MagicMock(
@@ -148,6 +154,74 @@ class ExplainErrorsMiddlewareDebugOffTest(SimpleTestCase):
         mw = ExplainErrorsMiddleware(lambda r: None)
         self.assertIsNone(mw.process_exception(factory.get("/"), Exception("x")))
 
+    def test_debug_false_handler_reraises(self):
+        factory = RequestFactory()
+
+        def boom(r):
+            raise ValueError("boom")
+
+        mw = ExplainErrorsMiddleware(boom)
+        with self.assertRaises(ValueError):
+            mw(factory.get("/"))
+
+
+@override_settings(DEBUG=True, OPENAI_API_KEY="test-key")
+class ExplainErrorsMiddlewarePreserveDebugPageTest(SimpleTestCase):
+
+    def setUp(self):
+        self.factory = RequestFactory()
+        self.patcher, self.client = _mock_openai()
+        self.addCleanup(self.patcher.stop)
+
+    @override_settings(EXPLAIN_ERRORS_PRESERVE_DEBUG_PAGE=True)
+    def test_preserve_flag_reraises_original_exception_sync(self):
+        def boom(r):
+            raise ValueError("boom")
+
+        mw = ExplainErrorsMiddleware(boom)
+        with self.assertRaises(ValueError) as ctx:
+            mw(self.factory.get("/"))
+        self.assertEqual(str(ctx.exception), "boom")
+
+    @override_settings(EXPLAIN_ERRORS_PRESERVE_DEBUG_PAGE=True)
+    def test_preserve_flag_still_prints_explanation(self):
+        def boom(r):
+            raise ValueError("boom")
+
+        mw = ExplainErrorsMiddleware(boom)
+        with patch("builtins.print") as mock_print:
+            with self.assertRaises(ValueError):
+                mw(self.factory.get("/"))
+        mock_print.assert_any_call("Error Explanation by OpenAI:\n", "Mocked explanation.")
+
+    def test_preserve_flag_default_off_returns_json_500(self):
+        def boom(r):
+            raise ValueError("boom")
+
+        mw = ExplainErrorsMiddleware(boom)
+        resp = mw(self.factory.get("/"))
+        self.assertIsInstance(resp, JsonResponse)
+        self.assertEqual(resp.status_code, 500)
+        self.assertIn("error", json.loads(resp.content))
+
+
+@override_settings(DEBUG=True, OPENAI_API_KEY="test-key", EXPLAIN_ERRORS_PRESERVE_DEBUG_PAGE=True)
+class ExplainErrorsMiddlewarePreserveDebugPageAsyncTest(SimpleTestCase):
+
+    def setUp(self):
+        self.factory = AsyncRequestFactory()
+        self.patcher, self.client = _mock_openai()
+        self.addCleanup(self.patcher.stop)
+
+    async def test_preserve_flag_reraises_original_exception_async(self):
+        async def boom(r):
+            raise ValueError("async boom")
+
+        mw = ExplainErrorsMiddleware(boom)
+        with self.assertRaises(ValueError) as ctx:
+            await mw(self.factory.get("/"))
+        self.assertEqual(str(ctx.exception), "async boom")
+
 
 @override_settings(DEBUG=True, OPENAI_API_KEY="test-key")
 class OpenAICallConfigTest(SimpleTestCase):
@@ -185,6 +259,6 @@ class OpenAICallConfigTest(SimpleTestCase):
 
     def test_timeout_passed_to_client(self):
         with override_settings(OPENAI_TIMEOUT=7):
-            with patch("explain_errors.middleware.OpenAI") as mock_cls:
+            with patch("openai.OpenAI") as mock_cls:
                 ExplainErrorsMiddleware(lambda r: None)
                 self.assertEqual(mock_cls.call_args.kwargs["timeout"], 7)

{django_explain_errors-0.3.1 → django_explain_errors-0.4.0}/tests/test_rag.py

@@ -39,17 +39,15 @@ requires_sqlite_vec = unittest.skipUnless(
 TESTS_DIR = os.path.dirname(os.path.abspath(__file__))
 
 
-
-def test_get_openai_client_loads_dotenv(self):
-    with patch("explain_errors.rag.indexer.load_dotenv") as mock_load:
-        with patch.dict(os.environ, {"OPENAI_API_KEY": "sk-test"}):
-            get_openai_client()
-    mock_load.assert_called_once()
-
-
 def _mock_openai():
-    """Patch the chat-completion OpenAI client used by the middleware."""
-    patcher = patch("explain_errors.middleware.OpenAI")
+    """Patch the chat-completion OpenAI client used by the middleware.
+
+    Patches openai.OpenAI at its source (rather than
+    explain_errors.middleware.OpenAI) because the client is now built by
+    explain_errors.client.get_openai_client(), which imports OpenAI
+    locally inside its function body.
+    """
+    patcher = patch("openai.OpenAI")
     mock_cls = patcher.start()
     client = MagicMock()
     client.chat.completions.create.return_value = MagicMock(