add client overloads for call method to use local files at specified path for agents and prompts

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
+from humanloop.overload import overload_call, overload_log, overload_call_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
@@ -99,6 +99,7 @@ def __init__(
         httpx_client: typing.Optional[httpx.Client] = None,
         opentelemetry_tracer_provider: Optional[TracerProvider] = None,
         opentelemetry_tracer: Optional[Tracer] = None,
+        use_local_files: bool = False,
     ):
         """
         Extends the base client with custom evaluation utilities and
@@ -118,6 +119,7 @@ def __init__(
             httpx_client=httpx_client,
         )
 
+        self.use_local_files = use_local_files
         self.sync_client = SyncClient(client=self)
         eval_client = ExtendedEvalsClient(client_wrapper=self._client_wrapper)
         eval_client.client = self
@@ -128,6 +130,16 @@ 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(
+            client=self.prompts, 
+            use_local_files=self.use_local_files, 
+            file_type="prompt"
+        )
+        self.agents = overload_call_with_local_files(
+            client=self.agents, 
+            use_local_files=self.use_local_files, 
+            file_type="agent"
+        )
         self.flows = overload_log(client=self.flows)
         self.tools = overload_log(client=self.tools)
 

src/humanloop/overload.py

@@ -1,8 +1,8 @@
 import inspect
 import logging
 import types
-from typing import TypeVar, Union
-
+from typing import TypeVar, Union, Literal
+from pathlib import Path
 from humanloop.context import (
     get_decorator_context,
     get_evaluation_context,
@@ -13,6 +13,7 @@
 from humanloop.evaluators.client import EvaluatorsClient
 from humanloop.flows.client import FlowsClient
 from humanloop.prompts.client import PromptsClient
+from humanloop.agents.client import AgentsClient
 from humanloop.tools.client import ToolsClient
 from humanloop.types.create_evaluator_log_response import CreateEvaluatorLogResponse
 from humanloop.types.create_flow_log_response import CreateFlowLogResponse
@@ -112,6 +113,7 @@ def _overload_call(self, **kwargs) -> PromptCallResponse:
             }
 
         try:
+            logger.info(f"Calling inner overload")
             response = self._call(**kwargs)
         except Exception as e:
             # Re-raising as HumanloopDecoratorError so the decorators don't catch it
@@ -122,3 +124,43 @@ def _overload_call(self, **kwargs) -> PromptCallResponse:
     # Replace the original log method with the overloaded one
     client.call = types.MethodType(_overload_call, client)  # type: ignore [assignment]
     return client
+
+def overload_call_with_local_files(
+    client: Union[PromptsClient, AgentsClient], 
+    use_local_files: bool,
+    file_type: Literal["prompt", "agent"]
+) -> Union[PromptsClient, AgentsClient]:
+    """Overload call to handle local files when use_local_files is True.
+    
+    Args:
+        client: The client to overload (PromptsClient or AgentsClient)
+        use_local_files: Whether to use local files
+        file_type: Type of file ("prompt" or "agent")
+    """
+    original_call = client._call if hasattr(client, '_call') else client.call
+
+    def _overload_call(self, **kwargs) -> PromptCallResponse:
+        if use_local_files and "path" in kwargs:
+            try:
+                # Construct path to local file
+                local_path = Path("humanloop") / kwargs["path"]
+                # Add appropriate extension
+                local_path = local_path.parent / f"{local_path.stem}.{file_type}"
+                
+                if local_path.exists():
+                    # Read the file content
+                    with open(local_path) as f:
+                        file_content = f.read()
+                    
+                    kwargs[file_type] = file_content  # "prompt" or "agent"
+                    
+                    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)
+
+    client.call = types.MethodType(_overload_call, client)
+    return client

tests/sync/test_sync.py

@@ -74,10 +74,10 @@ def cleanup_local_files():
         shutil.rmtree(local_dir)
 
 
-def test_sync_basic(humanloop_client: Humanloop, test_file_structure: List[SyncableFile], cleanup_local_files):
+def test_pull_basic(humanloop_client: Humanloop, test_file_structure: List[SyncableFile], cleanup_local_files):
     """Test that humanloop.sync() correctly syncs remote files to local filesystem"""
     # Run the sync
-    successful_files = humanloop_client.sync()
+    successful_files = humanloop_client.pull()
 
     # Verify each file was synced correctly
     for file in test_file_structure: