knowledge2 0.4.0__py3-none-any.whl

sdk/models/_registry.py

@@ -0,0 +1,457 @@
+"""Registry mapping TypedDict names to Pydantic model classes."""
+
+from __future__ import annotations
+
+import threading
+from typing import Any
+
+_REGISTRY: dict[str, type] | None = None
+_REGISTRY_LOCK = threading.Lock()
+
+
+def _build_registry() -> dict[str, type]:
+    """Build the TypedDict-name -> PydanticModel mapping."""
+    from sdk.models.a2a import (
+        A2AAgentAuthenticationModel,
+        A2AAgentCapabilitiesModel,
+        A2AAgentCardResponseModel,
+        A2AAgentProviderModel,
+        A2AAgentSkillModel,
+        A2AArtifactModel,
+        A2ADataPartModel,
+        A2AJsonRpcErrorModel,
+        A2AJsonRpcResponseModel,
+        A2ATaskResponseModel,
+        A2ATaskStatusModel,
+        A2ATextPartModel,
+    )
+    from sdk.models.agents import (
+        AgentChatResponseModel,
+        AgentListResponseModel,
+        AgentModelInfoModel,
+        AgentModelsResponseModel,
+        AgentResponseModel,
+        AgentRunListItemModel,
+        AgentRunListResponseModel,
+        AgentRunResponseModel,
+        AgentTaskTypesResponseModel,
+        SourceAgentInfoModel,
+        SubscriptionListResponseModel,
+        SubscriptionResponseModel,
+        TaskTypeInfoModel,
+        ToolConfigResponseModel,
+    )
+    from sdk.models.audit import (
+        AuditLogActionsResponseModel,
+        AuditLogItemModel,
+        AuditLogListResponseModel,
+    )
+    from sdk.models.auth import (
+        ApiKeyCreateResponseModel,
+        ApiKeyListItemModel,
+        ApiKeyListResponseModel,
+        ApiKeyRevokeResponseModel,
+        ApiKeyRotateResponseModel,
+        WhoAmIResponseModel,
+    )
+    from sdk.models.chunks import (
+        ChunkListItemModel,
+        ChunkListResponseModel,
+    )
+    from sdk.models.common import (
+        ApiErrorDetailModel,
+    )
+    from sdk.models.console import (
+        ConsoleBootstrapResponseModel,
+        ConsoleMeResponseModel,
+        ConsoleOrgResponseModel,
+        ConsoleProjectItemModel,
+        ConsoleProjectListResponseModel,
+        ConsoleSummaryResponseModel,
+        InviteAcceptResponseModel,
+        InviteCreateResponseModel,
+        InviteListItemModel,
+        InviteListResponseModel,
+        MemberRemoveResponseModel,
+        MemberUpdateResponseModel,
+        TeamListResponseModel,
+        TeamMemberModel,
+    )
+    from sdk.models.corpora import (
+        CorpusDeleteResponseModel,
+        CorpusListResponseModel,
+        CorpusResponseModel,
+        CorpusStatusResponseModel,
+    )
+    from sdk.models.deployments import (
+        DeploymentResponseModel,
+    )
+    from sdk.models.documents import (
+        ChunkingConfigModel,
+        DocumentBatchItemModel,
+        DocumentBatchUploadResponseModel,
+        DocumentCreateResponseModel,
+        DocumentDeleteResponseModel,
+        DocumentDetailResponseModel,
+        DocumentListItemModel,
+        DocumentListResponseModel,
+        DocumentManifestIngestResponseModel,
+        DocumentUrlIngestResponseModel,
+        DocumentUrlItemModel,
+    )
+    from sdk.models.embeddings import (
+        EmbeddingItemModel,
+        EmbeddingModelInfoModel,
+        EmbeddingModelListResponseModel,
+        EmbeddingsResponseModel,
+    )
+    from sdk.models.evaluation import (
+        EvalRunDetailResponseModel,
+        EvalRunResponseModel,
+    )
+    from sdk.models.feedback import (
+        FeedbackResponseModel,
+    )
+    from sdk.models.feeds import (
+        FeedListResponseModel,
+        FeedResponseModel,
+        FeedRunResponseModel,
+        FeedSubscriptionResponseModel,
+    )
+    from sdk.models.indexes import (
+        IndexBuildResponseModel,
+        IndexCompactResponseModel,
+        IndexStatusResponseModel,
+    )
+    from sdk.models.jobs import (
+        JobActionsModel,
+        JobListItemModel,
+        JobListResponseModel,
+        JobResponseModel,
+        JobStatusResponseModel,
+        ReconcileJobsResponseModel,
+    )
+    from sdk.models.models import (
+        ModelDeleteResponseModel,
+        ModelListResponseModel,
+        ModelResponseModel,
+    )
+    from sdk.models.onboarding import (
+        DatasetAnalysisDetailsModel,
+        DatasetAnalysisRequestModel,
+        DatasetAnalysisResponseModel,
+        DatasetAnalysisSummaryModel,
+        DocumentSummaryResponseModel,
+        EvaluationDetailsModel,
+        EvaluationListResponseModel,
+        EvaluationMetricsModel,
+        EvaluationReportResponseModel,
+        EvaluationRequestModel,
+        EvaluationResponseModel,
+        EvaluationSummaryModel,
+        GoldLabelChunkReferenceModel,
+        GoldLabelDocumentReferenceModel,
+        GoldLabelEntryModel,
+        GoldLabelListItemModel,
+        GoldLabelsListResponseModel,
+        GoldLabelsUploadRequestModel,
+        GoldLabelsUploadResponseModel,
+        LexicalStrategyModel,
+        OnboardingStatusResponseModel,
+        ResolvedLabelInfoModel,
+        SearcherPersonaModel,
+        SummarizationRequestModel,
+        SummarizationResponseModel,
+        SummarizationStatusResponseModel,
+        SyntheticQueryBatchDetailsModel,
+        SyntheticQueryBatchListResponseModel,
+        SyntheticQueryBatchResponseModel,
+        SyntheticQueryBatchSummaryModel,
+        SyntheticQueryGenerationRequestModel,
+        SyntheticQuerySampleModel,
+    )
+    from sdk.models.orgs import (
+        OrgResponseModel,
+    )
+    from sdk.models.pipelines import (
+        ApplyResultModel,
+        ArchiveResultModel,
+        DraftActivateResultModel,
+        DriftReportModel,
+        DriftSummaryModel,
+        DryRunResultModel,
+        EntityDiffModel,
+        FieldDiffModel,
+        GraphEdgeModel,
+        GraphNodeModel,
+        GraphResponseModel,
+        PipelineSpecListResponseModel,
+        PipelineSpecResponseModel,
+        RefreshChangesModel,
+        RefreshResultModel,
+        SkippedEntityModel,
+        ValidationIssueModel,
+    )
+    from sdk.models.projects import (
+        ProjectListResponseModel,
+        ProjectResponseModel,
+    )
+    from sdk.models.search import (
+        MetadataFilterModel,
+        MetadataFiltersModel,
+        SearchBatchOptionsModel,
+        SearchBatchResponseModel,
+        SearchColdStartMetaModel,
+        SearchGenerateOptionsModel,
+        SearchGenerateResponseModel,
+        SearchGenerationConfigModel,
+        SearchGraphRagConfigModel,
+        SearchGraphRagMetaModel,
+        SearchHybridConfigModel,
+        SearchMetaModel,
+        SearchOptionsModel,
+        SearchRerankConfigModel,
+        SearchResponseModel,
+        SearchResultModel,
+        SearchReturnConfigModel,
+    )
+    from sdk.models.training import (
+        CancelTuningRunResponseModel,
+        PromoteResponseModel,
+        TrainingDataBuildResponseModel,
+        TrainingDatasetListResponseModel,
+        TrainingDatasetResponseModel,
+        TuningRunBuildResponseModel,
+        TuningRunDetailResponseModel,
+        TuningRunListResponseModel,
+        TuningRunLogsResponseModel,
+        TuningRunResponseModel,
+    )
+    from sdk.models.usage import (
+        UsageByCorpusItemModel,
+        UsageByCorpusResponseModel,
+        UsageByKeyItemModel,
+        UsageByKeyResponseModel,
+        UsageSeriesPointModel,
+        UsageSummaryResponseModel,
+    )
+
+    return {
+        # a2a
+        "A2ATextPart": A2ATextPartModel,
+        "A2ADataPart": A2ADataPartModel,
+        "A2AArtifact": A2AArtifactModel,
+        "A2ATaskStatus": A2ATaskStatusModel,
+        "A2ATaskResponse": A2ATaskResponseModel,
+        "A2AJsonRpcError": A2AJsonRpcErrorModel,
+        "A2AJsonRpcResponse": A2AJsonRpcResponseModel,
+        "A2AAgentSkill": A2AAgentSkillModel,
+        "A2AAgentCapabilities": A2AAgentCapabilitiesModel,
+        "A2AAgentProvider": A2AAgentProviderModel,
+        "A2AAgentAuthentication": A2AAgentAuthenticationModel,
+        "A2AAgentCardResponse": A2AAgentCardResponseModel,
+        # agents
+        "SourceAgentInfo": SourceAgentInfoModel,
+        "ToolConfigResponse": ToolConfigResponseModel,
+        "AgentResponse": AgentResponseModel,
+        "AgentListResponse": AgentListResponseModel,
+        "AgentChatResponse": AgentChatResponseModel,
+        "AgentModelInfo": AgentModelInfoModel,
+        "AgentModelsResponse": AgentModelsResponseModel,
+        "TaskTypeInfo": TaskTypeInfoModel,
+        "AgentTaskTypesResponse": AgentTaskTypesResponseModel,
+        "AgentRunResponse": AgentRunResponseModel,
+        "AgentRunListItem": AgentRunListItemModel,
+        "AgentRunListResponse": AgentRunListResponseModel,
+        "SubscriptionResponse": SubscriptionResponseModel,
+        "SubscriptionListResponse": SubscriptionListResponseModel,
+        # audit
+        "AuditLogItem": AuditLogItemModel,
+        "AuditLogListResponse": AuditLogListResponseModel,
+        "AuditLogActionsResponse": AuditLogActionsResponseModel,
+        # auth
+        "ApiKeyCreateResponse": ApiKeyCreateResponseModel,
+        "ApiKeyRotateResponse": ApiKeyRotateResponseModel,
+        "ApiKeyRevokeResponse": ApiKeyRevokeResponseModel,
+        "ApiKeyListItem": ApiKeyListItemModel,
+        "ApiKeyListResponse": ApiKeyListResponseModel,
+        "WhoAmIResponse": WhoAmIResponseModel,
+        # chunks
+        "ChunkListItem": ChunkListItemModel,
+        "ChunkListResponse": ChunkListResponseModel,
+        # common
+        "ApiErrorDetail": ApiErrorDetailModel,
+        # console
+        "ConsoleMeResponse": ConsoleMeResponseModel,
+        "ConsoleBootstrapResponse": ConsoleBootstrapResponseModel,
+        "ConsoleSummaryResponse": ConsoleSummaryResponseModel,
+        "ConsoleProjectItem": ConsoleProjectItemModel,
+        "ConsoleProjectListResponse": ConsoleProjectListResponseModel,
+        "ConsoleOrgResponse": ConsoleOrgResponseModel,
+        "TeamMember": TeamMemberModel,
+        "TeamListResponse": TeamListResponseModel,
+        "InviteListItem": InviteListItemModel,
+        "InviteListResponse": InviteListResponseModel,
+        "InviteCreateResponse": InviteCreateResponseModel,
+        "InviteAcceptResponse": InviteAcceptResponseModel,
+        "MemberUpdateResponse": MemberUpdateResponseModel,
+        "MemberRemoveResponse": MemberRemoveResponseModel,
+        # corpora
+        "CorpusResponse": CorpusResponseModel,
+        "CorpusListResponse": CorpusListResponseModel,
+        "CorpusDeleteResponse": CorpusDeleteResponseModel,
+        "CorpusStatusResponse": CorpusStatusResponseModel,
+        # deployments
+        "DeploymentResponse": DeploymentResponseModel,
+        # documents
+        "ChunkingConfig": ChunkingConfigModel,
+        "DocumentCreateResponse": DocumentCreateResponseModel,
+        "DocumentBatchItem": DocumentBatchItemModel,
+        "DocumentUrlItem": DocumentUrlItemModel,
+        "DocumentUrlIngestResponse": DocumentUrlIngestResponseModel,
+        "DocumentManifestIngestResponse": DocumentManifestIngestResponseModel,
+        "DocumentListItem": DocumentListItemModel,
+        "DocumentListResponse": DocumentListResponseModel,
+        "DocumentDetailResponse": DocumentDetailResponseModel,
+        "DocumentDeleteResponse": DocumentDeleteResponseModel,
+        "DocumentBatchUploadResponse": DocumentBatchUploadResponseModel,
+        # embeddings
+        "EmbeddingItem": EmbeddingItemModel,
+        "EmbeddingModelInfo": EmbeddingModelInfoModel,
+        "EmbeddingModelListResponse": EmbeddingModelListResponseModel,
+        "EmbeddingsResponse": EmbeddingsResponseModel,
+        # evaluation
+        "EvalRunResponse": EvalRunResponseModel,
+        "EvalRunDetailResponse": EvalRunDetailResponseModel,
+        # feedback
+        "FeedbackResponse": FeedbackResponseModel,
+        # feeds
+        "FeedSubscriptionResponse": FeedSubscriptionResponseModel,
+        "FeedResponse": FeedResponseModel,
+        "FeedListResponse": FeedListResponseModel,
+        "FeedRunResponse": FeedRunResponseModel,
+        # indexes
+        "IndexBuildResponse": IndexBuildResponseModel,
+        "IndexStatusResponse": IndexStatusResponseModel,
+        "IndexCompactResponse": IndexCompactResponseModel,
+        # jobs
+        "JobActions": JobActionsModel,
+        "JobResponse": JobResponseModel,
+        "JobListItem": JobListItemModel,
+        "JobListResponse": JobListResponseModel,
+        "JobStatusResponse": JobStatusResponseModel,
+        "ReconcileJobsResponse": ReconcileJobsResponseModel,
+        # models
+        "ModelResponse": ModelResponseModel,
+        "ModelListResponse": ModelListResponseModel,
+        "ModelDeleteResponse": ModelDeleteResponseModel,
+        # onboarding
+        "GoldLabelDocumentReference": GoldLabelDocumentReferenceModel,
+        "GoldLabelChunkReference": GoldLabelChunkReferenceModel,
+        "GoldLabelEntry": GoldLabelEntryModel,
+        "GoldLabelsUploadRequest": GoldLabelsUploadRequestModel,
+        "ResolvedLabelInfo": ResolvedLabelInfoModel,
+        "GoldLabelsUploadResponse": GoldLabelsUploadResponseModel,
+        "GoldLabelListItem": GoldLabelListItemModel,
+        "GoldLabelsListResponse": GoldLabelsListResponseModel,
+        "DatasetAnalysisRequest": DatasetAnalysisRequestModel,
+        "DatasetAnalysisResponse": DatasetAnalysisResponseModel,
+        "DatasetAnalysisSummary": DatasetAnalysisSummaryModel,
+        "OnboardingStatusResponse": OnboardingStatusResponseModel,
+        "SearcherPersona": SearcherPersonaModel,
+        "LexicalStrategy": LexicalStrategyModel,
+        "DatasetAnalysisDetails": DatasetAnalysisDetailsModel,
+        "SyntheticQueryGenerationRequest": SyntheticQueryGenerationRequestModel,
+        "SyntheticQueryBatchResponse": SyntheticQueryBatchResponseModel,
+        "SyntheticQueryBatchSummary": SyntheticQueryBatchSummaryModel,
+        "SyntheticQueryBatchListResponse": SyntheticQueryBatchListResponseModel,
+        "SyntheticQuerySample": SyntheticQuerySampleModel,
+        "SyntheticQueryBatchDetails": SyntheticQueryBatchDetailsModel,
+        "EvaluationRequest": EvaluationRequestModel,
+        "EvaluationResponse": EvaluationResponseModel,
+        "EvaluationMetrics": EvaluationMetricsModel,
+        "EvaluationSummary": EvaluationSummaryModel,
+        "EvaluationListResponse": EvaluationListResponseModel,
+        "EvaluationDetails": EvaluationDetailsModel,
+        "EvaluationReportResponse": EvaluationReportResponseModel,
+        "SummarizationRequest": SummarizationRequestModel,
+        "SummarizationResponse": SummarizationResponseModel,
+        "SummarizationStatusResponse": SummarizationStatusResponseModel,
+        "DocumentSummaryResponse": DocumentSummaryResponseModel,
+        # orgs
+        "OrgResponse": OrgResponseModel,
+        # pipelines
+        "ValidationIssue": ValidationIssueModel,
+        "PipelineSpecResponse": PipelineSpecResponseModel,
+        "PipelineSpecListResponse": PipelineSpecListResponseModel,
+        "DryRunResult": DryRunResultModel,
+        "ApplyResult": ApplyResultModel,
+        "SkippedEntity": SkippedEntityModel,
+        "ArchiveResult": ArchiveResultModel,
+        "DraftActivateResult": DraftActivateResultModel,
+        "FieldDiff": FieldDiffModel,
+        "EntityDiff": EntityDiffModel,
+        "DriftSummary": DriftSummaryModel,
+        "DriftReport": DriftReportModel,
+        "RefreshChanges": RefreshChangesModel,
+        "RefreshResult": RefreshResultModel,
+        "GraphNode": GraphNodeModel,
+        "GraphEdge": GraphEdgeModel,
+        "GraphResponse": GraphResponseModel,
+        # projects
+        "ProjectResponse": ProjectResponseModel,
+        "ProjectListResponse": ProjectListResponseModel,
+        # search
+        "SearchHybridConfig": SearchHybridConfigModel,
+        "SearchRerankConfig": SearchRerankConfigModel,
+        "SearchGraphRagConfig": SearchGraphRagConfigModel,
+        "SearchReturnConfig": SearchReturnConfigModel,
+        "MetadataFilter": MetadataFilterModel,
+        "MetadataFilters": MetadataFiltersModel,
+        "SearchGenerationConfig": SearchGenerationConfigModel,
+        "SearchOptions": SearchOptionsModel,
+        "SearchGenerateOptions": SearchGenerateOptionsModel,
+        "SearchBatchOptions": SearchBatchOptionsModel,
+        "SearchResult": SearchResultModel,
+        "SearchColdStartMeta": SearchColdStartMetaModel,
+        "SearchGraphRagMeta": SearchGraphRagMetaModel,
+        "SearchMeta": SearchMetaModel,
+        "SearchResponse": SearchResponseModel,
+        "SearchBatchResponse": SearchBatchResponseModel,
+        "SearchGenerateResponse": SearchGenerateResponseModel,
+        # training
+        "TrainingDataBuildResponse": TrainingDataBuildResponseModel,
+        "TrainingDatasetResponse": TrainingDatasetResponseModel,
+        "TrainingDatasetListResponse": TrainingDatasetListResponseModel,
+        "TuningRunResponse": TuningRunResponseModel,
+        "TuningRunBuildResponse": TuningRunBuildResponseModel,
+        "TuningRunDetailResponse": TuningRunDetailResponseModel,
+        "TuningRunLogsResponse": TuningRunLogsResponseModel,
+        "TuningRunListResponse": TuningRunListResponseModel,
+        "PromoteResponse": PromoteResponseModel,
+        "CancelTuningRunResponse": CancelTuningRunResponseModel,
+        # usage
+        "UsageSeriesPoint": UsageSeriesPointModel,
+        "UsageSummaryResponse": UsageSummaryResponseModel,
+        "UsageByCorpusItem": UsageByCorpusItemModel,
+        "UsageByCorpusResponse": UsageByCorpusResponseModel,
+        "UsageByKeyItem": UsageByKeyItemModel,
+        "UsageByKeyResponse": UsageByKeyResponseModel,
+    }
+
+
+def get_model_for_type(type_name: str) -> type | None:
+    """Look up the Pydantic model class for a given TypedDict name.
+
+    Args:
+        type_name: The TypedDict class name (e.g. "SearchResponse").
+
+    Returns:
+        The corresponding Pydantic model class, or None if not found.
+    """
+    global _REGISTRY
+    if _REGISTRY is None:
+        with _REGISTRY_LOCK:
+            if _REGISTRY is None:
+                _REGISTRY = _build_registry()
+    return _REGISTRY.get(type_name)

