fix(cli): ignore session_db_kwargs for sqlite session services

Close issue #4317

Co-authored-by: Liang Wu <wuliang@google.com>
PiperOrigin-RevId: 864979145
This commit is contained in:
Liang Wu
2026-02-03 11:49:13 -08:00
committed by Copybara-Service
parent 706a6dda81
commit ce07cd8144
2 changed files with 28 additions and 9 deletions
+10 -3
View File
@@ -250,9 +250,16 @@ def _register_builtin_services(registry: ServiceRegistry) -> None:
return memory_session_factory("memory://", **kwargs)
elif db_path.startswith("/"):
db_path = db_path[1:]
kwargs_copy = kwargs.copy()
kwargs_copy.pop("agents_dir", None)
return SqliteSessionService(db_path=db_path, **kwargs_copy)
# SqliteSessionService only accepts db_path, warn if extra kwargs provided
ignored_kwargs = {k: v for k, v in kwargs.items() if k != "agents_dir"}
if ignored_kwargs:
logger.warning(
"SqliteSessionService does not support additional kwargs. "
"The following parameters will be ignored: %s",
list(ignored_kwargs.keys()),
)
return SqliteSessionService(db_path=db_path)
registry.register_session_service("memory", memory_session_factory)
registry.register_session_service("agentengine", agentengine_session_factory)