cleanup

langfuse/client.py

@@ -17,6 +17,7 @@
 from langfuse.api.resources.score.types.create_score_request import CreateScoreRequest
 from langfuse.api.resources.trace.types.create_trace_request import CreateTraceRequest
 from langfuse.environment import get_common_release_envs
+from langfuse.logging import clean_logger
 from langfuse.model import (
     DatasetItem,
     CreateDatasetRunItemRequest,
@@ -58,16 +59,10 @@ def __init__(
             logging.basicConfig()
             self.log.setLevel(logging.DEBUG)
 
-            httpx_logger = logging.getLogger("httpx")
-            httpx_logger.setLevel(logging.WARNING)  # Set the desired log level
-            console_handler = logging.StreamHandler()
-            httpx_logger.addHandler(console_handler)
+            clean_logger()
         else:
             self.log.setLevel(logging.WARNING)
-            httpx_logger = logging.getLogger("httpx")
-            httpx_logger.setLevel(logging.WARNING)  # Set the desired log level
-            console_handler = logging.StreamHandler()
-            httpx_logger.addHandler(console_handler)
+            clean_logger()
 
         self.task_manager = TaskManager(debug=debug)
 

langfuse/logging.py

@@ -0,0 +1,12 @@
+import logging
+
+
+def clean_logger():
+    httpx_logger = logging.getLogger("httpx")
+    httpx_logger.setLevel(logging.WARNING)  # Set the desired log level
+    console_handler = logging.StreamHandler()
+    httpx_logger.addHandler(console_handler)
+
+    backoff_logger = logging.getLogger("backoff")
+    backoff_logger.setLevel(logging.WARNING)  # Set the desired log level
+    backoff_logger.addHandler(console_handler)

langfuse/task_manager.py

@@ -7,6 +7,8 @@
 
 import backoff
 
+from langfuse.logging import clean_logger
+
 
 class Task:
     def __init__(self, task_id, function, predecessor_id: str = None):
@@ -40,8 +42,10 @@ def __init__(self, debug=False, max_task_queue_size=10_000, max_task_age=600):
             # See https://docs.python.org/3/howto/logging.html#what-happens-if-no-configuration-is-provided
             logging.basicConfig()
             self.log.setLevel(logging.DEBUG)
+            clean_logger()
         else:
             self.log.setLevel(logging.WARNING)
+            clean_logger()
 
         self.init_resources()
 

tests/test_core_sdk.py

@@ -21,7 +21,7 @@
 
 def test_flush():
     # set up the consumer with more requests than a single batch will allow
-    langfuse = Langfuse(debug=True)
+    langfuse = Langfuse(debug=False)
 
     for i in range(2):
         langfuse.trace(
@@ -36,7 +36,7 @@ def test_flush():
 
 
 def test_shutdown():
-    langfuse = Langfuse(debug=True)
+    langfuse = Langfuse(debug=False)
 
     for i in range(2):
         langfuse.trace(
@@ -54,7 +54,7 @@ def test_shutdown():
 
 
 def test_create_score():
-    langfuse = Langfuse(debug=True)
+    langfuse = Langfuse(debug=False)
     api_wrapper = LangfuseAPI()
 
     trace = langfuse.trace(
@@ -97,7 +97,7 @@ def test_create_score():
 
 
 def test_create_trace():
-    langfuse = Langfuse(debug=True)
+    langfuse = Langfuse(debug=False)
     api_wrapper = LangfuseAPI()
     trace_name = create_uuid()
 
@@ -120,7 +120,7 @@ def test_create_trace():
 
 
 def test_create_generation():
-    langfuse = Langfuse(debug=True)
+    langfuse = Langfuse(debug=False)
     api_wrapper = LangfuseAPI()
 
     timestamp = datetime.now()
@@ -181,7 +181,7 @@ def test_create_generation():
 
 
 def test_create_span():
-    langfuse = Langfuse(debug=True)
+    langfuse = Langfuse(debug=False)
     api_wrapper = LangfuseAPI()
 
     timestamp = datetime.now()
@@ -225,7 +225,7 @@ def test_create_span():
 
 
 def test_score_trace():
-    langfuse = Langfuse(debug=True)
+    langfuse = Langfuse(debug=False)
     api_wrapper = LangfuseAPI()
 
     trace_name = create_uuid()
@@ -262,7 +262,7 @@ def test_score_trace():
 
 
 def test_score_span():
-    langfuse = Langfuse(debug=True)
+    langfuse = Langfuse(debug=False)
     api_wrapper = LangfuseAPI()
 
     spanId = create_uuid()
@@ -309,7 +309,7 @@ def test_score_span():
 
 
 def test_create_trace_and_span():
-    langfuse = Langfuse(debug=True)
+    langfuse = Langfuse(debug=False)
     api_wrapper = LangfuseAPI()
 
     trace_name = create_uuid()
@@ -331,7 +331,7 @@ def test_create_trace_and_span():
 
 
 def test_create_trace_and_generation():
-    langfuse = Langfuse(debug=True)
+    langfuse = Langfuse(debug=False)
     api_wrapper = LangfuseAPI()
 
     trace_name = create_uuid()
@@ -353,7 +353,7 @@ def test_create_trace_and_generation():
 
 
 def test_create_generation_and_trace():
-    langfuse = Langfuse(debug=True)
+    langfuse = Langfuse(debug=False)
     api_wrapper = LangfuseAPI()
 
     trace_name = create_uuid()
@@ -375,7 +375,7 @@ def test_create_generation_and_trace():
 
 
 def test_update_generation():
-    langfuse = Langfuse(debug=True)
+    langfuse = Langfuse(debug=False)
     api = get_api()
 
     generation = langfuse.generation(InitialGeneration(name="generation"))
@@ -395,7 +395,7 @@ def test_update_generation():
 
 
 def test_update_span():
-    langfuse = Langfuse(debug=True)
+    langfuse = Langfuse(debug=False)
     api = get_api()
 
     span = langfuse.span(InitialSpan(name="span"))
@@ -415,7 +415,7 @@ def test_update_span():
 
 
 def test_create_trace_and_event():
-    langfuse = Langfuse(debug=True)
+    langfuse = Langfuse(debug=False)
     api_wrapper = LangfuseAPI()
 
     trace_name = create_uuid()
@@ -439,7 +439,7 @@ def test_create_trace_and_event():
 def test_create_span_and_generation():
     api = get_api()
 
-    langfuse = Langfuse(debug=True)
+    langfuse = Langfuse(debug=False)
 
     span = langfuse.span(InitialSpan(name="span"))
     langfuse.generation(InitialGeneration(traceId=span.trace_id, name="generation"))
@@ -459,7 +459,7 @@ def test_create_span_and_generation():
 
 
 def test_create_trace_with_id_and_generation():
-    langfuse = Langfuse(debug=True)
+    langfuse = Langfuse(debug=False)
     api_wrapper = LangfuseAPI()
 
     trace_name = create_uuid()
@@ -549,7 +549,7 @@ def test_end_span():
 
 
 def test_get_generations():
-    langfuse = Langfuse(debug=True)
+    langfuse = Langfuse(debug=False)
 
     timestamp = datetime.now()
 
@@ -582,7 +582,7 @@ def test_get_generations():
 
 
 def test_get_generations_by_user():
-    langfuse = Langfuse(debug=True)
+    langfuse = Langfuse(debug=False)
 
     timestamp = datetime.now()
 

tests/test_datasets.py

@@ -16,7 +16,7 @@
 
 
 def test_create_and_get_dataset():
-    langfuse = Langfuse(debug=True)
+    langfuse = Langfuse(debug=False)
 
     name = create_uuid()
     langfuse.create_dataset(CreateDatasetRequest(name=name))
@@ -25,7 +25,7 @@ def test_create_and_get_dataset():
 
 
 def test_create_dataset_item():
-    langfuse = Langfuse(debug=True)
+    langfuse = Langfuse(debug=False)
     name = create_uuid()
     langfuse.create_dataset(CreateDatasetRequest(name=name))
 
@@ -38,7 +38,7 @@ def test_create_dataset_item():
 
 
 def test_linking_observation():
-    langfuse = Langfuse(debug=True)
+    langfuse = Langfuse(debug=False)
 
     dataset_name = create_uuid()
     langfuse.create_dataset(CreateDatasetRequest(name=dataset_name))
@@ -66,7 +66,7 @@ def test_linking_observation():
 
 
 def test_linking_via_id_observation():
-    langfuse = Langfuse(debug=True)
+    langfuse = Langfuse(debug=False)
 
     dataset_name = create_uuid()
     langfuse.create_dataset(CreateDatasetRequest(name=dataset_name))
@@ -96,7 +96,7 @@ def test_linking_via_id_observation():
 
 @pytest.mark.skip(reason="inference cost")
 def test_langchain_dataset():
-    langfuse = Langfuse(debug=True)
+    langfuse = Langfuse(debug=False)
     dataset_name = create_uuid()
     langfuse.create_dataset(CreateDatasetRequest(name=dataset_name))