aragora-sdk 2.7.2__py3-none-any.whl

aragora_sdk/__init__.py

@@ -0,0 +1,519 @@
+"""
+Aragora Python SDK.
+
+The canonical Python client for the Aragora multi-agent debate platform.
+
+Usage::
+
+    from aragora_sdk import AragoraClient, Debate, DebateCreateRequest
+
+    client = AragoraClient(api_key="your-key")
+    debate = client.debates.create(DebateCreateRequest(task="Evaluate options"))
+
+All generated Pydantic types are re-exported here for convenient access.
+"""
+
+from .client import AragoraAsyncClient, AragoraClient
+
+# Import typed event classes
+from .events import (
+    EVENT_CLASS_MAP,
+    AgentMessageEvent,
+    AudienceSuggestionEvent,
+    ConnectedEvent,
+    ConsensusEvent,
+    ConsensusReachedEvent,
+    CritiqueEvent,
+    DebateEndEvent,
+    DebateStartEvent,
+    DisconnectedEvent,
+    ErrorEvent,
+    MessageEvent,
+    PhaseChangeEvent,
+    ProposeEvent,
+    RevisionEvent,
+    RoundStartEvent,
+    SynthesisEvent,
+    UserVoteEvent,
+    VoteEvent,
+    WarningEvent,
+)
+from .exceptions import (
+    AragoraError,
+    AuthenticationError,
+    AuthorizationError,
+    ConnectionError,
+    NotFoundError,
+    RateLimitError,
+    ServerError,
+    TimeoutError,
+    ValidationError,
+)
+from .generated_types import (
+    # --- Core debate types ---
+    Agent,
+    # --- Agent registration types ---
+    AgentRegistration,
+    AutoSelectConfig,
+    # --- Budget types ---
+    Budget,
+    # --- Enums ---
+    BudgetAction,
+    BudgetAlert,
+    BudgetAlertListResponse,
+    BudgetCheckRequest,
+    BudgetCheckResponse,
+    BudgetCreateRequest,
+    BudgetListResponse,
+    BudgetOverrideRequest,
+    BudgetOverrideResponse,
+    BudgetPeriod,
+    BudgetStatus,
+    BudgetSummary,
+    BudgetThreshold,
+    BudgetUpdateRequest,
+    # --- Analytics / stats types ---
+    ByAgent,
+    ByTier,
+    Calibration,
+    # --- GitHub code review types ---
+    ChangedFile,
+    # --- Checkpoint types ---
+    CheckpointList,
+    CheckpointMetadata,
+    Code,
+    # --- Codebase / vulnerability types ---
+    CodebaseCVEResponse,
+    CodebaseDependencyAnalysisRequest,
+    CodebaseDependencyAnalysisResponse,
+    CodebaseDependencyScanRequest,
+    CodebaseDependencyScanResponse,
+    CodebaseDuplicateListResponse,
+    CodebaseFileMetricsResponse,
+    CodebaseHotspot,
+    CodebaseHotspotListResponse,
+    CodebaseLicenseCheckRequest,
+    CodebaseLicenseCheckResponse,
+    CodebaseMetricsHistoryResponse,
+    CodebaseMetricsReportResponse,
+    CodebaseMetricsStartResponse,
+    CodebasePackageVulnerabilityResponse,
+    CodebaseSBOMRequest,
+    CodebaseSBOMResponse,
+    CodebaseScanListResponse,
+    CodebaseScanResult,
+    CodebaseScanResultResponse,
+    CodebaseScanStartResponse,
+    CodebaseScanSummary,
+    CodebaseSecretsListResponse,
+    CodebaseSecretsScanListResponse,
+    CodebaseSecretsScanRequest,
+    CodebaseSecretsScanResultResponse,
+    CodebaseSecretsScanStartResponse,
+    CodebaseVulnerabilityListResponse,
+    Comment,
+    Consensus,
+    Consensus1,
+    ConsensusResult,
+    # --- Control plane types ---
+    ControlPlaneAgent,
+    ControlPlaneAgentList,
+    ControlPlaneHealth,
+    ControlPlaneMetrics,
+    ControlPlaneQueue,
+    ControlPlaneQueueJob,
+    ControlPlaneStats,
+    ControlPlaneTask,
+    ControlPlaneTaskClaimResponse,
+    ControlPlaneTaskCreated,
+    # --- Cost management types ---
+    CostAlert,
+    CostAlertsResponse,
+    CostBreakdownItem,
+    CostBreakdownResponse,
+    CostBudgetRequest,
+    CostBudgetResponse,
+    CostDailyItem,
+    CostDismissAlertResponse,
+    CostSummaryResponse,
+    CostTimelineResponse,
+    # --- Explainability types ---
+    Counterfactual,
+    Debate,
+    DebateCreateRequest,
+    DebateCreateResponse,
+    # --- Moments types ---
+    DebateMoment,
+    DebateStatus,
+    DecisionExplanation,
+    # --- Decision / deliberation types ---
+    DecisionList,
+    # --- Gauntlet & receipt types ---
+    DecisionReceipt,
+    DecisionRequest,
+    DecisionResult,
+    DecisionStatus,
+    DecisionSummary,
+    DeliberationQueuedResponse,
+    DeliberationRecord,
+    DeliberationRequest,
+    DeliberationStatus,
+    DeliberationSyncResponse,
+    DependencyInfo,
+    DisagreementStats,
+    EarlyStopStats,
+    Error,
+    Event,
+    # --- Workflow types ---
+    ExecutionList,
+    Factor,
+    # --- Flip & insight types ---
+    FlipsRecent,
+    FlipsSummary,
+    GitHubPRDetails,
+    GitHubPRDetailsResponse,
+    GitHubPRReviewListResponse,
+    GitHubPRReviewResult,
+    GitHubPRReviewStatusResponse,
+    GitHubPRReviewTriggerRequest,
+    GitHubPRReviewTriggerResponse,
+    GitHubPRSubmitReviewRequest,
+    GitHubPRSubmitReviewResponse,
+    GitHubReviewComment,
+    HealthCheck,
+    HeatmapList,
+    Insight,
+    InsightsDetailed,
+    InsightsRecent,
+    # --- Knowledge types ---
+    KnowledgeNode,
+    KnowledgeQueryResult,
+    KnowledgeStoreResult,
+    MemoryStats,
+    Message,
+    MomentsByType,
+    MomentsSummary,
+    MomentsTimeline,
+    MomentsTrending,
+    OAuthProvider,
+    OAuthProviders,
+    PaginatedResponse,
+    # --- Pattern types ---
+    PatternTemplate,
+    PatternTemplateList,
+    PatternType,
+    PolicyEvaluation,
+    PositionFlip,
+    ProvenanceItem,
+    RankingStats,
+    ReceiptList,
+    Relationship,
+    RestoreResult,
+    # --- Retention types ---
+    RetentionPolicy,
+    RetentionPolicyList,
+    ReviewType,
+    RiskHeatmap,
+    Role,
+    RoleRotationStats,
+    Round,
+    # --- Shared inbox types ---
+    RoutingRule,
+    RoutingRuleCreateRequest,
+    RoutingRuleDeleteResponse,
+    RoutingRuleListResponse,
+    RoutingRuleResponse,
+    RoutingRuleTestRequest,
+    RoutingRuleTestResponse,
+    RoutingRuleUpdateRequest,
+    SharedInbox,
+    SharedInboxAssignRequest,
+    SharedInboxCreateRequest,
+    SharedInboxListResponse,
+    SharedInboxMessage,
+    SharedInboxMessageListResponse,
+    SharedInboxMessageResponse,
+    SharedInboxResponse,
+    SharedInboxStatusRequest,
+    SharedInboxTagRequest,
+    StandardSuccessResponse,
+    Status,
+    Status2,
+    StepDefinition,
+    TaskStatus,
+    TopDebate,
+    TransitionRule,
+    TrendingCategory,
+    Type,
+    Type1,
+    VulnerabilityFinding,
+    VulnerabilityReference,
+    Workflow,
+    WorkflowList,
+    WorkflowTemplate,
+    WorkflowTemplateList,
+    WorkflowUpdate,
+    # --- Workspace types ---
+    Workspace,
+    WorkspaceList,
+)
+
+# Alias Code as ErrorCode for clarity in exception handling contexts
+from .generated_types import Code as ErrorCode
+
+# Pagination helpers
+from .pagination import AsyncPaginator, SyncPaginator
+from .websocket import AragoraWebSocket, WebSocketEvent, WebSocketOptions, stream_debate
+
+__version__ = "2.6.3"
+__all__ = [
+    # --- Client ---
+    "AragoraClient",
+    "AragoraAsyncClient",
+    # --- WebSocket ---
+    "AragoraWebSocket",
+    "WebSocketEvent",
+    "WebSocketOptions",
+    "stream_debate",
+    # --- Pagination ---
+    "AsyncPaginator",
+    "SyncPaginator",
+    # --- Typed WebSocket Events ---
+    "EVENT_CLASS_MAP",
+    "AgentMessageEvent",
+    "AudienceSuggestionEvent",
+    "ConnectedEvent",
+    "ConsensusEvent",
+    "ConsensusReachedEvent",
+    "CritiqueEvent",
+    "DebateEndEvent",
+    "DebateStartEvent",
+    "DisconnectedEvent",
+    "ErrorEvent",
+    "MessageEvent",
+    "PhaseChangeEvent",
+    "ProposeEvent",
+    "RevisionEvent",
+    "RoundStartEvent",
+    "SynthesisEvent",
+    "UserVoteEvent",
+    "VoteEvent",
+    "WarningEvent",
+    # --- Exceptions ---
+    "AragoraError",
+    "AuthenticationError",
+    "AuthorizationError",
+    "ConnectionError",
+    "NotFoundError",
+    "RateLimitError",
+    "ServerError",
+    "TimeoutError",
+    "ValidationError",
+    # --- Enums ---
+    "BudgetAction",
+    "BudgetPeriod",
+    "BudgetStatus",
+    "Code",
+    "ErrorCode",
+    "Consensus",
+    "DebateStatus",
+    "Event",
+    "PatternType",
+    "ReviewType",
+    "Role",
+    "Status",
+    "Status2",
+    "TrendingCategory",
+    "Type",
+    "Type1",
+    # --- Core debate types ---
+    "Agent",
+    "AutoSelectConfig",
+    "Calibration",
+    "Consensus1",
+    "ConsensusResult",
+    "Debate",
+    "DebateCreateRequest",
+    "DebateCreateResponse",
+    "Error",
+    "HealthCheck",
+    "Message",
+    "OAuthProvider",
+    "OAuthProviders",
+    "PaginatedResponse",
+    "Relationship",
+    "Round",
+    "StandardSuccessResponse",
+    # --- Workspace types ---
+    "Workspace",
+    "WorkspaceList",
+    # --- Retention types ---
+    "RetentionPolicy",
+    "RetentionPolicyList",
+    # --- Workflow types ---
+    "ExecutionList",
+    "StepDefinition",
+    "TransitionRule",
+    "Workflow",
+    "WorkflowList",
+    "WorkflowTemplate",
+    "WorkflowTemplateList",
+    "WorkflowUpdate",
+    # --- Gauntlet & receipt types ---
+    "DecisionReceipt",
+    "ReceiptList",
+    "RiskHeatmap",
+    "HeatmapList",
+    # --- Pattern types ---
+    "PatternTemplate",
+    "PatternTemplateList",
+    # --- Checkpoint types ---
+    "CheckpointList",
+    "CheckpointMetadata",
+    "RestoreResult",
+    # --- Explainability types ---
+    "Counterfactual",
+    "DecisionExplanation",
+    "Factor",
+    "ProvenanceItem",
+    # --- Control plane types ---
+    "ControlPlaneAgent",
+    "ControlPlaneAgentList",
+    "ControlPlaneHealth",
+    "ControlPlaneMetrics",
+    "ControlPlaneQueue",
+    "ControlPlaneQueueJob",
+    "ControlPlaneStats",
+    "ControlPlaneTask",
+    "ControlPlaneTaskClaimResponse",
+    "ControlPlaneTaskCreated",
+    # --- Codebase / vulnerability types ---
+    "CodebaseCVEResponse",
+    "CodebaseDependencyAnalysisRequest",
+    "CodebaseDependencyAnalysisResponse",
+    "CodebaseDependencyScanRequest",
+    "CodebaseDependencyScanResponse",
+    "CodebaseDuplicateListResponse",
+    "CodebaseFileMetricsResponse",
+    "CodebaseHotspot",
+    "CodebaseHotspotListResponse",
+    "CodebaseLicenseCheckRequest",
+    "CodebaseLicenseCheckResponse",
+    "CodebaseMetricsHistoryResponse",
+    "CodebaseMetricsReportResponse",
+    "CodebaseMetricsStartResponse",
+    "CodebasePackageVulnerabilityResponse",
+    "CodebaseSBOMRequest",
+    "CodebaseSBOMResponse",
+    "CodebaseScanListResponse",
+    "CodebaseScanResult",
+    "CodebaseScanResultResponse",
+    "CodebaseScanStartResponse",
+    "CodebaseScanSummary",
+    "CodebaseSecretsScanListResponse",
+    "CodebaseSecretsScanRequest",
+    "CodebaseSecretsScanResultResponse",
+    "CodebaseSecretsScanStartResponse",
+    "CodebaseSecretsListResponse",
+    "CodebaseVulnerabilityListResponse",
+    "DependencyInfo",
+    "VulnerabilityFinding",
+    "VulnerabilityReference",
+    # --- Decision / deliberation types ---
+    "DecisionList",
+    "DecisionRequest",
+    "DecisionResult",
+    "DecisionStatus",
+    "DecisionSummary",
+    "DeliberationQueuedResponse",
+    "DeliberationRecord",
+    "DeliberationRequest",
+    "DeliberationStatus",
+    "DeliberationSyncResponse",
+    # --- GitHub code review types ---
+    "ChangedFile",
+    "Comment",
+    "GitHubPRDetails",
+    "GitHubPRDetailsResponse",
+    "GitHubPRReviewListResponse",
+    "GitHubPRReviewResult",
+    "GitHubPRReviewStatusResponse",
+    "GitHubPRReviewTriggerRequest",
+    "GitHubPRReviewTriggerResponse",
+    "GitHubPRSubmitReviewRequest",
+    "GitHubPRSubmitReviewResponse",
+    "GitHubReviewComment",
+    # --- Cost management types ---
+    "CostAlert",
+    "CostAlertsResponse",
+    "CostBreakdownItem",
+    "CostBreakdownResponse",
+    "CostBudgetRequest",
+    "CostBudgetResponse",
+    "CostDailyItem",
+    "CostDismissAlertResponse",
+    "CostSummaryResponse",
+    "CostTimelineResponse",
+    # --- Shared inbox types ---
+    "RoutingRule",
+    "RoutingRuleCreateRequest",
+    "RoutingRuleDeleteResponse",
+    "RoutingRuleListResponse",
+    "RoutingRuleResponse",
+    "RoutingRuleTestRequest",
+    "RoutingRuleTestResponse",
+    "RoutingRuleUpdateRequest",
+    "SharedInbox",
+    "SharedInboxAssignRequest",
+    "SharedInboxCreateRequest",
+    "SharedInboxListResponse",
+    "SharedInboxMessage",
+    "SharedInboxMessageListResponse",
+    "SharedInboxMessageResponse",
+    "SharedInboxResponse",
+    "SharedInboxStatusRequest",
+    "SharedInboxTagRequest",
+    # --- Budget types ---
+    "Budget",
+    "BudgetAlert",
+    "BudgetAlertListResponse",
+    "BudgetCheckRequest",
+    "BudgetCheckResponse",
+    "BudgetCreateRequest",
+    "BudgetListResponse",
+    "BudgetOverrideRequest",
+    "BudgetOverrideResponse",
+    "BudgetSummary",
+    "BudgetThreshold",
+    "BudgetUpdateRequest",
+    # --- Analytics / stats types ---
+    "ByAgent",
+    "ByTier",
+    "DisagreementStats",
+    "EarlyStopStats",
+    "MemoryStats",
+    "RankingStats",
+    "RoleRotationStats",
+    # --- Knowledge types ---
+    "KnowledgeNode",
+    "KnowledgeQueryResult",
+    "KnowledgeStoreResult",
+    # --- Agent registration types ---
+    "AgentRegistration",
+    "PolicyEvaluation",
+    "TaskStatus",
+    # --- Flip & insight types ---
+    "FlipsRecent",
+    "FlipsSummary",
+    "Insight",
+    "InsightsDetailed",
+    "InsightsRecent",
+    "PositionFlip",
+    # --- Moments types ---
+    "DebateMoment",
+    "MomentsByType",
+    "MomentsSummary",
+    "MomentsTimeline",
+    "MomentsTrending",
+    "TopDebate",
+]