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