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

Added the check for Sentry not existing

Browse Source
This commit is contained in:
Zvonimir Sabljic
2025-01-30 13:18:45 +01:00
parent ef033dbdd4
commit edc35890f1
2 changed files with 34 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

42
core/cli/main.py
View File

@@ -5,8 +5,6 @@ import sys
from argparse import Namespace from argparse import Namespace
from asyncio import run from asyncio import run
import sentry_sdk
from core.agents.orchestrator import Orchestrator 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.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 import LLMProvider, get_config
@@ -24,6 +22,12 @@ log = get_logger(__name__)
telemetry_sent = False telemetry_sent = False
try:
import sentry_sdk
SENTRY_ENABLED = True
except ImportError:
SENTRY_ENABLED = False
async def cleanup(ui: UIBase): async def cleanup(ui: UIBase):
@@ -280,22 +284,24 @@ async def async_main(
return success return success
telemetry.set("user_contact", args.email) telemetry.set("user_contact", args.email)
sentry_sdk.set_user( if SENTRY_ENABLED:
{ sentry_sdk.set_user(
# TODO add user id {
# "id": args.user_id or "CLI only", # TODO add user id
"email": args.email or "CLI only", # "id": args.user_id or "CLI only",
} "email": args.email or "CLI only",
) }
sentry_sdk.set_context( )
"pythagora-data", sentry_sdk.set_context(
{ "pythagora-data",
"extension-version": args.extension_version or "CLI only", {
"core-version": get_version(), "extension-version": args.extension_version or "CLI only",
# TODO get environment from the extension "core-version": get_version(),
# "environment": args.environment, # TODO get environment from the extension
}, # "environment": args.environment,
) },
)
if args.extension_version: if args.extension_version:
telemetry.set("is_extension", True) telemetry.set("is_extension", True)
telemetry.set("extension_version", args.extension_version) telemetry.set("extension_version", args.extension_version)

17
main.py
View File

@@ -2,15 +2,18 @@
import os.path import os.path
import sys import sys
import sentry_sdk try:
import sentry_sdk
sentry_sdk.init( sentry_sdk.init(
dsn="https://4101633bc5560bae67d6eab013ba9686@o4508731634221056.ingest.us.sentry.io/4508732401909760", dsn="https://4101633bc5560bae67d6eab013ba9686@o4508731634221056.ingest.us.sentry.io/4508732401909760",
send_default_pii=True, send_default_pii=True,
traces_sample_rate=1.0, traces_sample_rate=1.0,
) )
sentry_sdk.profiler.start_profiler() sentry_sdk.profiler.start_profiler()
except ImportError:
SENTRY_ENABLED = False
try: try:
from core.cli.main import run_pythagora from core.cli.main import run_pythagora