fix: use original private stored log method in flow decorator

Author: AlePouroullis

src/humanloop/decorators/flow.py

@@ -67,7 +67,7 @@ def wrapper(*args: P.args, **kwargs: P.kwargs) -> Optional[R]:
                         "messages": func_args.get("messages"),
                         "trace_parent_id": trace_id,
                     }
-                    this_flow_log: FlowLogResponse = client.flows.log(
+                    this_flow_log: FlowLogResponse = client.flows._log(  # type: ignore [attr-defined]
                         path=decorator_context.path,
                         flow=decorator_context.version,
                         log_status="incomplete",

src/humanloop/overload.py

@@ -181,7 +181,7 @@ def overload_client(
     use_local_files: bool = False,
 ) -> Any:
     """Overloads client methods to add tracing, local file handling, and evaluation context."""
-    # Store original log method as _log for all clients
+    # Store original log method as _log for all clients. Used in flow decorator 
     if hasattr(client, "log") and not hasattr(client, "_log"):
         client._log = client.log  # type: ignore [attr-defined]
 

tests/custom/sync/test_client.py

@@ -4,6 +4,7 @@
 from unittest.mock import Mock, patch
 from humanloop.sync.sync_client import SyncClient
 from humanloop.error import HumanloopRuntimeError
+from typing import Literal
 
 
 @pytest.fixture
@@ -87,8 +88,8 @@ def test_cache_functionality(sync_client: SyncClient):
     # Save a test file
     content = "test content"
     path = "test/path"
-    file_type = "prompt"
-    sync_client._save_serialized_file(content, path, file_type)  # type: ignore [arg-type] Ignore because we're deliberately testing an invalid literal
+    file_type: Literal["prompt", "agent"] = "prompt"
+    sync_client._save_serialized_file(content, path, file_type)
 
     # First read should hit disk
     sync_client.get_file_content(path, file_type)