Fix missing type packages (#107)

This commit is contained in:
Sterling Dreyer 2024-10-14 20:38:27 -07:00 committed by GitHub
parent b88a07b526
commit b95e26e5ec
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 4 additions and 2 deletions

View file

@ -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)

View file

@ -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}