mirror of
https://github.com/OpenHands/OpenHands.git
synced 2025-12-26 05:48:36 +08:00
45 lines
1.4 KiB
Python
45 lines
1.4 KiB
Python
from enum import Enum
|
|
from types import UnionType
|
|
from typing import get_args, get_origin
|
|
|
|
OH_DEFAULT_AGENT = 'CodeActAgent'
|
|
OH_MAX_ITERATIONS = 100
|
|
|
|
|
|
class UndefinedString(str, Enum):
|
|
UNDEFINED = 'UNDEFINED'
|
|
|
|
|
|
def get_field_info(f):
|
|
"""Extract information about a dataclass field: type, optional, and default.
|
|
|
|
Args:
|
|
f: The field to extract information from.
|
|
|
|
Returns: A dict with the field's type, whether it's optional, and its default value.
|
|
"""
|
|
field_type = f.type
|
|
optional = False
|
|
|
|
# for types like str | None, find the non-None type and set optional to True
|
|
# this is useful for the frontend to know if a field is optional
|
|
# and to show the correct type in the UI
|
|
# Note: this only works for UnionTypes with None as one of the types
|
|
if get_origin(field_type) is UnionType:
|
|
types = get_args(field_type)
|
|
non_none_arg = next((t for t in types if t is not type(None)), None)
|
|
if non_none_arg is not None:
|
|
field_type = non_none_arg
|
|
optional = True
|
|
|
|
# type name in a pretty format
|
|
type_name = (
|
|
field_type.__name__ if hasattr(field_type, '__name__') else str(field_type)
|
|
)
|
|
|
|
# default is always present
|
|
default = f.default
|
|
|
|
# return a schema with the useful info for frontend
|
|
return {'type': type_name.lower(), 'optional': optional, 'default': default}
|