mutagent-sdk 0.1.0__py3-none-any.whl

mutagent/__init__.py

@@ -0,0 +1,231 @@
+"""
+mutagent SDK -- public API surface.
+
+This file is GENERATED by mutagent-xgen. Do not edit manually.
+
+Usage::
+
+    from mutagent import AsyncMutagent, Mutagent
+    from mutagent.models import *
+"""
+from __future__ import annotations
+
+# region: module-top
+# endregion: module-top
+
+from .client import Mutagent
+from .async_client import AsyncMutagent
+from .errors import MutagentError, SDKError, ResponseValidationError, SDKValidationError
+
+__all__ = [
+    "Mutagent",
+    "AsyncMutagent",
+    "MutagentError",
+    "SDKError",
+    "ResponseValidationError",
+    "SDKValidationError",
+    "Prompt",
+    "PromptList",
+    "PromptDataset",
+    "PromptDatasetList",
+    "PromptDatasetItem",
+    "PromptDatasetItemList",
+    "PromptDatasetExport",
+    "Evaluation",
+    "EvaluationList",
+    "EvaluationResult",
+    "EvaluationResultList",
+    "EvaluationHistory",
+    "EvaluationRun",
+    "EvaluationAggregationResults",
+    "PromptSuccessResponse",
+    "ErrorResponse",
+    "ErrorWithStatusResponse",
+    "SuccessResponse",
+    "DetailedErrorResponse",
+    "OptimizationJob",
+    "OptimizationJobStatus",
+    "OptimizationJobList",
+    "OptimizationScoreProgression",
+    "OptimizationStates",
+    "OptimizationResults",
+    "OptimizationError",
+    "Agent",
+    "AgentList",
+    "AgentSuccessResponse",
+    "Organization",
+    "OrganizationWithRole",
+    "OrganizationList",
+    "OrganizationOrError",
+    "OrganizationListOrError",
+    "OrgSuccessOrError",
+    "OrgMemberCountOrError",
+    "Workspace",
+    "WorkspaceWithRole",
+    "WorkspacesList",
+    "WsSuccess",
+    "WsError",
+    "ProviderConfig",
+    "ProviderConfigsList",
+    "TestConnectionResult",
+    "ModelsList",
+    "ProviderSuccess",
+    "ProviderError",
+    "TraceIngestionResponse",
+    "BatchIngestionResponse",
+    "TraceWithSpans",
+    "PaginatedTraceList",
+    "TraceStats",
+    "AnalyticsSummary",
+    "TraceError",
+    "Schemanullab",
+    "Schemaformat",
+    "Schemaformat2",
+    "ErrorMessage",
+    "ErrorMessageStatusCode",
+    "NameEmailAvatarUrl",
+    "SchemamaxLen",
+    "Schemaformat3",
+    "Schemaformat4",
+    "Schemapatter",
+    "CurrentPasswordNewPassword",
+    "SchemamaxLen2",
+    "Schemamaximu",
+    "Schemaminimu",
+    "Schemaminimu2",
+    "NameDescriptionMetadata",
+    "SchemamaxLen3",
+    "Schemaitems",
+    "TargetPromptIdTargetPromptGroupIdNewName",
+    "SchemamaxLen4",
+    "Schemaitems2",
+    "NameInputExpectedOutput",
+    "Schemaitems3",
+    "NameInputExpectedOutput2",
+    "PromptIdDatasetIdName",
+    "CriteriaThresholdModel",
+    "ArrayOfIdCriteriaEvaluationParameter",
+    "IdCriteriaEvaluationParameter",
+    "SchemaminLen",
+    "Schemamaximu2",
+    "NameDescriptionEvalConfig",
+    "PromptIdPromptGroupIdEvaluationId",
+    "SchemaanyOf",
+    "Schemadefaul",
+    "Schemaminimu3",
+    "ArrayOfIdModelTemperature",
+    "IdModelTemperature",
+    "Schemamaximu3",
+    "SchemaanyOf2",
+    "Schemaminimu4",
+    "StatusSummaryCompletedAt",
+    "Schemaenum",
+    "TotalEvaluationsCompletedEvaluationsPerModelStats",
+    "DatasetItemIdDatasetItemIndexModelColumnId",
+    "Schemamaximu4",
+    "Schemanullab2",
+    "ArrayOfNameScorePassed",
+    "NameScorePassed",
+    "Schemanullab3",
+    "ArrayOfIdPromptIdWorkspaceId",
+    "IdPromptIdWorkspaceId",
+    "PromptIdInputsModel",
+    "Schemaenum2",
+    "InputOutputTotal",
+    "PromptIdExecutions",
+    "ArrayOfInputsModelActualOutput",
+    "InputsModelActualOutput",
+    "InputLlmConfig",
+    "ModelTemperatureMaxTokens",
+    "Schemadefaul2",
+    "Schemadefaul3",
+    "Schemadefaul4",
+    "InputEvaluationIdLlmConfig",
+    "NameDescriptionVersion",
+    "Schemapatter2",
+    "ArrayOfRoleContent",
+    "RoleContent",
+    "NameDescriptionIsLatest",
+    "NewVersionChanges",
+    "NameDescriptionMetadata2",
+    "DatasetIdEvaluationIdConfig",
+    "MaxIterationsTargetScorePatience",
+    "Schemadefaul5",
+    "Schemamaximu5",
+    "SchemamaxLen5",
+    "SchemamaxLen6",
+    "NameSlugDescription",
+    "SchemamaxLen7",
+    "NameDescriptionSystemPrompt",
+    "NameSlugDescription2",
+    "SchemamaxLen8",
+    "NameSlugDescription3",
+    "SchemamaxLen9",
+    "Schemaformat5",
+    "Schemanullab4",
+    "NameSlugDescription4",
+    "SchemamaxLen10",
+    "NameSlugDescription5",
+    "SchemamaxLen11",
+    "IdUserIdRole",
+    "IdNameEmail",
+    "UserIdRoleStatus",
+    "Schemadefaul6",
+    "Schemadefaul7",
+    "SuccessMessage",
+    "IdUserIdRole2",
+    "UserIdRole",
+    "Schemadefaul8",
+    "NameProviderApiKey",
+    "Schemadefaul9",
+    "OrganizationIdWorkspaceIdUserId",
+    "Schemaformat6",
+    "Schemadefaul10",
+    "ArrayOfIdDisplayNameFamily",
+    "IdDisplayNameFamily",
+    "Schemaformat7",
+    "NameApiKeyBaseUrl",
+    "Schemaitems4",
+    "IdOrganizationIdEmail",
+    "EmailRoleExpiresInDays",
+    "Schemadefaul11",
+    "Schemamaximu6",
+    "TraceIdSessionIdName",
+    "SchemamaxLen12",
+    "Schemadefaul12",
+    "Schemadefaul13",
+    "ArrayOfSpanIdParentSpanIdName",
+    "SpanIdParentSpanIdName",
+    "SchemamaxLen13",
+    "Schemadefaul14",
+    "InputOutput",
+    "ArrayOfNameTimestampAttributes",
+    "NameTimestampAttributes",
+    "Schemaitems5",
+    "ArrayOfResourceScopeSpans",
+    "ResourceScopeSpans",
+    "ArrayOfKeyValue",
+    "KeyValue",
+    "StringValueIntValueDoubleValue",
+    "ArrayOfSpans",
+    "ArrayOfTraceIdSpanIdParentSpanId",
+    "TraceIdSpanIdParentSpanId",
+    "CodeMessage",
+    "ArrayOfNameTimeUnixNanoAttributes",
+    "NameTimeUnixNanoAttributes",
+]
+
+# region: exports-extra
+from .tracing import (
+    init_tracing,
+    shutdown_tracing,
+    is_tracing_initialized,
+    trace,
+    with_trace,
+    start_span,
+    end_span,
+    get_current_span,
+    get_current_trace_id,
+    run_in_span_context,
+)
+# endregion: exports-extra

