github/gpt-pilot
1
1
Fork 0
mirror of https://github.com/Pythagora-io/gpt-pilot.git synced 2026-01-10 13:37:55 -05:00
Code Issues Packages Projects Releases Wiki Activity

Enable saving LLM requests to db if user email is pythagora.ai

Browse Source
This commit is contained in:
mijauexe
2025-06-26 16:58:29 +02:00
parent b287ccba97
commit 30ffc3179d
2 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
core/cli/helpers.py
View File

@@ -973,7 +973,7 @@ def init() -> tuple[UIBase, SessionManager, Namespace]:
ui = PlainConsoleUI()
run_migrations(config.db)
db = SessionManager(config.db)
db = SessionManager(config.db, args)
return (ui, db, args)

5
core/db/session.py
View File

@@ -18,7 +18,7 @@ class SessionManager:
... # Do something with the session
"""
def __init__(self, config: DBConfig):
def __init__(self, config: DBConfig, args=None):
"""
Initialize the session manager with the given configuration.
@@ -31,7 +31,8 @@ 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
email_is_pythagora = args is not None and args.email is not None and args.email.endswith("@pythagora.ai")
self.save_llm_requests = config.save_llm_requests or email_is_pythagora
event.listen(self.engine.sync_engine, "connect", self._on_connect)
def _on_connect(self, dbapi_connection, _):