python-base-agent 2026.2.13__py3-none-any.whl

base_agent/contracts/context_management_notification.py

@@ -0,0 +1,20 @@
+# generated by datamodel-codegen:
+#   filename:  ContextManagementNotification.json
+#   timestamp: 2026-02-10T02:09:52+00:00
+
+from __future__ import annotations
+
+from pydantic import BaseModel
+
+
+class ContextManagementNotification(BaseModel):
+    bytesReduced: int | None = None
+    error: str | None = None
+    finalBytes: int | None = None
+    finalMessages: int | None = None
+    message: str | None = None
+    modelTag: str | None = None
+    originalBytes: int | None = None
+    originalMessages: int | None = None
+    step: int | None = None
+    strategyName: str | None = None

base_agent/contracts/context_management_pipeline.py

@@ -0,0 +1,21 @@
+# generated by datamodel-codegen:
+#   filename:  ContextManagementPipeline.json
+#   timestamp: 2026-02-10T02:09:53+00:00
+
+from __future__ import annotations
+
+from pydantic import BaseModel
+
+
+class ContextManagementPipeline(BaseModel):
+    bytesReduced: int | None = None
+    completed: int | None = None
+    currentBytes: int | None = None
+    currentMessages: int | None = None
+    failed: int | None = None
+    initialBytes: int | None = None
+    initialMessages: int | None = None
+    message: str | None = None
+    pipelineId: str | None = None
+    progressPercent: float | None = None
+    total: int | None = None

base_agent/contracts/cross_vector_similarity_search_request.py

@@ -0,0 +1,14 @@
+# generated by datamodel-codegen:
+#   filename:  CrossVectorSimilaritySearchRequest.json
+#   timestamp: 2026-02-10T02:09:54+00:00
+
+from __future__ import annotations
+
+from pydantic import BaseModel
+
+
+class CrossVectorSimilaritySearchRequest(BaseModel):
+    leftJobId: str
+    rightJobId: str
+    threshold: float
+    topK: int

base_agent/contracts/data_classification_tag.py

@@ -0,0 +1,12 @@
+# generated by datamodel-codegen:
+#   filename:  DataClassificationTag.json
+#   timestamp: 2026-02-10T02:09:55+00:00
+
+from __future__ import annotations
+
+from pydantic import BaseModel
+
+
+class DataClassificationTag(BaseModel):
+    displayName: str | None = None
+    tagId: int | None = None

base_agent/contracts/data_lineage_part.py

@@ -0,0 +1,46 @@
+# generated by datamodel-codegen:
+#   filename:  DataLineagePart.json
+#   timestamp: 2026-02-10T02:09:56+00:00
+
+from __future__ import annotations
+
+from collections.abc import Sequence
+
+from pydantic import BaseModel, Field
+
+from typing import Literal, Any
+
+
+class DataClassificationTag(BaseModel):
+    displayName: str | None = None
+    tagId: int | None = None
+
+
+class MapStringListDataClassificationTag(BaseModel):
+    pass
+
+
+class DataSource(BaseModel):
+    catalog: str | None = None
+    columnTags: MapStringListDataClassificationTag | None = None
+    columns: Sequence[str] | None = None
+    objectStoreLocation: str | None = None
+    schema_: str | None = Field(None, alias='schema')
+    sourceLineageId: str | None = None
+    sourceType: str | None = None
+    table: str | None = None
+    tableTags: Sequence[DataClassificationTag] | None = None
+
+
+class DataLineagePart(BaseModel):
+    lineageId: str | None = None
+    metadata: dict[str, Any] | None = None
+    operationType: str | None = None
+    outputReference: str | None = None
+    partType: str
+    sources: Sequence[DataSource] | None = None
+    timestamp: int | None = None
+
+
+# Rebuild models to resolve forward references with Any type
+DataLineagePart.model_rebuild()

base_agent/contracts/data_part.py

@@ -0,0 +1,15 @@
+# generated by datamodel-codegen:
+#   filename:  DataPart.json
+#   timestamp: 2026-02-10T02:09:56+00:00
+
+from __future__ import annotations
+
+from typing import Literal, Any
+
+from pydantic import BaseModel, Field
+
+
+class DataPart(BaseModel):
+    data: Any
+    name: str | None = None
+    partType: Literal["data"] = "data"

