feat: generalize sdk_integration to string (#282)

Author: marcklingen

langfuse/callback.py

@@ -10,7 +10,6 @@
 from langfuse.api.resources.ingestion.types.sdk_log_event import SdkLogEvent
 from langfuse.client import (
     Langfuse,
-    SDKIntegrationTypes,
     StatefulSpanClient,
     StatefulTraceClient,
     StateType,
@@ -124,7 +123,7 @@ def __init__(
             if timeout is not None:
                 args["timeout"] = timeout
 
-            args["sdk_integration"] = SDKIntegrationTypes.LANGCHAIN
+            args["sdk_integration"] = "langchain"
 
             self.langfuse = Langfuse(**args)
             self.trace = None

langfuse/client.py

@@ -68,7 +68,7 @@ def __init__(
         flush_interval: int = 0.5,
         max_retries=3,
         timeout=15,
-        sdk_integration: SDKIntegrationTypes = SDKIntegrationTypes.DEFAULT,
+        sdk_integration: str = "default",
     ):
         set_debug = debug if debug else (os.getenv("LANGFUSE_DEBUG", "False") == "True")
 
@@ -130,7 +130,7 @@ def __init__(
             "public_key": public_key,
             "sdk_name": "python",
             "sdk_version": version,
-            "sdk_integration": sdk_integration.value,
+            "sdk_integration": sdk_integration,
         }
 
         if threads is not None:

langfuse/openai.py

@@ -9,7 +9,7 @@
 from wrapt import wrap_function_wrapper
 
 from langfuse import Langfuse
-from langfuse.client import SDKIntegrationTypes, StatefulGenerationClient
+from langfuse.client import StatefulGenerationClient
 from langfuse.utils import _get_timestamp
 
 log = logging.getLogger("langfuse")
@@ -435,7 +435,7 @@ def initialize(self):
                         secret_key=openai.langfuse_secret_key,
                         host=openai.langfuse_host,
                         debug=openai.langfuse_debug,
-                        sdk_integration=SDKIntegrationTypes.OPENAI,
+                        sdk_integration="openai",
                     )
         return self._langfuse