sdk/models/a2a.py

@@ -0,0 +1,92 @@
+"""Pydantic models for A2A (Agent-to-Agent) types."""
+
+from __future__ import annotations
+
+from typing import Any, Literal
+
+from sdk.models._base import K2BaseModel
+
+
+class A2ATextPartModel(K2BaseModel):
+    kind: Literal["text"]
+    text: str
+
+
+class A2ADataPartModel(K2BaseModel):
+    kind: Literal["data"]
+    data: dict[str, Any]
+
+
+class A2AArtifactModel(K2BaseModel):
+    artifactId: str | None = None
+    name: str | None = None
+    parts: list[A2ATextPartModel | A2ADataPartModel] | None = None
+    metadata: dict[str, Any] | None = None
+
+
+class A2ATaskStatusModel(K2BaseModel):
+    state: str | None = None
+    timestamp: str | None = None
+    message: dict[str, Any] | None = None
+
+
+class A2ATaskResponseModel(K2BaseModel):
+    id: str | None = None
+    contextId: str | None = None
+    status: A2ATaskStatusModel | None = None
+    artifacts: list[A2AArtifactModel] | None = None
+    history: list[dict[str, Any]] | None = None
+    metadata: dict[str, Any] | None = None
+    kind: Literal["task"] | None = None
+
+
+class A2AJsonRpcErrorModel(K2BaseModel):
+    code: int | None = None
+    message: str | None = None
+    data: dict[str, Any] | None = None
+
+
+class A2AJsonRpcResponseModel(K2BaseModel):
+    jsonrpc: str | None = None
+    id: str | int | None = None
+    result: A2ATaskResponseModel | None = None
+    error: A2AJsonRpcErrorModel | None = None
+
+
+class A2AAgentSkillModel(K2BaseModel):
+    id: str | None = None
+    name: str | None = None
+    description: str | None = None
+    tags: list[str] | None = None
+    examples: list[str] | None = None
+    inputModes: list[str] | None = None
+    outputModes: list[str] | None = None
+
+
+class A2AAgentCapabilitiesModel(K2BaseModel):
+    streaming: bool | None = None
+    pushNotifications: bool | None = None
+    stateTransitionHistory: bool | None = None
+
+
+class A2AAgentProviderModel(K2BaseModel):
+    organization: str | None = None
+    url: str | None = None
+
+
+class A2AAgentAuthenticationModel(K2BaseModel):
+    schemes: list[str] | None = None
+    credentials: str | None = None
+
+
+class A2AAgentCardResponseModel(K2BaseModel):
+    name: str | None = None
+    description: str | None = None
+    url: str | None = None
+    provider: A2AAgentProviderModel | None = None
+    version: str | None = None
+    capabilities: A2AAgentCapabilitiesModel | None = None
+    authentication: A2AAgentAuthenticationModel | None = None
+    defaultInputModes: list[str] | None = None
+    defaultOutputModes: list[str] | None = None
+    skills: list[A2AAgentSkillModel] | None = None

