Skip to content

Commit 7aef4d3

Browse files
committed
cover trigger_coding_agent_launch
1 parent aaf382f commit 7aef4d3

File tree

1 file changed

+16
-9
lines changed

1 file changed

+16
-9
lines changed

src/sentry/seer/endpoints/seer_rpc.py

Lines changed: 16 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -83,6 +83,7 @@
8383
from sentry.seer.autofix.utils import (
8484
AutofixTriggerSource,
8585
get_project_seer_preferences,
86+
read_preference_from_sentry_db,
8687
resolve_repository_ids,
8788
write_preference_to_sentry_db,
8889
)
@@ -613,17 +614,23 @@ def trigger_coding_agent_launch(
613614
},
614615
)
615616
try:
616-
project = Project.objects.get_from_cache(id=project_id)
617617
organization = Organization.objects.get_from_cache(id=organization_id)
618618
if features.has("organizations:seer-project-settings-dual-write", organization):
619-
preference_response = get_project_seer_preferences(project.id)
620-
if preference_response and preference_response.preference:
621-
updated_preference = preference_response.preference.copy(
622-
update={"automation_handoff": None}
623-
)
624-
validated_pref = SeerProjectPreference.validate(updated_preference)
625-
resolved_pref = resolve_repository_ids(organization.id, [validated_pref])
626-
write_preference_to_sentry_db(project, resolved_pref[0])
619+
project = Project.objects.get_from_cache(id=project_id)
620+
621+
if features.has(
622+
"organizations:seer-project-settings-read-from-sentry", organization
623+
):
624+
preference = read_preference_from_sentry_db(project)
625+
else:
626+
preference = get_project_seer_preferences(project.id).preference
627+
628+
if preference:
629+
updated_preference = preference.copy(update={"automation_handoff": None})
630+
resolved_pref = resolve_repository_ids(
631+
organization.id, [SeerProjectPreference.validate(updated_preference)]
632+
)[0]
633+
write_preference_to_sentry_db(project, resolved_pref)
627634
except Exception:
628635
logger.exception(
629636
"coding_agent.clear_handoff_preference_failed",

0 commit comments

Comments
 (0)