Skip to content

chore(tasks): remove code home, workstream grouping, and auto runs #72861

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
k11kirky merged 2 commits into from
Jul 22, 2026
Merged

chore(tasks): remove code home, workstream grouping, and auto runs #72861

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
8b3b3c9
chore(tasks): remove code home, workstream grouping, and auto runs
k11kirky Jul 22, 2026
99392b8
Merge branch 'master' into posthog-code/remove-code-home-workstreams
k11kirky Jul 22, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 0 additions & 6 deletions .semgrep/rules/security/idor-team-scoped-models.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -84,9 +84,6 @@ rules:
|ChangeRequest
|ClusteringConfig
|ClusteringJob
|CodePrSnapshot
|CodeWorkflowConfig
|CodeWorkstream
|Cohort
|CohortBackfillChunk
|CohortBackfillRun
Expand Down Expand Up @@ -398,9 +395,6 @@ rules:
|ChangeRequest
|ClusteringConfig
|ClusteringJob
|CodePrSnapshot
|CodeWorkflowConfig
|CodeWorkstream
|Cohort
|CohortBackfillChunk
|CohortBackfillRun
Expand Down
2 changes: 0 additions & 2 deletions posthog/temporal/schedule.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -101,7 +101,6 @@
from products.replay_vision.backend.temporal.reconciler import create_replay_vision_reconciler_schedule
from products.signals.backend.emission.conversations_schedule import create_conversations_signals_coordinator_schedule
from products.signals.backend.temporal.agentic.schedule import create_signals_scout_coordinator_schedule
from products.tasks.backend.facade.temporal import create_evaluate_code_workstreams_schedule
from products.web_analytics.backend.temporal.digest_notification.types import WADigestNotificationInput
from products.web_analytics.backend.temporal.weekly_digest.types import WAWeeklyDigestInput

Expand Down Expand Up @@ -835,7 +834,6 @@ async def create_error_tracking_recommendations_refresh_schedule(client: Client)
create_support_reply_coordinator_schedule,
create_replay_vision_reconciler_schedule,
create_replay_vision_estimates_schedule,
create_evaluate_code_workstreams_schedule,
create_github_job_logs_coordinator_schedule,
]

