mirror of
https://github.com/Pythagora-io/gpt-pilot.git
synced 2026-01-10 05:27:54 -05:00
@@ -8,6 +8,7 @@ from asyncio import run
|
||||
from core.agents.orchestrator import Orchestrator
|
||||
from core.cli.helpers import delete_project, init, list_projects, list_projects_json, load_project, show_config
|
||||
from core.config import LLMProvider, get_config
|
||||
from core.config.version import get_version
|
||||
from core.db.session import SessionManager
|
||||
from core.db.v0importer import LegacyDatabaseImporter
|
||||
from core.llm.anthropic_client import CustomAssertionError
|
||||
@@ -21,6 +22,12 @@ log = get_logger(__name__)
|
||||
|
||||
|
||||
telemetry_sent = False
|
||||
try:
|
||||
import sentry_sdk
|
||||
|
||||
SENTRY_ENABLED = True
|
||||
except ImportError:
|
||||
SENTRY_ENABLED = False
|
||||
|
||||
|
||||
async def cleanup(ui: UIBase):
|
||||
@@ -277,6 +284,24 @@ async def async_main(
|
||||
return success
|
||||
|
||||
telemetry.set("user_contact", args.email)
|
||||
if SENTRY_ENABLED:
|
||||
sentry_sdk.set_user(
|
||||
{
|
||||
# TODO add user id
|
||||
# "id": args.user_id or "CLI only",
|
||||
"email": args.email or "CLI only",
|
||||
}
|
||||
)
|
||||
sentry_sdk.set_context(
|
||||
"pythagora-data",
|
||||
{
|
||||
"extension-version": args.extension_version or "CLI only",
|
||||
"core-version": get_version(),
|
||||
# TODO get environment from the extension
|
||||
# "environment": args.environment,
|
||||
},
|
||||
)
|
||||
|
||||
if args.extension_version:
|
||||
telemetry.set("is_extension", True)
|
||||
telemetry.set("extension_version", args.extension_version)
|
||||
|
||||
15
main.py
15
main.py
@@ -1,9 +1,20 @@
|
||||
#!/usr/bin/env python
|
||||
|
||||
|
||||
import os.path
|
||||
import sys
|
||||
|
||||
try:
|
||||
import sentry_sdk
|
||||
|
||||
sentry_sdk.init(
|
||||
dsn="https://4101633bc5560bae67d6eab013ba9686@o4508731634221056.ingest.us.sentry.io/4508732401909760",
|
||||
send_default_pii=True,
|
||||
traces_sample_rate=1.0,
|
||||
)
|
||||
|
||||
sentry_sdk.profiler.start_profiler()
|
||||
except ImportError:
|
||||
SENTRY_ENABLED = False
|
||||
|
||||
try:
|
||||
from core.cli.main import run_pythagora
|
||||
except ImportError as err:
|
||||
|
||||
@@ -34,3 +34,4 @@ tqdm==4.67.1
|
||||
typing-extensions==4.12.2
|
||||
urllib3==2.2.3
|
||||
wcwidth==0.2.13
|
||||
sentry-sdk==2.20.0
|
||||
|
||||
Reference in New Issue
Block a user