mutagent/_funcs/__init__.py

@@ -0,0 +1,125 @@
+"""
+Auto-generated module index.
+This file is GENERATED by mutagent-xgen. Do not edit manually.
+"""
+from __future__ import annotations
+
+__all__ = [
+    "getProfile",
+    "updateProfile",
+    "deleteAccount",
+    "changePassword",
+    "listSessions",
+    "deleteSession",
+    "listPromptDatasets",
+    "getPromptDataset",
+    "updatePromptDataset",
+    "deletePromptDataset",
+    "clonePromptDataset",
+    "exportPromptDataset",
+    "listDatasetsForPrompt",
+    "createPromptDataset",
+    "listPromptDatasetItems",
+    "createPromptDatasetItem",
+    "bulkCreatePromptDatasetItems",
+    "getPromptDatasetItem",
+    "updatePromptDatasetItem",
+    "deletePromptDatasetItem",
+    "listEvaluations",
+    "createEvaluation",
+    "getEvaluationResultsAggregated",
+    "getEvaluation",
+    "updateEvaluation",
+    "deleteEvaluation",
+    "runEvaluation",
+    "getEvaluationResult",
+    "getEvaluationHistory",
+    "createEvaluationVersion",
+    "listExperiments",
+    "createExperiment",
+    "getExperiment",
+    "completeExperiment",
+    "deleteExperiment",
+    "addExperimentResult",
+    "updateExperimentName",
+    "executeExperiment",
+    "getPromptReadiness",
+    "listPlaygroundRuns",
+    "getPlaygroundRun",
+    "deletePlaygroundRun",
+    "listPromptExecutions",
+    "createPromptExecution",
+    "exportPromptExecutions",
+    "getPromptExecution",
+    "deletePromptExecution",
+    "createBatchPromptExecution",
+    "playgroundCall",
+    "playgroundEval",
+    "listPrompts",
+    "createPrompt",
+    "listPromptTags",
+    "getPrompt",
+    "updatePrompt",
+    "deletePrompt",
+    "listPromptVersions",
+    "createPromptVersion",
+    "getPromptAnalytics",
+    "getPromptModelComparisons",
+    "optimizePrompt",
+    "getOptimization",
+    "listOptimizations",
+    "getOptimizationProgress",
+    "getOptimizationStates",
+    "getOptimizationResults",
+    "pauseOptimization",
+    "resumeOptimization",
+    "cancelOptimization",
+    "listAgents",
+    "createAgent",
+    "getAgent",
+    "updateAgent",
+    "deleteAgent",
+    "getAgentBySlug",
+    "listOrganizations",
+    "createOrganization",
+    "getOrganization",
+    "updateOrganization",
+    "deleteOrganization",
+    "getOrganizationBySlug",
+    "checkOrganizationSlug",
+    "getOrganizationMemberCount",
+    "listWorkspaces",
+    "createWorkspace",
+    "getWorkspace",
+    "updateWorkspace",
+    "deleteWorkspace",
+    "setDefaultWorkspace",
+    "listOrganizationMembers",
+    "addOrganizationMember",
+    "updateOrganizationMember",
+    "removeOrganizationMember",
+    "listWorkspaceMembers",
+    "addWorkspaceMember",
+    "updateWorkspaceMember",
+    "removeWorkspaceMember",
+    "listProviders",
+    "createProvider",
+    "listAvailableModels",
+    "getProvider",
+    "updateProvider",
+    "deleteProvider",
+    "getModelsCatalog",
+    "testProvider",
+    "listInvitations",
+    "createInvitation",
+    "getInvitation",
+    "deleteInvitation",
+    "resendInvitation",
+    "listTraces",
+    "ingestTrace",
+    "ingestTraceBatch",
+    "ingestOtlp",
+    "getTraceStats",
+    "getAnalyticsSummary",
+    "getTrace",
+    "deleteTrace",]

