feat: support a2a mode

2026-04-01 11:21:55 +08:00
parent 9952af198a
commit 86447049a9
12 changed files with 3092 additions and 26 deletions
@@ -0,0 +1,134 @@
+from sqlalchemy import Boolean, Column, DateTime, ForeignKey, Integer, String, Text, func
+from sqlalchemy.orm import relationship
+
+from app.database import Base
+
+
+class A2ARemoteAgent(Base):
+    __tablename__ = "a2a_remote_agents"
+
+    id = Column(Integer, primary_key=True, index=True)
+    project_id = Column(Integer, ForeignKey("projects.id"), nullable=False, index=True)
+    name = Column(String, nullable=False)
+    base_url = Column(String, nullable=False)
+    auth_scheme = Column(String, nullable=False, default="none")
+    auth_token = Column(String, nullable=True)
+    protocol_version = Column(String, nullable=True)
+    capabilities_json = Column(Text, nullable=False, default="[]")
+    card_json = Column(Text, nullable=True)
+    card_fetched_at = Column(DateTime, nullable=True)
+    healthy = Column(Boolean, nullable=False, default=False)
+    failure_count = Column(Integer, nullable=False, default=0)
+    circuit_open_until = Column(DateTime, nullable=True)
+    created_by = Column(Integer, ForeignKey("users.id"), nullable=False, index=True)
+    created_at = Column(DateTime, default=func.now())
+    updated_at = Column(DateTime, default=func.now(), onupdate=func.now())
+
+    project = relationship("Project")
+
+
+class A2ATask(Base):
+    __tablename__ = "a2a_tasks"
+
+    id = Column(String, primary_key=True, index=True)
+    project_id = Column(Integer, ForeignKey("projects.id"), nullable=False, index=True)
+    tenant_id = Column(Integer, ForeignKey("users.id"), nullable=False, index=True)
+    source = Column(String, nullable=False, default="local")
+    remote_agent_id = Column(Integer, ForeignKey("a2a_remote_agents.id"), nullable=True, index=True)
+    idempotency_key = Column(String, nullable=True, index=True)
+    state = Column(String, nullable=False, index=True, default="SUBMITTED")
+    input_text = Column(Text, nullable=False, default="")
+    output_text = Column(Text, nullable=True)
+    error_message = Column(Text, nullable=True)
+    compatibility_mode = Column(Boolean, nullable=False, default=True)
+    metadata_json = Column(Text, nullable=False, default="{}")
+    created_at = Column(DateTime, default=func.now())
+    updated_at = Column(DateTime, default=func.now(), onupdate=func.now())
+    finished_at = Column(DateTime, nullable=True)
+
+    project = relationship("Project")
+    remote_agent = relationship("A2ARemoteAgent")
+
+
+class A2ATaskEvent(Base):
+    __tablename__ = "a2a_task_events"
+
+    id = Column(Integer, primary_key=True, index=True)
+    task_id = Column(String, ForeignKey("a2a_tasks.id", ondelete="CASCADE"), nullable=False, index=True)
+    event_type = Column(String, nullable=False)
+    payload_json = Column(Text, nullable=False, default="{}")
+    created_at = Column(DateTime, default=func.now(), index=True)
+
+    task = relationship("A2ATask")
+
+
+class A2ATaskWebhook(Base):
+    __tablename__ = "a2a_task_webhooks"
+
+    id = Column(Integer, primary_key=True, index=True)
+    task_id = Column(String, ForeignKey("a2a_tasks.id", ondelete="CASCADE"), nullable=False, index=True)
+    target_url = Column(String, nullable=False)
+    secret = Column(String, nullable=True)
+    auth_header = Column(String, nullable=True)
+    enabled = Column(Boolean, nullable=False, default=True)
+    created_by = Column(Integer, ForeignKey("users.id"), nullable=False, index=True)
+    created_at = Column(DateTime, default=func.now())
+    updated_at = Column(DateTime, default=func.now(), onupdate=func.now())
+
+    task = relationship("A2ATask")
+
+
+class A2AWebhookDelivery(Base):
+    __tablename__ = "a2a_webhook_deliveries"
+
+    id = Column(Integer, primary_key=True, index=True)
+    task_id = Column(String, ForeignKey("a2a_tasks.id", ondelete="CASCADE"), nullable=False, index=True)
+    webhook_id = Column(Integer, ForeignKey("a2a_task_webhooks.id", ondelete="CASCADE"), nullable=False, index=True)
+    event_id = Column(Integer, ForeignKey("a2a_task_events.id", ondelete="CASCADE"), nullable=False, index=True)
+    attempt = Column(Integer, nullable=False, default=0)
+    status = Column(String, nullable=False, default="PENDING")
+    response_code = Column(Integer, nullable=True)
+    response_body = Column(Text, nullable=True)
+    error_message = Column(Text, nullable=True)
+    next_retry_at = Column(DateTime, nullable=True)
+    delivered_at = Column(DateTime, nullable=True)
+    dead_letter = Column(Boolean, nullable=False, default=False, index=True)
+    created_at = Column(DateTime, default=func.now())
+    updated_at = Column(DateTime, default=func.now(), onupdate=func.now())
+
+    task = relationship("A2ATask")
+    webhook = relationship("A2ATaskWebhook")
+    event = relationship("A2ATaskEvent")
+
+
+class A2AProjectConfig(Base):
+    __tablename__ = "a2a_project_configs"
+
+    project_id = Column(Integer, ForeignKey("projects.id"), primary_key=True)
+    canary_enabled = Column(Boolean, nullable=False, default=False)
+    canary_percent = Column(Integer, nullable=False, default=0)
+    rollback_to_local = Column(Boolean, nullable=False, default=True)
+    compatibility_mode = Column(Boolean, nullable=False, default=True)
+    dual_event_write = Column(Boolean, nullable=False, default=True)
+    route_mode_default = Column(String, nullable=False, default="local_first")
+    fallback_chain_json = Column(Text, nullable=False, default='["local"]')
+    alert_thresholds_json = Column(Text, nullable=False, default="{}")
+    updated_by = Column(Integer, ForeignKey("users.id"), nullable=False)
+    updated_at = Column(DateTime, default=func.now(), onupdate=func.now())
+
+    project = relationship("Project")
+
+
+class A2AAuditLog(Base):
+    __tablename__ = "a2a_audit_logs"
+
+    id = Column(Integer, primary_key=True, index=True)
+    actor_user_id = Column(Integer, ForeignKey("users.id"), nullable=False, index=True)
+    action = Column(String, nullable=False)
+    target_type = Column(String, nullable=False)
+    target_id = Column(String, nullable=False)
+    project_id = Column(Integer, ForeignKey("projects.id"), nullable=True, index=True)
+    task_id = Column(String, nullable=True, index=True)
+    result = Column(String, nullable=False)
+    detail_json = Column(Text, nullable=False, default="{}")
+    created_at = Column(DateTime, default=func.now(), index=True)