base_agent/contracts/data_pointer_part.py

@@ -0,0 +1,20 @@
+# generated by datamodel-codegen:
+#   filename:  DataPointerPart.json
+#   timestamp: 2026-02-10T02:09:57+00:00
+
+from __future__ import annotations
+
+from pydantic import BaseModel, Field
+
+from typing import Literal
+
+
+class DataPointerPart(BaseModel):
+    columnCount: int
+    dataSizeBytes: int
+    mimeType: str = Field(..., min_length=1)
+    name: str = Field(..., min_length=1)
+    objectStoreLocation: str = Field(..., min_length=1)
+    partType: Literal["data-pointer"] = "data-pointer"
+    rowCount: int
+    storageType: str = Field(..., min_length=1)

base_agent/contracts/embedding_request.py

@@ -0,0 +1,15 @@
+# generated by datamodel-codegen:
+#   filename:  EmbeddingRequest.json
+#   timestamp: 2026-02-10T02:09:58+00:00
+
+from __future__ import annotations
+
+from collections.abc import Sequence
+
+from pydantic import BaseModel
+
+
+class EmbeddingRequest(BaseModel):
+    dimensions: int | None = None
+    model: str | None = None
+    texts: Sequence[str] | None = None

base_agent/contracts/embedding_response.py

@@ -0,0 +1,21 @@
+# generated by datamodel-codegen:
+#   filename:  EmbeddingResponse.json
+#   timestamp: 2026-02-10T02:09:59+00:00
+
+from __future__ import annotations
+
+from collections.abc import Sequence
+
+from pydantic import BaseModel
+
+
+class UsageMetadata(BaseModel):
+    totalTokens: int | None = None
+
+
+class EmbeddingResponse(BaseModel):
+    dimensions: int | None = None
+    embeddings: Sequence[Sequence[float]] | None = None
+    model: str | None = None
+    usage: UsageMetadata | None = None
+    vendor: str | None = None

base_agent/contracts/entity_matching_form_data.py

@@ -0,0 +1,79 @@
+# generated by datamodel-codegen:
+#   filename:  EntityMatchingFormData.json
+#   timestamp: 2026-02-10T02:09:59+00:00
+
+from __future__ import annotations
+
+from collections.abc import Sequence
+from enum import Enum
+
+from pydantic import BaseModel
+
+
+class FieldSelectionMode(Enum):
+    EXCLUDE = 'EXCLUDE'
+    INCLUDE = 'INCLUDE'
+
+
+class LLMJudgeParams(BaseModel):
+    fuzzyCandidatesJobId: str | None = None
+    minimumMatchProbability: float
+    semanticCandidatesJobId: str | None = None
+    systemPromptType: str
+    userPromptType: str
+
+
+class EntityMatchingMode(Enum):
+    SYNTHETIC = 'SYNTHETIC'
+    PRODUCTION = 'PRODUCTION'
+    DE_DUPE = 'DE_DUPE'
+
+
+class FieldConstraint(BaseModel):
+    field: str
+    op: str
+    value: str
+
+
+class SemanticMatchingParams(BaseModel):
+    leftVectorizationJobId: str | None = None
+    rightVectorizationJobId: str | None = None
+    threshold: float
+    topK: int
+    vectorizationJobId: str | None = None
+
+
+class SimpleLinkRule(BaseModel):
+    fields: Sequence[str]
+    name: str
+
+
+class FuzzyMatchingParams(BaseModel):
+    fields: Sequence[str]
+    mode: FieldSelectionMode
+    threshold: float
+
+
+class DeterministicLinkParams(BaseModel):
+    constraints: Sequence[FieldConstraint]
+    linkRules: Sequence[SimpleLinkRule]
+
+
+class EntityMatchingFormData(BaseModel):
+    activeTab: str
+    constraints: Sequence[FieldConstraint]
+    dataLoaded: bool | None = None
+    deterministicLinkParams: DeterministicLinkParams
+    fuzzyMatchingParams: FuzzyMatchingParams
+    goldenTable: str | None = None
+    groupByField: str | None = None
+    leftTable: str
+    leftUidColumn: str
+    linkRules: Sequence[SimpleLinkRule]
+    llmJudgeParams: LLMJudgeParams
+    mode: EntityMatchingMode
+    rightTable: str
+    rightUidColumn: str
+    schemaName: str | None = None
+    schemaSaved: bool | None = None
+    semanticMatchingParams: SemanticMatchingParams

