3.9 typing complaint

Author: Andrei Bratu

src/humanloop/decorators/tool.py

@@ -87,19 +87,21 @@ def _parse_tool_parameters_schema(func) -> dict[str, dict]:
             inspect.Parameter.VAR_KEYWORD,
         ):
             raise ValueError("Varargs and kwargs are not supported")
-        if isinstance(origin := typing.get_origin(parameter.annotation), dict):
+        origin = typing.get_origin(parameter.annotation)
+        print("HEY", origin, parameter.annotation, origin is None)
+        if origin is Union:
+            param_schema = _handle_union_annotation(parameter)
+        elif origin is None:
+            param_schema = _handle_simple_type(parameter)
+            required.append(parameter.name)
+        elif isinstance(origin, dict):
             param_schema = _handle_dict_annotation(parameter)
             parameters_schema["required"].append(parameter.name)
             required.append(parameter.name)
         elif isinstance(origin, list):
             param_schema = _handle_list_annotation(parameter)
             parameters_schema["required"].append(parameter.name)
             required.append(parameter.name)
-        elif isinstance(origin, Union):
-            param_schema = _handle_union_annotation(parameter)
-        elif origin is None:
-            param_schema = _handle_simple_type(parameter)
-            required.append(parameter.name)
         else:
             raise ValueError("Unsupported type hint ", parameter)
         parameters_schema["properties"][parameter.name] = param_schema