mutagent/_funcs/add_experiment_result.py

@@ -0,0 +1,42 @@
+"""
+Standalone function: addExperimentResult.
+
+This file is GENERATED by mutagent-xgen. Do not edit manually.
+"""
+from __future__ import annotations
+
+from typing import Any, AsyncIterator, Optional
+
+
+def add_experiment_result(client: Any, id_: str, body: DatasetItemIdDatasetItemIndexModelColumnId) -> Dict[str, Any]:
+    """Add experiment result"""
+    path_params = {
+            "id": id_,
+    }
+    query_params = {}
+    response = client._http.request(
+            "POST",
+            "/api/prompts/experiments/{id}/results",
+            path_params=path_params or None,
+            query_params=query_params or None,
+            json=body.model_dump(exclude_none=True) if hasattr(body, "model_dump") else body,
+    )
+    response.raise_for_status()
+    return response.json()
+
+
+async def a_add_experiment_result(client: Any, id_: str, body: DatasetItemIdDatasetItemIndexModelColumnId) -> Dict[str, Any]:
+    """Async variant of add_experiment_result. Add experiment result"""
+    path_params = {
+            "id": id_,
+    }
+    query_params = {}
+    response = await client._async_http.request(
+            "POST",
+            "/api/prompts/experiments/{id}/results",
+            path_params=path_params or None,
+            query_params=query_params or None,
+            json=body.model_dump(exclude_none=True) if hasattr(body, "model_dump") else body,
+    )
+    response.raise_for_status()
+    return response.json()

