fix bug in enum.Enum type annotation

comfuture · comfuture · commit c9d1b6e97544 · 2024-04-09T15:32:04.000+09:00
diff --git a/function_schema/core.py b/function_schema/core.py
@@ -80,7 +80,7 @@ def get_function_schema(
             # find enum in param_args tuple
             enum_ = next(
                 (
-                    arg.name
+                    [e.name for e in arg]
                     for arg in param_args
                     if isinstance(arg, type) and issubclass(arg, enum.Enum)
                 ),

Original file line number	Diff line number	Diff line change
`@@ -80,7 +80,7 @@ def get_function_schema(`
`80`	`80`	`# find enum in param_args tuple`
`81`	`81`	`enum_ = next(`
`82`	`82`	`(`
`83`		`- arg.name`
	`83`	`+ [e.name for e in arg]`
`84`	`84`	`for arg in param_args`
`85`	`85`	`if isinstance(arg, type) and issubclass(arg, enum.Enum)`
`86`	`86`	`),`