Skip to content

Commit f091c9d

Browse files
committed
fix(ag-ui): example loggers config
Fix example loggers configuration which wasn't wired up correctly.
1 parent 131d793 commit f091c9d

File tree

1 file changed

+8
-8
lines changed
  • examples/pydantic_ai_ag_ui_examples/cli

1 file changed

+8
-8
lines changed

examples/pydantic_ai_ag_ui_examples/cli/args.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -20,18 +20,18 @@ class Args:
2020
def log_config(self) -> dict[str, Any]:
2121
"""Return the logging configuration based on the log level."""
2222
log_config: dict[str, Any] = LOGGING_CONFIG.copy()
23-
log_config['loggers']['pydantic_ai_ag_ui.adapter'] = {
24-
'handlers': ['default'],
25-
'level': self.log_level.upper(),
26-
'propagate': False,
27-
}
23+
for logger in self.loggers:
24+
log_config['loggers'][logger] = {
25+
'handlers': ['default'],
26+
'level': self.log_level.upper(),
27+
'propagate': False,
28+
}
29+
2830
return log_config
2931

3032

3133
def parse_args(
32-
loggers: tuple[str, ...] = tuple(
33-
'pydantic_ai_ag_ui.adapter',
34-
),
34+
loggers: tuple[str, ...] = ('pydantic_ai_ag_ui.adapter',),
3535
) -> Args:
3636
"""Parse command line arguments for the PydanticAI AG-UI servers.
3737

0 commit comments

Comments
 (0)