Expand Down
216 changes: 0 additions & 216 deletions products/tasks/backend/facade/api.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -42,8 +42,6 @@
is_blocked_sandbox_env_key,
)
from products.tasks.backend.error_telemetry import truncate_error_message
from products.tasks.backend.logic.code_workstreams.default_workflow import build_default_bindings
from products.tasks.backend.logic.code_workstreams.validation import validate_bindings
from products.tasks.backend.logic.services.image_builder import (
ensure_image_builder_task,
is_custom_images_enabled,
Expand All @@ -55,8 +53,6 @@
ChannelFeedMessage,
CodeInvite,
CodeInviteRedemption,
CodeWorkflowConfig,
CodeWorkstream,
SandboxCustomImage,
SandboxEnvironment,
SandboxSnapshot,
Expand Down Expand Up @@ -91,26 +87,12 @@
CODE_INVITE_INVALID_CODE = "invalid_code"
CODE_INVITE_NOT_REDEEMABLE = "not_redeemable"

# --- Code-workflow save outcomes ---
# Returned on ``CodeWorkflowSaveResult.outcome``; the presentation layer maps each to an
# HTTP status (saved -> 200, conflict -> 409, invalid -> 422).
CODE_WORKFLOW_SAVED = "saved"
CODE_WORKFLOW_CONFLICT = "conflict"
CODE_WORKFLOW_INVALID = "invalid"

# --- Code-home tuning ---
# An agent run counts as "active" only if it updated within this window.
CODE_HOME_ACTIVE_AGENT_WINDOW = timedelta(minutes=30)
_CODE_HOME_RUNNING_STATUSES = (TaskRun.Status.QUEUED, TaskRun.Status.IN_PROGRESS)
WIZARD_PR_READY_EMAIL_FEATURE_FLAG = "wizard-cloud-run-pr-ready-email-enabled"

__all__ = [
"CODE_INVITE_INVALID_CODE",
"CODE_INVITE_NOT_REDEEMABLE",
"CODE_INVITE_REDEEMED",
"CODE_WORKFLOW_CONFLICT",
"CODE_WORKFLOW_INVALID",
"CODE_WORKFLOW_SAVED",
"SandboxNetworkAccessLevel",
"SandboxSnapshotStatus",
"TaskOriginProduct",
Expand Down Expand Up @@ -149,8 +131,6 @@
"finalize_task_run_artifact_uploads",
"finalize_task_staged_artifacts",
"get_active_wizard_cloud_run",
"get_code_home",
"get_code_workflow_config",
"get_conversation_task_dtos",
"get_latest_pr_url_by_task",
"get_merged_pr_task_ids",
Expand Down Expand Up @@ -192,14 +172,11 @@
"read_task_run_logs",
"redeem_code_invite",
"redispatch_task_run",
"refresh_team_code_workstreams",
"relay_task_run_message",
"reset_code_workflow_bindings",
"resolve_slack_thread_context",
"resume_task_run_in_cloud",
"run_task",
"run_task_automation_now",
"save_code_workflow_bindings",
"send_cancel",
"select_repository_for_message",
"set_task_run_output",
Expand Down Expand Up @@ -3021,7 +2998,6 @@ def bootstrap_task_run(
reasoning_effort = validated_data.get("reasoning_effort")
github_user_token = validated_data.get("github_user_token")
initial_permission_mode = validated_data.get("initial_permission_mode")
home_quick_action = validated_data.get("home_quick_action")
imported_mcp_servers = validated_data.get("imported_mcp_servers")
relayed_mcp_servers = validated_data.get("relayed_mcp_servers")
if run_source == RunSource.SIGNAL_REPORT:
Expand All @@ -3042,7 +3018,6 @@ def bootstrap_task_run(
"provider": provider,
"model": model,
"reasoning_effort": reasoning_effort,
"home_quick_action": home_quick_action,
"rtk_enabled": validated_data.get("rtk_enabled"),
}.items():
if value is not None:
Expand Down Expand Up @@ -4803,197 +4778,6 @@ def resolve_slack_thread_context(
return contracts.SlackThreadContextResult(outcome="ok", context=context)


# --- Code workflow config (presentation CRUD) ---
# A user's per-team binding configuration. Reads seed a default config on first access;
# saves are optimistic-locked on ``version`` and validate the bindings before persisting.


def _epoch_ms(dt: datetime) -> int:
return int(dt.timestamp() * 1000)


def _code_workflow_config_to_dto(config: CodeWorkflowConfig) -> contracts.CodeWorkflowConfigDTO:
return contracts.CodeWorkflowConfigDTO(
id=str(config.id),
version=config.version,
updated_at=config.updated_at,
bindings=config.bindings,
)


def get_code_workflow_config(team_id: int, user_id: int) -> contracts.CodeWorkflowConfigDTO:
"""Return the user's config for the team, seeding a default one on first access."""
config, _ = CodeWorkflowConfig.objects.get_or_create(
team_id=team_id,
user_id=user_id,
defaults={"bindings": build_default_bindings(), "version": 1},
)
return _code_workflow_config_to_dto(config)


def save_code_workflow_bindings(
team_id: int, user_id: int, *, bindings: dict, expected_version: object
) -> contracts.CodeWorkflowSaveResult:
"""Validate and save bindings under optimistic locking.

Returns a ``conflict`` result when ``expected_version`` is not an int or does not match
the stored version, an ``invalid`` result (with diagnostics) when validation fails, or a
``saved`` result with the version-bumped config.
"""
with transaction.atomic():
current, _ = CodeWorkflowConfig.objects.select_for_update().get_or_create(
team_id=team_id,
user_id=user_id,
defaults={"bindings": build_default_bindings(), "version": 1},
)
if not isinstance(expected_version, int) or current.version != expected_version:
return contracts.CodeWorkflowSaveResult(
outcome=CODE_WORKFLOW_CONFLICT,
config=_code_workflow_config_to_dto(current),
)

result = validate_bindings(bindings)
if not result.can_save:
return contracts.CodeWorkflowSaveResult(
outcome=CODE_WORKFLOW_INVALID,
config=_code_workflow_config_to_dto(current),
diagnostics=[
contracts.CodeWorkflowDiagnosticDTO(
severity=d.severity,
code=d.code,
message=d.message,
situation_id=d.situation_id,
action_id=d.action_id,
)
for d in result.diagnostics
],
)

current.bindings = bindings
current.version = current.version + 1
current.save(update_fields=["bindings", "version", "updated_at"])

return contracts.CodeWorkflowSaveResult(
outcome=CODE_WORKFLOW_SAVED,
config=_code_workflow_config_to_dto(current),
)


def reset_code_workflow_bindings(team_id: int, user_id: int) -> contracts.CodeWorkflowConfigDTO:
"""Reset the user's bindings back to the defaults and bump the version."""
with transaction.atomic():
config, _ = CodeWorkflowConfig.objects.select_for_update().get_or_create(
team_id=team_id,
user_id=user_id,
defaults={"bindings": build_default_bindings(), "version": 1},
)
config.bindings = build_default_bindings()
config.version = config.version + 1
config.save(update_fields=["bindings", "version", "updated_at"])
return _code_workflow_config_to_dto(config)


# --- Code home board ---
# Active agents are computed live off in-flight runs; workstreams are persisted by the
# worker and split into board columns by their stored ``state``.


def _code_home_workstream_to_dto(ws: CodeWorkstream) -> contracts.CodeHomeWorkstreamDTO:
return contracts.CodeHomeWorkstreamDTO(
id=ws.key,
repo_name=ws.repo_name,
repo_full_path=ws.repo_full_path,
branch=ws.branch,
pr_url=ws.pr_url,
pr=ws.pr,
primary_situation=ws.primary_situation,
last_activity_at=_epoch_ms(ws.last_activity_at),
tasks=[
contracts.CodeHomeWorkstreamTaskDTO(
id=t.get("id"),
title=t.get("title"),
status=t.get("status"),
is_generating=False,
needs_permission=False,
quick_action=t.get("quick_action"),
)
for t in (ws.tasks or [])
],
situations=ws.situations or [],
)


def _code_home_active_agents(team_id: int, user_id: int) -> list[contracts.CodeHomeActiveAgentDTO]:
cutoff = django_timezone.now() - CODE_HOME_ACTIVE_AGENT_WINDOW
runs = (
TaskRun.objects.filter(
team_id=team_id,
task__created_by_id=user_id,
task__archived=False,
task__deleted=False,
status__in=_CODE_HOME_RUNNING_STATUSES,
updated_at__gte=cutoff,
)
.select_related("task")
.order_by("-updated_at")
)

seen_tasks: set[str] = set()
agents: list[contracts.CodeHomeActiveAgentDTO] = []
for run in runs.iterator():
task = run.task
if str(task.id) in seen_tasks:
continue
if (run.output or {}).get("pr_url"):
continue
seen_tasks.add(str(task.id))
agents.append(
contracts.CodeHomeActiveAgentDTO(
task_id=str(task.id),
title=task.title,
repo_name=task.repository.split("/")[-1] if task.repository else None,
branch=run.branch,
status=run.status,
last_activity_at=_epoch_ms(run.updated_at),
needs_permission=False,
cloud_pr_url=None,
)
)
return agents


def get_code_home(team_id: int, user_id: int) -> contracts.CodeHomeDTO:
"""Assemble the code-home board: live active agents plus persisted workstreams by column."""
workstreams = CodeWorkstream.objects.filter(team_id=team_id, user_id=user_id)
needs_attention: list[contracts.CodeHomeWorkstreamDTO] = []
in_progress: list[contracts.CodeHomeWorkstreamDTO] = []
for ws in workstreams.iterator():
dto = _code_home_workstream_to_dto(ws)
if ws.state == CodeWorkstream.WorkstreamState.ATTENTION:
needs_attention.append(dto)
else:
in_progress.append(dto)

return contracts.CodeHomeDTO(
active_agents=_code_home_active_agents(team_id, user_id),
needs_attention=needs_attention,
in_progress=in_progress,
)


def refresh_team_code_workstreams(team_id: int) -> bool:
"""Trigger an on-demand evaluation of the team's code workstreams.

Returns whether a new evaluation workflow was started (``False`` if one was already
running).
"""
from products.tasks.backend.temporal.code_workstreams.client import ( # noqa: PLC0415 — keep temporalio off the api import path
trigger_team_code_workstreams_evaluation,
)

return trigger_team_code_workstreams_evaluation(team_id)


# --- Id-based bridges to the sandbox/agent-command surface ---
# These take a run id (not an ORM TaskRun) so callers never hold a tasks model. The heavy
# service modules are imported lazily to keep them off this module's import path.
Expand Down
Loading
Loading