overload log operation to use local files when relevant

Author: AlePouroullis

src/humanloop/client.py

@@ -13,7 +13,7 @@
 from humanloop.evals.types import Dataset, Evaluator, EvaluatorCheck, File
 
 from humanloop.base_client import AsyncBaseHumanloop, BaseHumanloop
-from humanloop.overload import overload_call, overload_log, overload_call_with_local_files
+from humanloop.overload import overload_call, overload_log, overload_with_local_files
 from humanloop.decorators.flow import flow as flow_decorator_factory
 from humanloop.decorators.prompt import prompt_decorator_factory
 from humanloop.decorators.tool import tool_decorator_factory as tool_decorator_factory
@@ -77,6 +77,7 @@ class ExtendedPromptsClient(PromptsClient):
     """
 
     populate_template = staticmethod(populate_template)  # type: ignore [assignment]
+    load = staticmethod(load)
 
 
 class Humanloop(BaseHumanloop):
@@ -130,11 +131,11 @@ def __init__(
         # and the @flow decorator providing the trace_id
         self.prompts = overload_log(client=self.prompts)
         self.prompts = overload_call(client=self.prompts)
-        self.prompts = overload_call_with_local_files(
+        self.prompts = overload_with_local_files(
             client=self.prompts, 
             use_local_files=self.use_local_files
         )
-        self.agents = overload_call_with_local_files(
+        self.agents = overload_with_local_files(
             client=self.agents, 
             use_local_files=self.use_local_files
         )
@@ -393,4 +394,4 @@ class AsyncHumanloop(AsyncBaseHumanloop):
     TODO: Add custom evaluation utilities for async case.
     """
 
-    pass
+    pass

src/humanloop/overload.py

@@ -126,7 +126,7 @@ def _overload_call(self, **kwargs) -> PromptCallResponse:
     client.call = types.MethodType(_overload_call, client)  # type: ignore [assignment]
     return client
 
-def overload_call_with_local_files(
+def overload_with_local_files(
     client: Union[PromptsClient, AgentsClient], 
     use_local_files: bool,
 ) -> Union[PromptsClient, AgentsClient]:
@@ -138,6 +138,7 @@ def overload_call_with_local_files(
         file_type: Type of file ("prompt" or "agent")
     """
     original_call = client._call if hasattr(client, '_call') else client.call
+    original_log = client._log if hasattr(client, '_log') else client.log
     # get file type from client type
     file_type: FileType
     if isinstance(client, PromptsClient):   
@@ -147,11 +148,11 @@ def overload_call_with_local_files(
     else:
         raise ValueError(f"Unsupported client type: {type(client)}")
 
-    def _overload_call(self, **kwargs) -> PromptCallResponse:
+    def _overload(self, function_name: str, **kwargs) -> PromptCallResponse:
         if use_local_files and "path" in kwargs:
             try:
                 # Construct path to local file
-                local_path = Path("humanloop") / kwargs["path"]
+                local_path = Path("humanloop") / kwargs["path"] # FLAG: ensure that when passing the path back to remote, it's using forward slashes
                 # Add appropriate extension
                 local_path = local_path.parent / f"{local_path.stem}.{file_type}"
 
@@ -160,15 +161,27 @@ def _overload_call(self, **kwargs) -> PromptCallResponse:
                     with open(local_path) as f:
                         file_content = f.read()
 
-                    kwargs[file_type] = file_content  # "prompt" or "agent" # TODO: raise warning if kernel passed in
+                    kwargs[file_type] = file_content  # "prompt" or "agent" # TODO: raise warning if kernel passed in
 
                     logger.debug(f"Using local file content from {local_path}")
                 else:
                     logger.warning(f"Local file not found: {local_path}, falling back to API")
             except Exception as e:
                 logger.error(f"Error reading local file: {e}, falling back to API")
 
-        return original_call(**kwargs)
+        if function_name == "call":
+            return original_call(**kwargs)
+        elif function_name == "log":
+            return original_log(**kwargs)
+        else:
+            raise ValueError(f"Unsupported function name: {function_name}")
+
+    def _overload_call(self, **kwargs) -> PromptCallResponse:
+        return _overload(self, "call", **kwargs)
+
+    def _overload_log(self, **kwargs) -> PromptCallResponse:
+        return _overload(self, "log", **kwargs)
 
     client.call = types.MethodType(_overload_call, client)
+    client.log = types.MethodType(_overload_log, client)
     return client

tests/conftest.py

@@ -194,7 +194,7 @@ def api_keys() -> APIKeys:
 def humanloop_client(api_keys: APIKeys) -> Humanloop:
     return Humanloop(
         api_key=api_keys.humanloop,
-        base_url="https://neostaging.humanloop.ml/v5/",
+        base_url="http://localhost:80/v5/",
     )