File tree Expand file tree Collapse file tree 1 file changed +11
-8
lines changed
examples/pydantic_ai_ag_ui_examples/cli Expand file tree Collapse file tree 1 file changed +11
-8
lines changed Original file line number Diff line number Diff line change @@ -15,7 +15,7 @@ class Args:
15
15
port : int
16
16
reload : bool
17
17
log_level : str
18
- loggers : tuple [str , ... ]
18
+ loggers : list [str ]
19
19
20
20
def log_config (self ) -> dict [str , Any ]:
21
21
"""Return the logging configuration based on the log level."""
@@ -30,14 +30,9 @@ def log_config(self) -> dict[str, Any]:
30
30
return log_config
31
31
32
32
33
- def parse_args (
34
- loggers : tuple [str , ...] = ('pydantic_ai_ag_ui.adapter' ,),
35
- ) -> Args :
33
+ def parse_args () -> Args :
36
34
"""Parse command line arguments for the PydanticAI AG-UI servers.
37
35
38
- Args:
39
- loggers: A tuple of logger names to configure. Defaults to the adapter logger.
40
-
41
36
Returns:
42
37
Args: A dataclass containing the parsed command line arguments.
43
38
"""
@@ -66,6 +61,14 @@ def parse_args(
66
61
default = 'info' ,
67
62
help = 'Adapter log level (default: info)' ,
68
63
)
64
+ parser .add_argument (
65
+ '--loggers' ,
66
+ nargs = '*' ,
67
+ default = [
68
+ 'pydantic_ai_ag_ui.adapter' ,
69
+ ],
70
+ help = 'Logger names to configure (default: adapter and model loggers)' ,
71
+ )
69
72
70
73
args : argparse .Namespace = parser .parse_args ()
71
- return Args (loggers = loggers , ** vars (args ))
74
+ return Args (** vars (args ))
You can’t perform that action at this time.
0 commit comments