sdk/models/agents.py

@@ -0,0 +1,109 @@
+"""Pydantic models for agent types."""
+
+from __future__ import annotations
+
+from typing import Any
+
+from sdk.models._base import K2BaseModel
+
+
+class SourceAgentInfoModel(K2BaseModel):
+    agent_id: str | None = None
+    agent_name: str | None = None
+    mode: str | None = None
+
+
+class ToolConfigResponseModel(K2BaseModel):
+    webhook_url: str | None = None
+    webhook_secret: str | None = None
+    webhook_retries: int | None = None
+
+
+class AgentResponseModel(K2BaseModel):
+    id: str | None = None
+    org_id: str | None = None
+    project_id: str | None = None
+    name: str | None = None
+    slug: str | None = None
+    description: str | None = None
+    corpus_id: str | None = None
+    task_type: str | None = None
+    instructions: str | None = None
+    model: str | None = None
+    status: str | None = None
+    parent_agent_id: str | None = None
+    source_agents: list[SourceAgentInfoModel] | None = None
+    tool_config: ToolConfigResponseModel | None = None
+    created_at: str | None = None
+    updated_at: str | None = None
+
+
+class AgentListResponseModel(K2BaseModel):
+    agents: list[AgentResponseModel]
+    total: int
+    limit: int
+    offset: int
+
+
+class AgentChatResponseModel(K2BaseModel):
+    answer: str | None = None
+    model: str | None = None
+    results: list[dict[str, Any]] | None = None
+    used_sources: list[str] | None = None
+    meta: dict[str, Any] | None = None
+
+
+class AgentModelInfoModel(K2BaseModel):
+    id: str | None = None
+    name: str | None = None
+    provider: str | None = None
+    default: bool | None = None
+
+
+class AgentModelsResponseModel(K2BaseModel):
+    models: list[AgentModelInfoModel]
+
+
+class TaskTypeInfoModel(K2BaseModel):
+    id: str | None = None
+    name: str | None = None
+    description: str | None = None
+    output_schema: str | None = None
+    supports_chat: bool | None = None
+
+
+class AgentTaskTypesResponseModel(K2BaseModel):
+    task_types: list[TaskTypeInfoModel]
+
+
+class AgentRunResponseModel(K2BaseModel):
+    job_id: str
+    status: str
+
+
+class AgentRunListItemModel(K2BaseModel):
+    job_id: str | None = None
+    status: str | None = None
+    created_at: str | None = None
+    completed_at: str | None = None
+    result: dict[str, Any] | None = None
+    error_message: str | None = None
+
+
+class AgentRunListResponseModel(K2BaseModel):
+    runs: list[AgentRunListItemModel]
+    total: int
+
+
+class SubscriptionResponseModel(K2BaseModel):
+    id: str | None = None
+    agent_id: str | None = None
+    feed_id: str | None = None
+    feed_name: str | None = None
+    feed_activation_status: str | None = None
+    role: str | None = None
+    created_at: str | None = None
+
+
+class SubscriptionListResponseModel(K2BaseModel):
+    subscriptions: list[SubscriptionResponseModel]

sdk/models/audit.py

@@ -0,0 +1,28 @@
+"""Pydantic models for audit types."""
+
+from __future__ import annotations
+
+from sdk.models._base import K2BaseModel
+
+
+class AuditLogItemModel(K2BaseModel):
+    id: str | None = None
+    action: str | None = None
+    entity_type: str | None = None
+    entity_id: str | None = None
+    org_id: str | None = None
+    project_id: str | None = None
+    corpus_id: str | None = None
+    api_key_id: str | None = None
+    user_id: str | None = None
+    payload: dict | None = None
+    created_at: str | None = None
+
+
+class AuditLogListResponseModel(K2BaseModel):
+    logs: list[AuditLogItemModel]
+    total: int
+
+
+class AuditLogActionsResponseModel(K2BaseModel):
+    actions: list[str]