yi
228 lines
10 KiB
Python
228 lines
10 KiB
Python
from sqlalchemy import Boolean, Column, DateTime, ForeignKey, Integer, String, Text, JSON, Enum as SQLEnum, func
|
|
from sqlalchemy.orm import relationship
|
|
import enum
|
|
|
|
from app.database import Base
|
|
|
|
|
|
class A2ATaskState(str, enum.Enum):
|
|
SUBMITTED = "SUBMITTED"
|
|
WORKING = "WORKING"
|
|
COMPLETED = "COMPLETED"
|
|
FAILED = "FAILED"
|
|
CANCELED = "CANCELED"
|
|
INPUT_REQUIRED = "INPUT_REQUIRED"
|
|
AUTH_REQUIRED = "AUTH_REQUIRED"
|
|
REJECTED = "REJECTED"
|
|
|
|
|
|
class A2APartType(str, enum.Enum):
|
|
TEXT = "text"
|
|
RAW = "raw"
|
|
URL = "url"
|
|
DATA = "data"
|
|
|
|
|
|
class A2AMessageRole(str, enum.Enum):
|
|
USER = "user"
|
|
AGENT = "agent"
|
|
SYSTEM = "system"
|
|
|
|
|
|
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)
|
|
shared_secret = Column(String, nullable=True)
|
|
mtls_ca_cert = Column(Text, nullable=True)
|
|
mtls_client_cert = Column(Text, nullable=True)
|
|
mtls_client_key = Column(Text, nullable=True)
|
|
oauth2_client_id = Column(String, nullable=True)
|
|
oauth2_client_secret = Column(String, nullable=True)
|
|
oauth2_token_url = Column(String, nullable=True)
|
|
oauth2_scopes = Column(String, nullable=True)
|
|
oidc_issuer_url = Column(String, nullable=True)
|
|
oidc_client_id = Column(String, nullable=True)
|
|
oidc_client_secret = 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 A2APart(Base):
|
|
__tablename__ = "a2a_parts"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
message_id = Column(Integer, ForeignKey("a2a_messages.id", ondelete="CASCADE"), nullable=True, index=True)
|
|
artifact_id = Column(Integer, ForeignKey("a2a_artifacts.id", ondelete="CASCADE"), nullable=True, index=True)
|
|
part_type = Column(SQLEnum(A2APartType), nullable=False)
|
|
text_content = Column(Text, nullable=True)
|
|
raw_content = Column(Text, nullable=True)
|
|
url_content = Column(String, nullable=True)
|
|
data_content = Column(Text, nullable=True)
|
|
media_type = Column(String, nullable=True)
|
|
filename = Column(String, nullable=True)
|
|
metadata_json = Column(Text, nullable=False, default="{}")
|
|
created_at = Column(DateTime, default=func.now())
|
|
|
|
message = relationship("A2AMessage", back_populates="parts", foreign_keys=[message_id])
|
|
artifact = relationship("A2AArtifact", back_populates="parts", foreign_keys=[artifact_id])
|
|
|
|
|
|
class A2AMessage(Base):
|
|
__tablename__ = "a2a_messages"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
message_id = Column(String, nullable=False, unique=True, index=True)
|
|
context_id = Column(String, nullable=True, index=True)
|
|
task_id = Column(String, ForeignKey("a2a_tasks.id", ondelete="CASCADE"), nullable=True, index=True)
|
|
role = Column(SQLEnum(A2AMessageRole), nullable=False)
|
|
extensions_json = Column(Text, nullable=False, default="{}")
|
|
reference_task_ids_json = Column(Text, nullable=False, default="[]")
|
|
created_at = Column(DateTime, default=func.now(), index=True)
|
|
|
|
task = relationship("A2ATask", back_populates="messages", foreign_keys=[task_id])
|
|
parts = relationship("A2APart", back_populates="message", cascade="all, delete-orphan")
|
|
|
|
|
|
class A2AArtifact(Base):
|
|
__tablename__ = "a2a_artifacts"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
artifact_id = Column(String, nullable=False, unique=True, index=True)
|
|
task_id = Column(String, ForeignKey("a2a_tasks.id", ondelete="CASCADE"), nullable=False, index=True)
|
|
name = Column(String, nullable=True)
|
|
description = Column(Text, nullable=True)
|
|
metadata_json = Column(Text, nullable=False, default="{}")
|
|
extensions_json = Column(Text, nullable=False, default="{}")
|
|
created_at = Column(DateTime, default=func.now())
|
|
updated_at = Column(DateTime, default=func.now(), onupdate=func.now())
|
|
|
|
task = relationship("A2ATask", back_populates="artifacts")
|
|
parts = relationship("A2APart", back_populates="artifact", cascade="all, delete-orphan")
|
|
|
|
|
|
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)
|
|
context_id = Column(String, nullable=True, 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(SQLEnum(A2ATaskState), nullable=False, index=True, default=A2ATaskState.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="{}")
|
|
history_length = Column(Integer, nullable=False, default=0)
|
|
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")
|
|
messages = relationship("A2AMessage", back_populates="task", cascade="all, delete-orphan", foreign_keys=[A2AMessage.task_id])
|
|
artifacts = relationship("A2AArtifact", back_populates="task", cascade="all, delete-orphan")
|
|
|
|
|
|
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)
|