base_agent/contracts/entity_vector_text_search_match.py

@@ -0,0 +1,12 @@
+# generated by datamodel-codegen:
+#   filename:  EntityVectorTextSearchMatch.json
+#   timestamp: 2026-02-10T02:10:04+00:00
+
+from __future__ import annotations
+
+from pydantic import BaseModel
+
+
+class EntityVectorTextSearchMatch(BaseModel):
+    entityId: str
+    similarity: float

base_agent/contracts/entity_vector_text_search_request.py

@@ -0,0 +1,14 @@
+# generated by datamodel-codegen:
+#   filename:  EntityVectorTextSearchRequest.json
+#   timestamp: 2026-02-10T02:10:05+00:00
+
+from __future__ import annotations
+
+from pydantic import BaseModel
+
+
+class EntityVectorTextSearchRequest(BaseModel):
+    jobId: str
+    query: str
+    threshold: float | None = None
+    topK: int | None = None

base_agent/contracts/entity_vector_text_search_result.py

@@ -0,0 +1,24 @@
+# generated by datamodel-codegen:
+#   filename:  EntityVectorTextSearchResult.json
+#   timestamp: 2026-02-10T02:10:05+00:00
+
+from __future__ import annotations
+
+from collections.abc import Sequence
+
+from pydantic import BaseModel
+
+
+class EntityVectorTextSearchMatch(BaseModel):
+    entityId: str
+    similarity: float
+
+
+class EntityVectorTextSearchResult(BaseModel):
+    entityClass: str | None = None
+    entityIdField: str
+    errorMessage: str | None = None
+    jobId: str | None = None
+    matchCount: int
+    matches: Sequence[EntityVectorTextSearchMatch] | None = None
+    status: str

base_agent/contracts/entity_vectorization_job_info.py

@@ -0,0 +1,24 @@
+# generated by datamodel-codegen:
+#   filename:  EntityVectorizationJobInfo.json
+#   timestamp: 2026-02-10T02:10:00+00:00
+
+from __future__ import annotations
+
+from collections.abc import Sequence
+
+from pydantic import BaseModel
+
+
+class EntityVectorizationJobInfo(BaseModel):
+    completedAt: str | None = None
+    createdAt: str
+    entityClass: str
+    entityIdField: str
+    failedCount: int
+    includedFields: Sequence[str]
+    jobId: str
+    modelName: str
+    processedCount: int
+    sessionId: str | None = None
+    status: str
+    totalCount: int | None = None

base_agent/contracts/entity_vectorization_jobs_result.py

@@ -0,0 +1,30 @@
+# generated by datamodel-codegen:
+#   filename:  EntityVectorizationJobsResult.json
+#   timestamp: 2026-02-10T02:10:01+00:00
+
+from __future__ import annotations
+
+from collections.abc import Sequence
+
+from pydantic import BaseModel
+
+
+class EntityVectorizationJobInfo(BaseModel):
+    completedAt: str | None = None
+    createdAt: str
+    entityClass: str
+    entityIdField: str
+    failedCount: int
+    includedFields: Sequence[str]
+    jobId: str
+    modelName: str
+    processedCount: int
+    sessionId: str | None = None
+    status: str
+    totalCount: int | None = None
+
+
+class EntityVectorizationJobsResult(BaseModel):
+    jobs: Sequence[EntityVectorizationJobInfo]
+    status: str
+    total: int

base_agent/contracts/entity_vectorization_progress.py

@@ -0,0 +1,14 @@
+# generated by datamodel-codegen:
+#   filename:  EntityVectorizationProgress.json
+#   timestamp: 2026-02-10T02:10:02+00:00
+
+from __future__ import annotations
+
+from pydantic import BaseModel
+
+
+class EntityVectorizationProgress(BaseModel):
+    completed: int | None = None
+    errorMessage: str | None = None
+    status: str | None = None
+    total: int | None = None

