diff --git a/arcade/arcade/core/catalog.py b/arcade/arcade/core/catalog.py index a9451a4b..496e8284 100644 --- a/arcade/arcade/core/catalog.py +++ b/arcade/arcade/core/catalog.py @@ -384,7 +384,7 @@ def create_output_definition(func: Callable) -> ToolOutput: ) if hasattr(return_type, "__metadata__"): - description = return_type.__metadata__[0] if return_type.__metadata__ else None + description = return_type.__metadata__[0] if return_type.__metadata__ else None # type: ignore[assignment] return_type = return_type.__origin__ # Unwrap Optional types @@ -542,7 +542,7 @@ def get_wire_type_info(_type: type) -> WireTypeInfo: # Special case: Enum can be enumerated on the wire elif issubclass(type_to_check, Enum): is_enum = True - enum_values = [e.value for e in type_to_check] + enum_values = [e.value for e in type_to_check] # type: ignore[union-attr] return WireTypeInfo(wire_type, inner_wire_type, enum_values if is_enum else None) diff --git a/arcade/pyproject.toml b/arcade/pyproject.toml index 569161ab..0eff8b01 100644 --- a/arcade/pyproject.toml +++ b/arcade/pyproject.toml @@ -22,6 +22,8 @@ tomlkit = "^0.12.4" openai = "^1.36.0" # TODO: relax to an earlier version that still has what we need pyjwt = "^2.8.0" loguru = "^0.7.0" +types-python-dateutil = "2.9.0.20241003" +types-pytz = "2024.2.0.20241003" opentelemetry-instrumentation-fastapi = {version = "0.48b0", optional = true} opentelemetry-exporter-otlp-proto-http = {version = "1.27.0", optional = true} opentelemetry-exporter-otlp-proto-common = {version = "1.27.0", optional = true}