mutagent/_funcs/add_organization_member.py

@@ -0,0 +1,42 @@
+"""
+Standalone function: addOrganizationMember.
+
+This file is GENERATED by mutagent-xgen. Do not edit manually.
+"""
+from __future__ import annotations
+
+from typing import Any, AsyncIterator, Optional
+
+
+def add_organization_member(client: Any, org_id: str, body: UserIdRoleStatus) -> IdUserIdRole:
+    """Add organization member"""
+    path_params = {
+            "orgId": org_id,
+    }
+    query_params = {}
+    response = client._http.request(
+            "POST",
+            "/api/organizations/{orgId}/members/",
+            path_params=path_params or None,
+            query_params=query_params or None,
+            json=body.model_dump(exclude_none=True) if hasattr(body, "model_dump") else body,
+    )
+    response.raise_for_status()
+    return response.json()
+
+
+async def a_add_organization_member(client: Any, org_id: str, body: UserIdRoleStatus) -> IdUserIdRole:
+    """Async variant of add_organization_member. Add organization member"""
+    path_params = {
+            "orgId": org_id,
+    }
+    query_params = {}
+    response = await client._async_http.request(
+            "POST",
+            "/api/organizations/{orgId}/members/",
+            path_params=path_params or None,
+            query_params=query_params or None,
+            json=body.model_dump(exclude_none=True) if hasattr(body, "model_dump") else body,
+    )
+    response.raise_for_status()
+    return response.json()

mutagent/_funcs/add_workspace_member.py

@@ -0,0 +1,42 @@
+"""
+Standalone function: addWorkspaceMember.
+
+This file is GENERATED by mutagent-xgen. Do not edit manually.
+"""
+from __future__ import annotations
+
+from typing import Any, AsyncIterator, Optional
+
+
+def add_workspace_member(client: Any, ws_id: str, body: UserIdRole) -> IdUserIdRole2:
+    """Add workspace member"""
+    path_params = {
+            "wsId": ws_id,
+    }
+    query_params = {}
+    response = client._http.request(
+            "POST",
+            "/api/workspaces/{wsId}/members/",
+            path_params=path_params or None,
+            query_params=query_params or None,
+            json=body.model_dump(exclude_none=True) if hasattr(body, "model_dump") else body,
+    )
+    response.raise_for_status()
+    return response.json()
+
+
+async def a_add_workspace_member(client: Any, ws_id: str, body: UserIdRole) -> IdUserIdRole2:
+    """Async variant of add_workspace_member. Add workspace member"""
+    path_params = {
+            "wsId": ws_id,
+    }
+    query_params = {}
+    response = await client._async_http.request(
+            "POST",
+            "/api/workspaces/{wsId}/members/",
+            path_params=path_params or None,
+            query_params=query_params or None,
+            json=body.model_dump(exclude_none=True) if hasattr(body, "model_dump") else body,
+    )
+    response.raise_for_status()
+    return response.json()

mutagent/_funcs/bulk_create_prompt_dataset_items.py

@@ -0,0 +1,42 @@
+"""
+Standalone function: bulkCreatePromptDatasetItems.
+
+This file is GENERATED by mutagent-xgen. Do not edit manually.
+"""
+from __future__ import annotations
+
+from typing import Any, AsyncIterator, Optional
+
+
+def bulk_create_prompt_dataset_items(client: Any, id_: Schemaminimu2, body: Dict[str, Any]) -> List[Dict[str, Any]]:
+    """Bulk add items"""
+    path_params = {
+            "id": id_,
+    }
+    query_params = {}
+    response = client._http.request(
+            "POST",
+            "/api/prompts/datasets/{id}/items/bulk",
+            path_params=path_params or None,
+            query_params=query_params or None,
+            json=body.model_dump(exclude_none=True) if hasattr(body, "model_dump") else body,
+    )
+    response.raise_for_status()
+    return response.json()
+
+
+async def a_bulk_create_prompt_dataset_items(client: Any, id_: Schemaminimu2, body: Dict[str, Any]) -> List[Dict[str, Any]]:
+    """Async variant of bulk_create_prompt_dataset_items. Bulk add items"""
+    path_params = {
+            "id": id_,
+    }
+    query_params = {}
+    response = await client._async_http.request(
+            "POST",
+            "/api/prompts/datasets/{id}/items/bulk",
+            path_params=path_params or None,
+            query_params=query_params or None,
+            json=body.model_dump(exclude_none=True) if hasattr(body, "model_dump") else body,
+    )
+    response.raise_for_status()
+    return response.json()

mutagent/_funcs/cancel_optimization.py

@@ -0,0 +1,40 @@
+"""
+Standalone function: cancelOptimization.
+
+This file is GENERATED by mutagent-xgen. Do not edit manually.
+"""
+from __future__ import annotations
+
+from typing import Any, AsyncIterator, Optional
+
+
+def cancel_optimization(client: Any, id_: SchemamaxLen4) -> OptimizationJob:
+    """Cancel job"""
+    path_params = {
+            "id": id_,
+    }
+    query_params = {}
+    response = client._http.request(
+            "POST",
+            "/api/optimization/{id}/cancel",
+            path_params=path_params or None,
+            query_params=query_params or None,
+    )
+    response.raise_for_status()
+    return response.json()
+
+
+async def a_cancel_optimization(client: Any, id_: SchemamaxLen4) -> OptimizationJob:
+    """Async variant of cancel_optimization. Cancel job"""
+    path_params = {
+            "id": id_,
+    }
+    query_params = {}
+    response = await client._async_http.request(
+            "POST",
+            "/api/optimization/{id}/cancel",
+            path_params=path_params or None,
+            query_params=query_params or None,
+    )
+    response.raise_for_status()
+    return response.json()

mutagent/_funcs/change_password.py