base_agent/contracts/entity_vectorization_request.py

@@ -0,0 +1,15 @@
+# generated by datamodel-codegen:
+#   filename:  EntityVectorizationRequest.json
+#   timestamp: 2026-02-10T02:10:02+00:00
+
+from __future__ import annotations
+
+from collections.abc import Sequence
+
+from pydantic import BaseModel, Field
+
+
+class EntityVectorizationRequest(BaseModel):
+    entityClass: str = Field(..., min_length=1)
+    entityIdField: str = Field(..., min_length=1)
+    selectedFields: Sequence[str]

base_agent/contracts/entity_vectorization_result.py

@@ -0,0 +1,17 @@
+# generated by datamodel-codegen:
+#   filename:  EntityVectorizationResult.json
+#   timestamp: 2026-02-10T02:10:03+00:00
+
+from __future__ import annotations
+
+from collections.abc import Sequence
+
+from pydantic import BaseModel, Field
+
+
+class EntityVectorizationResult(BaseModel):
+    errorMessages: Sequence[str] | None = None
+    failedEntityIds: Sequence[str] | None = None
+    status: str = Field(..., min_length=1)
+    totalFailed: int
+    totalProcessed: int

base_agent/contracts/envelope.py

@@ -0,0 +1,20 @@
+# generated by datamodel-codegen:
+#   filename:  Envelope.json
+#   timestamp: 2026-02-10T02:10:06+00:00
+
+from __future__ import annotations
+
+from typing import Any
+
+from pydantic import BaseModel, Field
+
+
+class MapIntegerString(BaseModel):
+    pass
+
+
+from .header import Header
+
+class Envelope(BaseModel):
+    header: Header
+    payload: Any

base_agent/contracts/error_part.py

@@ -0,0 +1,24 @@
+# generated by datamodel-codegen:
+#   filename:  ErrorPart.json
+#   timestamp: 2026-02-10T02:10:07+00:00
+
+from __future__ import annotations
+
+from enum import Enum
+
+from pydantic import BaseModel, Field
+
+from typing import Literal
+
+
+class ErrorPartType(Enum):
+    SYSTEM_ERROR = 'SYSTEM_ERROR'
+    LLM_ERROR = 'LLM_ERROR'
+    EMBEDDING_ERROR = 'EMBEDDING_ERROR'
+    ERROR = 'ERROR'
+
+
+class ErrorPart(BaseModel):
+    errorType: ErrorPartType
+    partType: Literal["error"] = "error"
+    text: str = Field(..., min_length=1)

base_agent/contracts/error_part_type.py

@@ -0,0 +1,14 @@
+# generated by datamodel-codegen:
+#   filename:  ErrorPartType.json
+#   timestamp: 2026-02-10T02:10:08+00:00
+
+from __future__ import annotations
+
+from enum import Enum
+
+
+class ErrorPartType(Enum):
+    SYSTEM_ERROR = 'SYSTEM_ERROR'
+    LLM_ERROR = 'LLM_ERROR'
+    EMBEDDING_ERROR = 'EMBEDDING_ERROR'
+    ERROR = 'ERROR'

base_agent/contracts/file_part.py

@@ -0,0 +1,17 @@
+# generated by datamodel-codegen:
+#   filename:  FilePart.json
+#   timestamp: 2026-02-10T02:10:09+00:00
+
+from __future__ import annotations
+
+from pydantic import BaseModel, Field
+
+from typing import Literal
+
+
+class FilePart(BaseModel):
+    bytes: str = Field(..., min_length=1)
+    mimeType: str = Field(..., min_length=1)
+    name: str = Field(..., min_length=1)
+    partType: Literal["file"] = "file"
+    uri: str = Field(..., min_length=1)

base_agent/contracts/file_pointer_part.py

@@ -0,0 +1,17 @@
+# generated by datamodel-codegen:
+#   filename:  FilePointerPart.json
+#   timestamp: 2026-02-10T02:10:09+00:00
+
+from __future__ import annotations
+
+from pydantic import BaseModel, Field
+
+from typing import Literal
+
+
+class FilePointerPart(BaseModel):
+    mimeType: str = Field(..., min_length=1)
+    name: str = Field(..., min_length=1)
+    objectStoreLocation: str = Field(..., min_length=1)
+    originalFileName: str | None = None
+    partType: Literal["file-pointer"] = "file-pointer"

