Fix missing type packages (#107)
This commit is contained in:
parent
b88a07b526
commit
b95e26e5ec
2 changed files with 4 additions and 2 deletions
|
|
@ -384,7 +384,7 @@ def create_output_definition(func: Callable) -> ToolOutput:
|
||||||
)
|
)
|
||||||
|
|
||||||
if hasattr(return_type, "__metadata__"):
|
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__
|
return_type = return_type.__origin__
|
||||||
|
|
||||||
# Unwrap Optional types
|
# Unwrap Optional types
|
||||||
|
|
@ -542,7 +542,7 @@ def get_wire_type_info(_type: type) -> WireTypeInfo:
|
||||||
# Special case: Enum can be enumerated on the wire
|
# Special case: Enum can be enumerated on the wire
|
||||||
elif issubclass(type_to_check, Enum):
|
elif issubclass(type_to_check, Enum):
|
||||||
is_enum = True
|
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)
|
return WireTypeInfo(wire_type, inner_wire_type, enum_values if is_enum else None)
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -22,6 +22,8 @@ tomlkit = "^0.12.4"
|
||||||
openai = "^1.36.0" # TODO: relax to an earlier version that still has what we need
|
openai = "^1.36.0" # TODO: relax to an earlier version that still has what we need
|
||||||
pyjwt = "^2.8.0"
|
pyjwt = "^2.8.0"
|
||||||
loguru = "^0.7.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-instrumentation-fastapi = {version = "0.48b0", optional = true}
|
||||||
opentelemetry-exporter-otlp-proto-http = {version = "1.27.0", optional = true}
|
opentelemetry-exporter-otlp-proto-http = {version = "1.27.0", optional = true}
|
||||||
opentelemetry-exporter-otlp-proto-common = {version = "1.27.0", optional = true}
|
opentelemetry-exporter-otlp-proto-common = {version = "1.27.0", optional = true}
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue