test: fix type errors in local file operations test by using proper ChatMessageParams type

Author: AlePouroullis

src/humanloop/cli/__main__.py

@@ -81,7 +81,6 @@ def get_client(
     """
     if not api_key:
         api_key = load_api_key(env_file)
-    print(api_key)
     return Humanloop(api_key=api_key, base_url=base_url)
 
 

src/humanloop/client.py

@@ -1,36 +1,34 @@
+import logging
 import os
 import typing
 from typing import Any, List, Optional, Sequence, Tuple
-import logging
 
 import httpx
 from opentelemetry.sdk.resources import Resource
 from opentelemetry.sdk.trace import TracerProvider
 from opentelemetry.trace import Tracer
 
+from humanloop.base_client import AsyncBaseHumanloop, BaseHumanloop
 from humanloop.core.client_wrapper import SyncClientWrapper
-
+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
+from humanloop.environment import HumanloopEnvironment
 from humanloop.evals import run_eval
 from humanloop.evals.types import (
     DatasetEvalConfig,
-    EvaluatorEvalConfig,
     EvaluatorCheck,
+    EvaluatorEvalConfig,
     FileEvalConfig,
 )
-
-from humanloop.base_client import AsyncBaseHumanloop, BaseHumanloop
-from humanloop.overload import overload_client
-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
-from humanloop.environment import HumanloopEnvironment
 from humanloop.evaluations.client import EvaluationsClient
 from humanloop.otel import instrument_provider
 from humanloop.otel.exporter import HumanloopSpanExporter
 from humanloop.otel.processor import HumanloopSpanProcessor
+from humanloop.overload import overload_client
 from humanloop.prompt_utils import populate_template
 from humanloop.prompts.client import PromptsClient
-from humanloop.sync.sync_client import SyncClient, DEFAULT_CACHE_SIZE
+from humanloop.sync.sync_client import DEFAULT_CACHE_SIZE, SyncClient
 
 logger = logging.getLogger("humanloop.sdk")
 
@@ -431,11 +429,6 @@ def pull(self, path: Optional[str] = None, environment: Optional[str] = None) ->
 
         If you specify `local_files_directory="data/humanloop"`, files will be saved in ./data/humanloop/ instead.
 
-        Important note about paths:
-        - Paths should not contain leading or trailing slashes
-        - When specifying a file path, include the file extension (`.prompt` or `.agent`)
-        - When specifying a directory path, do not include a trailing slash
-
         :param path: Optional path to either a specific file (e.g. "path/to/file.prompt") or a directory (e.g. "path/to/directory").
                     If not provided, all Prompt and Agent files will be pulled.
         :param environment: The environment to pull the files from.

src/humanloop/path_utils.py

@@ -1,17 +1,11 @@
 from pathlib import Path
 
-from humanloop.error import HumanloopRuntimeError
-
 
 def normalize_path(path: str, strip_extension: bool = False) -> str:
     """Normalize a path to the standard API format: path/to/resource, where resource can be a file or a directory."""
     # Remove leading/trailing slashes
     path_obj = Path(path.strip("/"))
 
-    # Check if path is absolute
-    if path_obj.is_absolute():
-        raise HumanloopRuntimeError("Absolute paths are not supported. Ensure the ")
-
     # Handle extension
     if strip_extension:
         normalized = str(path_obj.with_suffix(""))

tests/custom/integration/test_local_file_operations.py

@@ -3,6 +3,8 @@
 import pytest
 
 from humanloop.error import HumanloopRuntimeError
+from humanloop.requests.chat_message import ChatMessageParams
+from humanloop.types.chat_role import ChatRole
 from tests.custom.types import GetHumanloopClientFn, SyncableFile
 
 
@@ -99,17 +101,16 @@ def test_local_file_call(
     client = get_humanloop_client(use_local_files=True, local_files_directory=str(tmp_path))
 
     # WHEN calling the API with the local file path (without extension)
-    response = client.prompts.call(
-        path=test_path, messages=[{"role": "user", "content": "What is the capital of France?"}]
-    )
+    call_messages = [ChatMessageParams(role="user", content="What is the capital of France?")]
+    response = client.prompts.call(path=test_path, messages=call_messages)
 
     # THEN the response should be successful
     assert response is not None
     assert response.logs is not None
     assert len(response.logs) > 0
 
     # AND the response should contain the expected output format (lowercase city name)
-    assert "paris" in response.logs[0].output.lower()
+    assert response.logs[0].output is not None and "paris" in response.logs[0].output.lower()
 
     # AND the prompt used should match our expected path
     assert response.prompt is not None
@@ -150,11 +151,11 @@ def test_local_file_log(
     client = get_humanloop_client(use_local_files=True, local_files_directory=str(tmp_path))
 
     # GIVEN message content to log
-    test_messages = [{"role": "user", "content": "What is the capital of France?"}]
     test_output = "Paris is the capital of France."
 
     # WHEN logging the data with the local file path
-    response = client.prompts.log(path=test_path, messages=test_messages, output=test_output)
+    messages = [ChatMessageParams(role="user", content="What is the capital of France?")]
+    response = client.prompts.log(path=test_path, messages=messages, output=test_output)
 
     # THEN the log should be successful
     assert response is not None
@@ -179,6 +180,8 @@ def test_overload_version_environment_handling(
     humanloop_client = get_humanloop_client(use_local_files=True, local_files_directory=str(tmp_path))
     humanloop_client.pull()
 
+    test_message = [ChatMessageParams(role="user", content="Testing")]
+
     # GIVEN a test file that exists locally
     test_file = syncable_files_fixture[0]
     extension = f".{test_file.type}"
@@ -195,13 +198,13 @@ def test_overload_version_environment_handling(
             humanloop_client.prompts.call(
                 path=test_file.path,
                 version_id=test_file.version_id,
-                messages=[{"role": "user", "content": "Testing"}],
+                messages=test_message,
             )
         elif test_file.type == "agent":
             humanloop_client.agents.call(
                 path=test_file.path,
                 version_id=test_file.version_id,
-                messages=[{"role": "user", "content": "Testing"}],
+                messages=test_message,
             )
 
     # WHEN calling with environment
@@ -211,13 +214,13 @@ def test_overload_version_environment_handling(
             humanloop_client.prompts.call(
                 path=test_file.path,
                 environment="production",
-                messages=[{"role": "user", "content": "Testing"}],
+                messages=test_message,
             )
         elif test_file.type == "agent":
             humanloop_client.agents.call(
                 path=test_file.path,
                 environment="production",
-                messages=[{"role": "user", "content": "Testing"}],
+                messages=test_message,
             )
 
     # WHEN calling with both version_id and environment
@@ -228,14 +231,14 @@ def test_overload_version_environment_handling(
                 path=test_file.path,
                 version_id=test_file.version_id,
                 environment="staging",
-                messages=[{"role": "user", "content": "Testing"}],
+                messages=test_message,
             )
         elif test_file.type == "agent":
             humanloop_client.agents.call(
                 path=test_file.path,
                 version_id=test_file.version_id,
                 environment="staging",
-                messages=[{"role": "user", "content": "Testing"}],
+                messages=test_message,
             )
 
 
@@ -280,8 +283,9 @@ def test_overload_version_environment_handling(
 #     client = get_humanloop_client(use_local_files=True, local_files_directory=str(tmp_path))
 #
 #     # WHEN calling the API with the local file path (without extension)
+#     agent_call_messages = [ChatMessageParams(role="user", content="What is the capital of France?")]
 #     response = client.agents.call(
-#         path=test_path, messages=[{"role": "user", "content": "What is the capital of France?"}]
+#         path=test_path, messages=agent_call_messages
 #     )
 #
 #     # THEN the response should be successful
@@ -290,16 +294,16 @@ def test_overload_version_environment_handling(
 #     assert len(response.logs) > 0
 #
 #     # AND the response should contain the expected output format (lowercase city name)
-#     assert "paris" in response.logs[0].output.lower()
+#     assert response.logs[0].output is not None and "paris" in response.logs[0].output.lower()
 #
 #     # AND the agent used should match our expected path
 #     assert response.agent is not None
 #     assert response.agent.path == test_path
 #
 #     # WHEN logging with the local agent file
-#     test_messages = [{"role": "user", "content": "What is the capital of Germany?"}]
 #     test_output = "Berlin is the capital of Germany."
-#     log_response = client.agents.log(path=test_path, messages=test_messages, output=test_output)
+#     agent_messages = [ChatMessageParams(role="user", content="What is the capital of Germany?")]
+#     log_response = client.agents.log(path=test_path, messages=agent_messages, output=test_output)
 #
 #     # THEN the log should be successful
 #     assert log_response is not None