fix(proxy-extras): name the libpq load failure when the partition check is skipped

psycopg imports fine without libpq and only raises when the pq wrapper
is resolved, so the skipped-check warning now carries the ImportError
text and points at the system libpq the images ship

Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>

litellm-proxy-extras/litellm_proxy_extras/utils.py

@@ -604,13 +604,15 @@ class ProxyExtrasDBManager:
 
         try:
             import psycopg
-        except ImportError:
+        except ImportError as exc:
             logger.warning(
-                "psycopg is not installed; skipping the LiteLLM_SpendLogs "
-                "partition check. If this table is partitioned (see "
+                "psycopg is not installed or found no libpq (%s); skipping the "
+                "LiteLLM_SpendLogs partition check. If this table is partitioned (see "
                 "db_scripts/partition_spend_logs.sql), schema reconciliation "
                 "will try to rewrite its primary key and fail. Install the "
-                "litellm[extra_proxy] extra, which now includes psycopg."
+                "litellm[extra_proxy] extra, which includes psycopg, plus the "
+                "system libpq it loads (the litellm images ship it).",
+                exc,
             )
             return False
 
@@ -798,10 +800,11 @@ class ProxyExtrasDBManager:
         try:
             import psycopg
             from psycopg import sql
-        except ImportError:
+        except ImportError as exc:
             logger.warning(
-                "psycopg is not installed; skipping the invalid index check. "
-                "Install the litellm[extra_proxy] extra, which includes psycopg."
+                "psycopg is not installed or found no libpq (%s); skipping the invalid index check. "
+                "Install the litellm[extra_proxy] extra, which includes psycopg, plus the system libpq.",
+                exc,
             )
             return False
 

tests/litellm-proxy-extras/test_litellm_proxy_extras_utils.py

@@ -3,6 +3,7 @@ import os
 import re
 import sys
 from pathlib import Path
+from typing import Final
 
 import pytest
 
@@ -705,6 +706,23 @@ class TestSpendLogsPartitionDetectionMissingPsycopg:
             "psycopg is not installed" in record.message for record in caplog.records
         )
 
+    def test_missing_libpq_warning_names_the_libpq_error(self, monkeypatch, caplog):
+        class NoLibpq:
+            @staticmethod
+            def find_spec(name, path=None, target=None):
+                if name == "psycopg":
+                    raise ImportError("no pq wrapper available.\nAttempts made:\n- couldn't import psycopg 'python'")
+
+        for name in [m for m in sys.modules if m == "psycopg" or m.startswith("psycopg.")]:
+            monkeypatch.delitem(sys.modules, name)
+        monkeypatch.setattr(sys, "meta_path", [NoLibpq(), *sys.meta_path])
+        monkeypatch.setenv("DATABASE_URL", "postgresql://u:p@localhost:5432/db")
+        with caplog.at_level("WARNING", logger="litellm_proxy_extras"):
+            assert ProxyExtrasDBManager.spend_logs_is_partitioned() is False
+        warning: Final = next(r.getMessage() for r in caplog.records if "partition check" in r.getMessage())
+        assert "no pq wrapper available" in warning
+        assert "libpq" in warning
+
 
 _ATTEMPT_BUDGET = 4