tkmind/tkmind_go
1
0
Fork 0
Code Issues Pull Requests Actions 4 Packages Projects Releases Wiki Activity

fix: Simplifed langfuse auth check (#177)

Browse Source
This commit is contained in:
Lifei Zhou
2024-10-22 14:39:58 +11:00
committed by GitHub
parent be612c5a19
commit 7749b8aaf4
3 changed files with 19 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files
packages/exchange/tests/test_langfuse_wrapper.py
+6 -4
View File
@@ -9,9 +9,10 @@ def mock_langfuse_context():
yield mock
@patch("exchange.langfuse_wrapper.HAS_LANGFUSE_CREDENTIALS", True)
def test_function_is_wrapped(mock_langfuse_context):
@patch("exchange.langfuse_wrapper.auth_check")
def test_function_is_wrapped(mock_auth_check, mock_langfuse_context):
mock_observe = MagicMock(side_effect=lambda *args, **kwargs: lambda fn: fn)
mock_auth_check.return_value = True
mock_langfuse_context.observe = mock_observe
def original_function(x: int, y: int) -> int:
@@ -31,9 +32,10 @@ def test_function_is_wrapped(mock_langfuse_context):
mock_observe.assert_called_with("arg1", kwarg1="kwarg1")
@patch("exchange.langfuse_wrapper.HAS_LANGFUSE_CREDENTIALS", False)
def test_function_is_not_wrapped(mock_langfuse_context):
@patch("exchange.langfuse_wrapper.auth_check")
def test_function_is_not_wrapped(mock_auth_check, mock_langfuse_context):
mock_observe = MagicMock(return_value=lambda f: f)
mock_auth_check.return_value = False
mock_langfuse_context.observe = mock_observe
@observe_wrapper("arg1", kwarg1="kwarg1")