Move arg to config.json

cloud/config-docker.json

@@ -111,7 +111,8 @@
   },
   "db": {
     "url": "sqlite+aiosqlite:///data/database/pythagora.db",
-    "debug_sql": false
+    "debug_sql": false,
+    "save_llm_requests": false
   },
   "ui": {
     "type": "plain"

core/cli/helpers.py

@@ -248,12 +248,6 @@ def parse_arguments() -> Namespace:
         dest="auto_confirm_breakdown",
         required=False,
     )
-    parser.add_argument(
-        "--save-llm-req",
-        action="store_true",
-        default=False,
-        help="Enable saving LLM requests to the database",
-    )
     parser.add_argument("--access-token", help="Access token", required=False)
     parser.add_argument(
         "--enable-api-server",

core/cli/main.py

@@ -242,8 +242,6 @@ async def async_main(
 
     if not args.auto_confirm_breakdown:
         sm.auto_confirm_breakdown = False
-    if args.save_llm_req:
-        sm.save_llm_requests = True
     ui_started = await ui.start()
     if not ui_started:
         if api_server:

core/config/__init__.py

@@ -243,6 +243,7 @@ class DBConfig(_StrictModel):
         description="Database connection URL",
     )
     debug_sql: bool = Field(False, description="Log all SQL queries to the console")
+    save_llm_requests: bool = Field(False, description="Save LLM requests to db")
 
     @field_validator("url")
     @classmethod

core/db/session.py

@@ -31,7 +31,7 @@ class SessionManager:
         self.SessionClass = async_sessionmaker(self.engine, expire_on_commit=False)
         self.session = None
         self.recursion_depth = 0
-
+        self.save_llm_requests = config.save_llm_requests
         event.listen(self.engine.sync_engine, "connect", self._on_connect)
 
     def _on_connect(self, dbapi_connection, _):

core/state/state_manager.py

@@ -394,7 +394,7 @@ class StateManager:
         :param request_log: The request log to log.
         """
         # removed logging of LLM requests
-        if self.save_llm_requests:
+        if self.session_manager.save_llm_requests:
             async with self.db_blocker():
                 try:
                     telemetry.record_llm_request(

tests/cli/test_cli.py

@@ -63,7 +63,6 @@ def test_parse_arguments(mock_ArgumentParser):
         "--use-git",
         "--access-token",
         "--no-auto-confirm-breakdown",
-        "--save-llm-req",
     }
 
     parser.parse_args.assert_called_once_with()