import

Author: Andrei Bratu

src/humanloop/decorators/tool.py

@@ -12,12 +12,17 @@
 
 from humanloop.eval_utils import File
 from humanloop.otel import TRACE_FLOW_CONTEXT, FlowContext
-from humanloop.otel.constants import HUMANLOOP_FILE_KEY, HUMANLOOP_FILE_TYPE_KEY, HUMANLOOP_LOG_KEY, HUMANLOOP_PATH_KEY
+from humanloop.otel.constants import (
+    HUMANLOOP_FILE_KEY,
+    HUMANLOOP_FILE_TYPE_KEY,
+    HUMANLOOP_LOG_KEY,
+    HUMANLOOP_PATH_KEY,
+)
 from humanloop.otel.helpers import generate_span_id, write_to_opentelemetry_span
 from humanloop.requests.tool_function import ToolFunctionParams
 from humanloop.requests.tool_kernel_request import ToolKernelRequestParams
 
-from humanloop.otel.helpers import args_to_inputs
+from humanloop.decorators.helpers import args_to_inputs
 
 logger = logging.getLogger("humanloop.sdk")
 
@@ -163,7 +168,9 @@ def _build_function_parameters_property(func) -> _JSONSchemaFunctionParameters:
             inspect.Parameter.VAR_POSITIONAL,
             inspect.Parameter.VAR_KEYWORD,
         ):
-            raise ValueError(f"{func.__name__}: *args and **kwargs are not supported by the @tool decorator")
+            raise ValueError(
+                f"{func.__name__}: *args and **kwargs are not supported by the @tool decorator"
+            )
 
     for parameter in signature.parameters.values():
         try:
@@ -267,7 +274,16 @@ def _parse_annotation(annotation: typing.Type) -> _ParsedAnnotation:
     if origin is None:
         # Either not a nested type or no type hint
         # Parameter.empty is used for parameters without type hints
-        if annotation not in (str, int, float, bool, Parameter.empty, dict, list, tuple):
+        if annotation not in (
+            str,
+            int,
+            float,
+            bool,
+            Parameter.empty,
+            dict,
+            list,
+            tuple,
+        ):
             raise ValueError(f"Unsupported type hint: {annotation}")
 
         # Check if it's a complex type with no inner type
@@ -321,7 +337,9 @@ def _parse_annotation(annotation: typing.Type) -> _ParsedAnnotation:
             # Union has sub_types and is Optional
             return _ParsedOptionalAnnotation(
                 annotation=_ParsedUnionAnnotation(
-                    annotation=[_parse_annotation(sub_type) for sub_type in sub_types[:-1]],
+                    annotation=[
+                        _parse_annotation(sub_type) for sub_type in sub_types[:-1]
+                    ],
                 )
             )
         # Union type that is not Optional
@@ -355,7 +373,10 @@ def _annotation_parse_to_json_schema(
 
     if isinstance(arg, _ParsedUnionAnnotation):
         arg_type = {
-            "anyOf": [_annotation_parse_to_json_schema(sub_type) for sub_type in arg.annotation],
+            "anyOf": [
+                _annotation_parse_to_json_schema(sub_type)
+                for sub_type in arg.annotation
+            ],
         }
 
     elif isinstance(arg, _ParsedTupleAnnotation):
@@ -370,7 +391,10 @@ def _annotation_parse_to_json_schema(
         else:
             arg_type = {
                 "type": "array",
-                "items": [_annotation_parse_to_json_schema(sub_type) for sub_type in arg.annotation],
+                "items": [
+                    _annotation_parse_to_json_schema(sub_type)
+                    for sub_type in arg.annotation
+                ],
             }
 
     elif isinstance(arg, _ParsedListAnnotation):
@@ -439,7 +463,10 @@ def _annotation_parse_to_json_schema(
     if is_optional:
         if isinstance(arg, _ParsedUnionAnnotation):
             for type_option in arg_type["anyOf"]:
-                if isinstance(type_option["type"], list) and "null" not in type_option["type"]:  # type: ignore
+                if (
+                    isinstance(type_option["type"], list)
+                    and "null" not in type_option["type"]
+                ):  # type: ignore
                     type_option["type"] = [*type_option["type"], "null"]  # type: ignore
                 elif not isinstance(type_option["type"], list):  # type: ignore
                     type_option["type"] = [type_option["type"], "null"]  # type: ignore