add param

Author: nina-kollman

packages/traceloop-sdk/traceloop/sdk/__init__.py

@@ -48,7 +48,7 @@ class Traceloop:
     @staticmethod
     def init(
         app_name: str = sys.argv[0],
-        api_endpoint: Optional[str] = None,
+        api_endpoint: str = "https://api.traceloop.com",
         api_key: Optional[str] = None,
         enabled: bool = True,
         headers: Dict[str, str] = {},
@@ -69,6 +69,7 @@ def init(
         block_instruments: Optional[Set[Instruments]] = None,
         image_uploader: Optional[ImageUploader] = None,
         span_postprocess_callback: Optional[Callable[[ReadableSpan], None]] = None,
+        traceloop_backend: Optional[bool] = False,
     ) -> Optional[Client]:
         if not enabled:
             TracerWrapper.set_disabled(True)
@@ -79,8 +80,8 @@ def init(
             )
             return
 
-        api_endpoint = api_endpoint or os.getenv("TRACELOOP_BASE_URL") or "https://api.traceloop.com"
-        api_key = api_key or os.getenv("TRACELOOP_API_KEY")
+        api_endpoint = os.getenv("TRACELOOP_BASE_URL") or api_endpoint
+        api_key = os.getenv("TRACELOOP_API_KEY") or api_key
         Traceloop.__app_name = app_name
 
         if not is_tracing_enabled():
@@ -180,7 +181,7 @@ def init(
             Traceloop.__logger_wrapper = LoggerWrapper(exporter=logging_exporter)
 
         if (
-            api_endpoint.find("traceloop") != -1
+            (api_endpoint.find("traceloop.com") != -1 or traceloop_backend)
             and api_key
             and (exporter is None)
             and (processor is None)