add releases to callback

maxdeichmann · maxdeichmann · commit c1354f6fd067 · 2023-09-07T11:48:51.000+02:00
diff --git a/langfuse/callback.py b/langfuse/callback.py
@@ -32,6 +32,7 @@ def __init__(
         host: str = "https://cloud.langfuse.com",
         debug: bool = False,
         statefulTraceClient: Optional[StatefulTraceClient] = None,
+        release: Optional[str] = None,
     ) -> None:
         # If we're provided a stateful trace client directly
         if statefulTraceClient:
@@ -40,7 +41,7 @@ def __init__(
 
         # Otherwise, initialize stateless using the provided keys
         elif public_key and secret_key:
-            self.langfuse = Langfuse(public_key, secret_key, host, debug=debug)
+            self.langfuse = Langfuse(public_key, secret_key, host, debug=debug, release=release)
             self.trace = None
             self.runs = {}
             if debug:
diff --git a/tests/test_langchain.py b/tests/test_langchain.py
@@ -37,6 +37,13 @@ def test_langfuse_init():
     assert not callback.runs
 
 
+def test_langfuse_release_init():
+    callback = CallbackHandler(
+        os.environ.get("LF_PK"), os.environ.get("LF_SK"), os.environ.get("HOST"), release="something"
+    )
+    assert callback.langfuse.release == "something"
+
+
 @pytest.mark.skip(reason="inference cost")
 def test_callback_generated_from_trace():
     api_wrapper = LangfuseAPI(os.environ.get("LF_PK"), os.environ.get("LF_SK"), os.environ.get("HOST"))