@@ -0,0 +1,38 @@
+"""
+Standalone function: changePassword.
+
+This file is GENERATED by mutagent-xgen. Do not edit manually.
+"""
+from __future__ import annotations
+
+from typing import Any, AsyncIterator, Optional
+
+
+def change_password(client: Any, body: CurrentPasswordNewPassword) -> Dict[str, Any]:
+    """Change password"""
+    path_params = {}
+    query_params = {}
+    response = client._http.request(
+            "POST",
+            "/api/users/me/password",
+            path_params=path_params or None,
+            query_params=query_params or None,
+            json=body.model_dump(exclude_none=True) if hasattr(body, "model_dump") else body,
+    )
+    response.raise_for_status()
+    return response.json()
+
+
+async def a_change_password(client: Any, body: CurrentPasswordNewPassword) -> Dict[str, Any]:
+    """Async variant of change_password. Change password"""
+    path_params = {}
+    query_params = {}
+    response = await client._async_http.request(
+            "POST",
+            "/api/users/me/password",
+            path_params=path_params or None,
+            query_params=query_params or None,
+            json=body.model_dump(exclude_none=True) if hasattr(body, "model_dump") else body,
+    )
+    response.raise_for_status()
+    return response.json()

mutagent/_funcs/check_organization_slug.py

@@ -0,0 +1,40 @@
+"""
+Standalone function: checkOrganizationSlug.
+
+This file is GENERATED by mutagent-xgen. Do not edit manually.
+"""
+from __future__ import annotations
+
+from typing import Any, AsyncIterator, Optional
+
+
+def check_organization_slug(client: Any, slug: str) -> Dict[str, Any]:
+    """Check slug availability"""
+    path_params = {
+            "slug": slug,
+    }
+    query_params = {}
+    response = client._http.request(
+            "GET",
+            "/api/organizations/check-slug/{slug}",
+            path_params=path_params or None,
+            query_params=query_params or None,
+    )
+    response.raise_for_status()
+    return response.json()
+
+
+async def a_check_organization_slug(client: Any, slug: str) -> Dict[str, Any]:
+    """Async variant of check_organization_slug. Check slug availability"""
+    path_params = {
+            "slug": slug,
+    }
+    query_params = {}
+    response = await client._async_http.request(
+            "GET",
+            "/api/organizations/check-slug/{slug}",
+            path_params=path_params or None,
+            query_params=query_params or None,
+    )
+    response.raise_for_status()
+    return response.json()

mutagent/_funcs/clone_prompt_dataset.py

@@ -0,0 +1,42 @@
+"""
+Standalone function: clonePromptDataset.
+
+This file is GENERATED by mutagent-xgen. Do not edit manually.
+"""
+from __future__ import annotations
+
+from typing import Any, AsyncIterator, Optional
+
+
+def clone_prompt_dataset(client: Any, id_: Schemaminimu2, body: TargetPromptIdTargetPromptGroupIdNewName) -> PromptDataset:
+    """Clone dataset"""
+    path_params = {
+            "id": id_,
+    }
+    query_params = {}
+    response = client._http.request(
+            "POST",
+            "/api/prompts/datasets/{id}/clone",
+            path_params=path_params or None,
+            query_params=query_params or None,
+            json=body.model_dump(exclude_none=True) if hasattr(body, "model_dump") else body,
+    )
+    response.raise_for_status()
+    return response.json()
+
+
+async def a_clone_prompt_dataset(client: Any, id_: Schemaminimu2, body: TargetPromptIdTargetPromptGroupIdNewName) -> PromptDataset:
+    """Async variant of clone_prompt_dataset. Clone dataset"""
+    path_params = {
+            "id": id_,
+    }
+    query_params = {}
+    response = await client._async_http.request(
+            "POST",
+            "/api/prompts/datasets/{id}/clone",
+            path_params=path_params or None,
+            query_params=query_params or None,
+            json=body.model_dump(exclude_none=True) if hasattr(body, "model_dump") else body,
+    )
+    response.raise_for_status()
+    return response.json()