refactor: move langfuse wrapper to a module in exchange instead of a package (#138)

Co-authored-by: Alice Hau <ahau@squareup.com>
2024-10-16 09:30:13 -04:00
parent 4cdc1004c3
commit 8cf7b9f26c
22 changed files with 392 additions and 7 deletions
@@ -34,7 +34,9 @@ def test_session_start_command_with_session_name(mock_session):
    mock_session_class, mock_session_instance = mock_session
    runner = CliRunner()
    runner.invoke(goose_cli, ["session", "start", "session1", "--profile", "default"])
-    mock_session_class.assert_called_once_with(name="session1", profile="default", plan=None, log_level="INFO")
+    mock_session_class.assert_called_once_with(
+        name="session1", profile="default", plan=None, log_level="INFO", tracing=False
+    )
    mock_session_instance.run.assert_called_once()