github/gpt-pilot
1
1
Fork 0
mirror of https://github.com/Pythagora-io/gpt-pilot.git synced 2026-01-09 21:27:53 -05:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'main' into staging

Browse Source
This commit is contained in:
zvone187
2025-06-30 19:20:25 +02:00
committed by GitHub
parent e242a83fa6 c73a17760c
commit 31713db91b
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

@@ -976,7 +976,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, _):