From 4ef0f1b247fcb7d9cae604a6df71f7056272c1d1 Mon Sep 17 00:00:00 2001
From: Alexsander Hamir <alexsanderhamirgomesbaptista@gmail.com>
Date: Fri, 23 Jan 2026 15:26:48 -0800
Subject: [PATCH] Fix Langfuse mock client: use proper URL parsing and remove
 unnecessary send patching

---
 .../langfuse/langfuse_mock_client.py          | 23 +++++++++++++++----
 1 file changed, 18 insertions(+), 5 deletions(-)

diff --git a/litellm/integrations/langfuse/langfuse_mock_client.py b/litellm/integrations/langfuse/langfuse_mock_client.py
index 8b79f951829..27a1a886878 100644
--- a/litellm/integrations/langfuse/langfuse_mock_client.py
+++ b/litellm/integrations/langfuse/langfuse_mock_client.py
@@ -51,16 +51,29 @@ class MockLangfuseResponse:
             raise Exception(f"HTTP {self.status_code}")
 
 
+def _is_langfuse_url(url) -> bool:
+    """Check if URL is a Langfuse domain."""
+    try:
+        parsed_url = httpx.URL(url) if isinstance(url, str) else url
+        hostname = parsed_url.host or ""
+        
+        return (
+            hostname.endswith(".langfuse.com") or
+            hostname == "langfuse.com" or
+            (hostname in ("localhost", "127.0.0.1") and "langfuse" in str(parsed_url).lower())
+        )
+    except Exception:
+        return False
+
+
 def _mock_httpx_post(self, url, **kwargs):
     """Monkey-patched httpx.Client.post that intercepts Langfuse calls."""
-    if isinstance(url, str) and ("langfuse.com" in url or "langfuse" in url.lower()):
-        verbose_logger.debug(f"[LANGFUSE MOCK] POST to {url}")
+    if _is_langfuse_url(url):
+        verbose_logger.info(f"[LANGFUSE MOCK] POST to {url}")
         return MockLangfuseResponse(status_code=200, json_data={"status": "success"}, url=url)
+    
     if _original_httpx_post is not None:
         return _original_httpx_post(self, url, **kwargs)
-    import httpx
-    with httpx.Client() as client:
-        return client.post(url, **kwargs)
 
 
 def create_mock_langfuse_client():