base_agent/contracts/file_revectorization_request_event.py

@@ -0,0 +1,19 @@
+# generated by datamodel-codegen:
+#   filename:  FileRevectorizationRequestEvent.json
+#   timestamp: 2026-02-10T02:10:10+00:00
+
+from __future__ import annotations
+
+from pydantic import AwareDatetime, BaseModel
+
+
+class FileRevectorizationRequestEvent(BaseModel):
+    correlationId: str | None = None
+    embeddingModel: str | None = None
+    fileId: int | None = None
+    mimeType: str | None = None
+    objectLocation: str | None = None
+    originalFilename: str | None = None
+    timestamp: AwareDatetime | None = None
+    triggeredByUserId: str | None = None
+    vectorizationJobId: str | None = None

base_agent/contracts/file_vectorization_request_event.py

@@ -0,0 +1,22 @@
+# generated by datamodel-codegen:
+#   filename:  FileVectorizationRequestEvent.json
+#   timestamp: 2026-02-10T02:10:11+00:00
+
+from __future__ import annotations
+
+from pydantic import AwareDatetime, BaseModel
+
+
+class FileVectorizationRequestEvent(BaseModel):
+    correlationId: str | None = None
+    description: str | None = None
+    displayName: str | None = None
+    embeddingModel: str | None = None
+    fileId: int | None = None
+    folderId: int | None = None
+    mimeType: str | None = None
+    objectLocation: str | None = None
+    originalFilename: str | None = None
+    tags: str | None = None
+    timestamp: AwareDatetime | None = None
+    vectorizationJobId: str | None = None

base_agent/contracts/for_each_part.py

@@ -0,0 +1,30 @@
+# generated by datamodel-codegen:
+#   filename:  ForEachPart.json
+#   timestamp: 2026-02-10T02:10:12+00:00
+
+from __future__ import annotations
+
+from collections.abc import Sequence
+
+from pydantic import BaseModel, Field
+
+from typing import Literal, Any
+
+
+class TaskRef(BaseModel):
+    inputPointer: str | None = None
+    overrides: dict[str, Any] | None = None
+    taskSha: str = Field(..., min_length=1)
+
+
+class ForEachPart(BaseModel):
+    arrayPointer: str = Field(..., min_length=1)
+    elementSchemaHash: str = Field(..., min_length=1)
+    name: str = Field(..., min_length=1)
+    parallel: bool
+    partType: Literal["for-each"] = "for-each"
+    tasksToRunPerElement: Sequence[TaskRef] = Field(..., min_length=1)
+
+
+# Rebuild models to resolve forward references with Any type
+TaskRef.model_rebuild()

base_agent/contracts/fuzzy_matching_params.py

@@ -0,0 +1,21 @@
+# generated by datamodel-codegen:
+#   filename:  FuzzyMatchingParams.json
+#   timestamp: 2026-02-10T02:10:12+00:00
+
+from __future__ import annotations
+
+from collections.abc import Sequence
+from enum import Enum
+
+from pydantic import BaseModel
+
+
+class FieldSelectionMode(Enum):
+    EXCLUDE = 'EXCLUDE'
+    INCLUDE = 'INCLUDE'
+
+
+class FuzzyMatchingParams(BaseModel):
+    fields: Sequence[str]
+    mode: FieldSelectionMode
+    threshold: float

base_agent/contracts/group_plan.py

@@ -0,0 +1,16 @@
+# generated by datamodel-codegen:
+#   filename:  GroupPlan.json
+#   timestamp: 2026-02-10T02:10:13+00:00
+
+from __future__ import annotations
+
+from collections.abc import Sequence
+
+from pydantic import BaseModel, Field
+
+
+class GroupPlan(BaseModel):
+    executionMode: str = Field(..., min_length=1)
+    toolCallIds: Sequence[str]
+    toolCount: int | None = None
+    toolName: str = Field(..., min_length=1)