@databricks/sdk-supervisoragents 0.0.0-dev → 0.1.0-dev.1

package/dist/v1/model.d.ts

@@ -0,0 +1,295 @@
+import { Temporal } from '@js-temporal/polyfill';
+import { FieldMask } from '@databricks/sdk-core/wkt';
+import { z } from 'zod';
+/** <Databricks> app. Supported app: custom mcp, custom agent. */
+export interface App {
+    /** App name */
+    name?: string | undefined;
+}
+/** Create an example. */
+export interface CreateExampleRequest {
+    /**
+     * Parent resource where this example will be created.
+     * Format: supervisor-agents/{supervisor_agent_id}
+     */
+    parent?: string | undefined;
+    /** The example to create under the parent Supervisor Agent. */
+    example?: Example | undefined;
+}
+export interface CreateSupervisorAgentRequest {
+    /** The Supervisor Agent to create. */
+    supervisorAgent?: SupervisorAgent | undefined;
+}
+export interface CreateToolRequest {
+    /**
+     * Parent resource where this tool will be created.
+     * Format: supervisor-agents/{supervisor_agent_id}
+     */
+    parent?: string | undefined;
+    tool?: Tool | undefined;
+    /**
+     * The ID to use for the tool, which will become the final component of
+     * the tool's resource name.
+     */
+    toolId?: string | undefined;
+}
+/** Delete an example. */
+export interface DeleteExampleRequest {
+    /**
+     * The resource name of the example to delete.
+     * Format: supervisor-agents/{supervisor_agent_id}/examples/{example_id}
+     */
+    name?: string | undefined;
+}
+export interface DeleteSupervisorAgentRequest {
+    /**
+     * The resource name of the Supervisor Agent.
+     * Format: supervisor-agents/{supervisor_agent_id}
+     */
+    name?: string | undefined;
+}
+export interface DeleteToolRequest {
+    /**
+     * The resource name of the Tool.
+     * Format: supervisor-agents/{supervisor_agent_id}/tools/{tool_id}
+     */
+    name?: string | undefined;
+}
+/**
+ * An example associated with a Supervisor Agent.
+ * Contains a question and guidelines for how the agent should respond.
+ */
+export interface Example {
+    /**
+     * Full resource name:
+     * supervisor-agents/{supervisor_agent_id}/examples/{example_id}
+     */
+    name?: string | undefined;
+    /** The example question. */
+    question?: string | undefined;
+    /** Guidelines for answering the question. */
+    guidelines?: string[] | undefined;
+    /** The universally unique identifier (UUID) of the example. */
+    exampleId?: string | undefined;
+}
+export interface GenieSpace {
+    /**
+     * Deprecated: use space_id instead. Still REQUIRED for backward compatibility
+     * until a future API version removes it.
+     */
+    id?: string | undefined;
+}
+/** Get an example. */
+export interface GetExampleRequest {
+    /**
+     * The resource name of the example.
+     * Format: supervisor-agents/{supervisor_agent_id}/examples/{example_id}
+     */
+    name?: string | undefined;
+}
+export interface GetSupervisorAgentRequest {
+    /**
+     * The resource name of the Supervisor Agent.
+     * Format: supervisor-agents/{supervisor_agent_id}
+     */
+    name?: string | undefined;
+}
+export interface GetToolRequest {
+    /**
+     * The resource name of the Tool.
+     * Format: supervisor-agents/{supervisor_agent_id}/tools/{tool_id}
+     */
+    name?: string | undefined;
+}
+export interface KnowledgeAssistant {
+    /** Deprecated: use knowledge_assistant_id instead. */
+    servingEndpointName?: string | undefined;
+    /** The ID of the knowledge assistant. */
+    knowledgeAssistantId?: string | undefined;
+}
+/** List examples. */
+export interface ListExamplesRequest {
+    /**
+     * Parent resource to list from.
+     * Format: supervisor-agents/{supervisor_agent_id}
+     */
+    parent?: string | undefined;
+    /**
+     * The maximum number of examples to return.
+     * If unspecified, at most 100 examples will be returned.
+     * The maximum value is 100; values above 100 will be coerced to 100.
+     */
+    pageSize?: number | undefined;
+    /**
+     * A page token, received from a previous `ListExamples` call.
+     * Provide this to retrieve the subsequent page.
+     * If unspecified, the first page will be returned.
+     */
+    pageToken?: string | undefined;
+}
+/** A list of Supervisor Agent examples. */
+export interface ListExamplesResponse {
+    examples?: Example[] | undefined;
+    nextPageToken?: string | undefined;
+}
+export interface ListSupervisorAgentsRequest {
+    /**
+     * The maximum number of supervisor agents to return.
+     * If unspecified, at most 100 supervisor agents will be returned.
+     * The maximum value is 100; values above 100 will be coerced to 100.
+     */
+    pageSize?: number | undefined;
+    /**
+     * A page token, received from a previous `ListSupervisorAgents` call.
+     * Provide this to retrieve the subsequent page.
+     * If unspecified, the first page will be returned.
+     */
+    pageToken?: string | undefined;
+}
+export interface ListSupervisorAgentsResponse {
+    supervisorAgents?: SupervisorAgent[] | undefined;
+    /**
+     * A token that can be sent as `page_token` to retrieve the next page.
+     * If this field is omitted, there are no subsequent pages.
+     */
+    nextPageToken?: string | undefined;
+}
+export interface ListToolsRequest {
+    /**
+     * Parent resource to list from.
+     * Format: supervisor-agents/{supervisor_agent_id}
+     */
+    parent?: string | undefined;
+    pageSize?: number | undefined;
+    pageToken?: string | undefined;
+}
+export interface ListToolsResponse {
+    tools?: Tool[] | undefined;
+    nextPageToken?: string | undefined;
+}
+export interface SupervisorAgent {
+    /**
+     * The resource name of the SupervisorAgent.
+     * Format: supervisor-agents/{supervisor_agent_id}
+     */
+    name?: string | undefined;
+    /** The display name of the Supervisor Agent, unique at workspace level. */
+    displayName?: string | undefined;
+    /** Description of what this agent can do (user-facing). */
+    description?: string | undefined;
+    /** Optional natural-language instructions for the supervisor agent. */
+    instructions?: string | undefined;
+    /** Deprecated: Use supervisor_agent_id instead. */
+    id?: string | undefined;
+    /** The universally unique identifier (UUID) of the Supervisor Agent. */
+    supervisorAgentId?: string | undefined;
+    /** The creator of the Supervisor Agent. */
+    creator?: string | undefined;
+    /** Creation timestamp. */
+    createTime?: Temporal.Instant | undefined;
+    /** The name of the supervisor agent's serving endpoint. */
+    endpointName?: string | undefined;
+    /** The MLflow experiment ID. */
+    experimentId?: string | undefined;
+}
+export interface Tool {
+    /**
+     * Full resource name:
+     * supervisor-agents/{supervisor_agent_id}/tools/{tool_id}
+     */
+    name?: string | undefined;
+    /** Deprecated: Use tool_id instead. */
+    id?: string | undefined;
+    /** Tool type. Must be one of: "genie_space", "knowledge_assistant", "uc_function", "uc_connection", "uc_mcp", "app", "volume", "dashboard", "serving_endpoint", "table", "vector_search_index", "catalog", "schema", "supervisor_agent", "web_search", "skill". The legacy values "lakeview_dashboard" and "uc_table" are also accepted and remain equivalent to "dashboard" and "table" respectively. */
+    toolType?: string | undefined;
+    /** Specification for the tool type. */
+    spec?: {
+        $case: 'genieSpace';
+        genieSpace: GenieSpace;
+    } | {
+        $case: 'knowledgeAssistant';
+        knowledgeAssistant: KnowledgeAssistant;
+    } | {
+        $case: 'ucFunction';
+        ucFunction: UcFunction;
+    } | {
+        $case: 'app';
+        app: App;
+    } | {
+        $case: 'volume';
+        volume: Volume;
+    } | {
+        $case: 'ucConnection';
+        ucConnection: UcConnection;
+    } | undefined;
+    /** Description of what this tool does (user-facing). */
+    description?: string | undefined;
+    /** User specified id of the Tool. */
+    toolId?: string | undefined;
+}
+/** Databricks UC connection. Supported connection: external mcp server. */
+export interface UcConnection {
+    name?: string | undefined;
+}
+export interface UcFunction {
+    /** Full uc function name */
+    name?: string | undefined;
+}
+/** Update an example. */
+export interface UpdateExampleRequest {
+    /**
+     * The resource name of the example to update.
+     * Format: supervisor-agents/{supervisor_agent_id}/examples/{example_id}
+     */
+    name?: string | undefined;
+    example?: Example | undefined;
+    /**
+     * Comma-delimited list of fields to update on the example.
+     * Allowed values: `question`, `guidelines`.
+     * Examples:
+     * - `question`
+     * - `question,guidelines`
+     */
+    updateMask?: FieldMask<Example> | undefined;
+}
+export interface UpdateSupervisorAgentRequest {
+    /** The SupervisorAgent to update. */
+    supervisorAgent?: SupervisorAgent | undefined;
+    /** Field mask for fields to be updated. */
+    updateMask?: FieldMask<SupervisorAgent> | undefined;
+}
+export interface UpdateToolRequest {
+    /** The Tool to update. */
+    tool?: Tool | undefined;
+    /** Field mask for fields to be updated. */
+    updateMask?: FieldMask<Tool> | undefined;
+}
+export interface Volume {
+    /** Full uc volume name */
+    name?: string | undefined;
+}
+export declare const unmarshalAppSchema: z.ZodType<App>;
+export declare const unmarshalExampleSchema: z.ZodType<Example>;
+export declare const unmarshalGenieSpaceSchema: z.ZodType<GenieSpace>;
+export declare const unmarshalKnowledgeAssistantSchema: z.ZodType<KnowledgeAssistant>;
+export declare const unmarshalListExamplesResponseSchema: z.ZodType<ListExamplesResponse>;
+export declare const unmarshalListSupervisorAgentsResponseSchema: z.ZodType<ListSupervisorAgentsResponse>;
+export declare const unmarshalListToolsResponseSchema: z.ZodType<ListToolsResponse>;
+export declare const unmarshalSupervisorAgentSchema: z.ZodType<SupervisorAgent>;
+export declare const unmarshalToolSchema: z.ZodType<Tool>;
+export declare const unmarshalUcConnectionSchema: z.ZodType<UcConnection>;
+export declare const unmarshalUcFunctionSchema: z.ZodType<UcFunction>;
+export declare const unmarshalVolumeSchema: z.ZodType<Volume>;
+export declare const marshalAppSchema: z.ZodType;
+export declare const marshalExampleSchema: z.ZodType;
+export declare const marshalGenieSpaceSchema: z.ZodType;
+export declare const marshalKnowledgeAssistantSchema: z.ZodType;
+export declare const marshalSupervisorAgentSchema: z.ZodType;
+export declare const marshalToolSchema: z.ZodType;
+export declare const marshalUcConnectionSchema: z.ZodType;
+export declare const marshalUcFunctionSchema: z.ZodType;
+export declare const marshalVolumeSchema: z.ZodType;
+export declare function exampleFieldMask(...paths: string[]): FieldMask<Example>;
+export declare function supervisorAgentFieldMask(...paths: string[]): FieldMask<SupervisorAgent>;
+export declare function toolFieldMask(...paths: string[]): FieldMask<Tool>;
+//# sourceMappingURL=model.d.ts.map

package/dist/v1/model.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"model.d.ts","sourceRoot":"","sources":["../../src/v1/model.ts"],"names":[],"mappings":"AAEA,OAAO,EAAC,QAAQ,EAAC,MAAM,uBAAuB,CAAC;AAC/C,OAAO,EAAC,SAAS,EAAC,MAAM,0BAA0B,CAAC;AAEnD,OAAO,EAAC,CAAC,EAAC,MAAM,KAAK,CAAC;AAEtB,iEAAiE;AACjE,MAAM,WAAW,GAAG;IAClB,eAAe;IACf,IAAI,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;CAC3B;AAED,yBAAyB;AACzB,MAAM,WAAW,oBAAoB;IACnC;;;OAGG;IACH,MAAM,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC5B,+DAA+D;IAC/D,OAAO,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;CAC/B;AAED,MAAM,WAAW,4BAA4B;IAC3C,sCAAsC;IACtC,eAAe,CAAC,EAAE,eAAe,GAAG,SAAS,CAAC;CAC/C;AAED,MAAM,WAAW,iBAAiB;IAChC;;;OAGG;IACH,MAAM,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC5B,IAAI,CAAC,EAAE,IAAI,GAAG,SAAS,CAAC;IACxB;;;OAGG;IACH,MAAM,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;CAC7B;AAED,yBAAyB;AACzB,MAAM,WAAW,oBAAoB;IACnC;;;OAGG;IACH,IAAI,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;CAC3B;AAED,MAAM,WAAW,4BAA4B;IAC3C;;;OAGG;IACH,IAAI,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;CAC3B;AAED,MAAM,WAAW,iBAAiB;IAChC;;;OAGG;IACH,IAAI,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;CAC3B;AAED;;;GAGG;AACH,MAAM,WAAW,OAAO;IACtB;;;OAGG;IACH,IAAI,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC1B,4BAA4B;IAC5B,QAAQ,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC9B,6CAA6C;IAC7C,UAAU,CAAC,EAAE,MAAM,EAAE,GAAG,SAAS,CAAC;IAClC,+DAA+D;IAC/D,SAAS,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;CAChC;AAED,MAAM,WAAW,UAAU;IACzB;;;OAGG;IACH,EAAE,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;CACzB;AAED,sBAAsB;AACtB,MAAM,WAAW,iBAAiB;IAChC;;;OAGG;IACH,IAAI,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;CAC3B;AAED,MAAM,WAAW,yBAAyB;IACxC;;;OAGG;IACH,IAAI,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;CAC3B;AAED,MAAM,WAAW,cAAc;IAC7B;;;OAGG;IACH,IAAI,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;CAC3B;AAED,MAAM,WAAW,kBAAkB;IACjC,sDAAsD;IACtD,mBAAmB,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IACzC,yCAAyC;IACzC,oBAAoB,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;CAC3C;AAED,qBAAqB;AACrB,MAAM,WAAW,mBAAmB;IAClC;;;OAGG;IACH,MAAM,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC5B;;;;OAIG;IACH,QAAQ,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC9B;;;;OAIG;IACH,SAAS,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;CAChC;AAED,2CAA2C;AAC3C,MAAM,WAAW,oBAAoB;IACnC,QAAQ,CAAC,EAAE,OAAO,EAAE,GAAG,SAAS,CAAC;IACjC,aAAa,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;CACpC;AAED,MAAM,WAAW,2BAA2B;IAC1C;;;;OAIG;IACH,QAAQ,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC9B;;;;OAIG;IACH,SAAS,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;CAChC;AAED,MAAM,WAAW,4BAA4B;IAC3C,gBAAgB,CAAC,EAAE,eAAe,EAAE,GAAG,SAAS,CAAC;IACjD;;;OAGG;IACH,aAAa,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;CACpC;AAED,MAAM,WAAW,gBAAgB;IAC/B;;;OAGG;IACH,MAAM,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC5B,QAAQ,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC9B,SAAS,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;CAChC;AAED,MAAM,WAAW,iBAAiB;IAChC,KAAK,CAAC,EAAE,IAAI,EAAE,GAAG,SAAS,CAAC;IAC3B,aAAa,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;CACpC;AAED,MAAM,WAAW,eAAe;IAC9B;;;OAGG;IACH,IAAI,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC1B,2EAA2E;IAC3E,WAAW,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IACjC,2DAA2D;IAC3D,WAAW,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IACjC,uEAAuE;IACvE,YAAY,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAClC,mDAAmD;IACnD,EAAE,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IACxB,wEAAwE;IACxE,iBAAiB,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IACvC,2CAA2C;IAC3C,OAAO,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC7B,0BAA0B;IAC1B,UAAU,CAAC,EAAE,QAAQ,CAAC,OAAO,GAAG,SAAS,CAAC;IAC1C,2DAA2D;IAC3D,YAAY,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAClC,gCAAgC;IAChC,YAAY,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;CACnC;AAED,MAAM,WAAW,IAAI;IACnB;;;OAGG;IACH,IAAI,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC1B,uCAAuC;IACvC,EAAE,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IACxB,0YAA0Y;IAC1Y,QAAQ,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC9B,uCAAuC;IACvC,IAAI,CAAC,EACD;QAAC,KAAK,EAAE,YAAY,CAAC;QAAC,UAAU,EAAE,UAAU,CAAA;KAAC,GAC7C;QAAC,KAAK,EAAE,oBAAoB,CAAC;QAAC,kBAAkB,EAAE,kBAAkB,CAAA;KAAC,GACrE;QAAC,KAAK,EAAE,YAAY,CAAC;QAAC,UAAU,EAAE,UAAU,CAAA;KAAC,GAC7C;QAAC,KAAK,EAAE,KAAK,CAAC;QAAC,GAAG,EAAE,GAAG,CAAA;KAAC,GACxB;QAAC,KAAK,EAAE,QAAQ,CAAC;QAAC,MAAM,EAAE,MAAM,CAAA;KAAC,GACjC;QAAC,KAAK,EAAE,cAAc,CAAC;QAAC,YAAY,EAAE,YAAY,CAAA;KAAC,GACnD,SAAS,CAAC;IACd,wDAAwD;IACxD,WAAW,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IACjC,qCAAqC;IACrC,MAAM,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;CAC7B;AAED,2EAA2E;AAC3E,MAAM,WAAW,YAAY;IAC3B,IAAI,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;CAC3B;AAED,MAAM,WAAW,UAAU;IACzB,4BAA4B;IAC5B,IAAI,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;CAC3B;AAED,yBAAyB;AACzB,MAAM,WAAW,oBAAoB;IACnC;;;OAGG;IACH,IAAI,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC1B,OAAO,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;IAC9B;;;;;;OAMG;IACH,UAAU,CAAC,EAAE,SAAS,CAAC,OAAO,CAAC,GAAG,SAAS,CAAC;CAC7C;AAED,MAAM,WAAW,4BAA4B;IAC3C,qCAAqC;IACrC,eAAe,CAAC,EAAE,eAAe,GAAG,SAAS,CAAC;IAC9C,2CAA2C;IAC3C,UAAU,CAAC,EAAE,SAAS,CAAC,eAAe,CAAC,GAAG,SAAS,CAAC;CACrD;AAED,MAAM,WAAW,iBAAiB;IAChC,0BAA0B;IAC1B,IAAI,CAAC,EAAE,IAAI,GAAG,SAAS,CAAC;IACxB,2CAA2C;IAC3C,UAAU,CAAC,EAAE,SAAS,CAAC,IAAI,CAAC,GAAG,SAAS,CAAC;CAC1C;AAED,MAAM,WAAW,MAAM;IACrB,0BAA0B;IAC1B,IAAI,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;CAC3B;AAED,eAAO,MAAM,kBAAkB,EAAE,CAAC,CAAC,OAAO,CAAC,GAAG,CAMzC,CAAC;AAEN,eAAO,MAAM,sBAAsB,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAYjD,CAAC;AAEN,eAAO,MAAM,yBAAyB,EAAE,CAAC,CAAC,OAAO,CAAC,UAAU,CAMvD,CAAC;AAEN,eAAO,MAAM,iCAAiC,EAAE,CAAC,CAAC,OAAO,CAAC,kBAAkB,CASrE,CAAC;AAER,eAAO,MAAM,mCAAmC,EAAE,CAAC,CAAC,OAAO,CAAC,oBAAoB,CASzE,CAAC;AAER,eAAO,MAAM,2CAA2C,EAAE,CAAC,CAAC,OAAO,CAAC,4BAA4B,CAWzF,CAAC;AAER,eAAO,MAAM,gCAAgC,EAAE,CAAC,CAAC,OAAO,CAAC,iBAAiB,CAQrE,CAAC;AAEN,eAAO,MAAM,8BAA8B,EAAE,CAAC,CAAC,OAAO,CAAC,eAAe,CA2BjE,CAAC;AAEN,eAAO,MAAM,mBAAmB,EAAE,CAAC,CAAC,OAAO,CAAC,IAAI,CA0C3C,CAAC;AAEN,eAAO,MAAM,2BAA2B,EAAE,CAAC,CAAC,OAAO,CAAC,YAAY,CAM3D,CAAC;AAEN,eAAO,MAAM,yBAAyB,EAAE,CAAC,CAAC,OAAO,CAAC,UAAU,CAMvD,CAAC;AAEN,eAAO,MAAM,qBAAqB,EAAE,CAAC,CAAC,OAAO,CAAC,MAAM,CAM/C,CAAC;AAEN,eAAO,MAAM,gBAAgB,EAAE,CAAC,CAAC,OAM5B,CAAC;AAEN,eAAO,MAAM,oBAAoB,EAAE,CAAC,CAAC,OAYhC,CAAC;AAEN,eAAO,MAAM,uBAAuB,EAAE,CAAC,CAAC,OAMnC,CAAC;AAEN,eAAO,MAAM,+BAA+B,EAAE,CAAC,CAAC,OAQ3C,CAAC;AAEN,eAAO,MAAM,4BAA4B,EAAE,CAAC,CAAC,OA2BxC,CAAC;AAEN,eAAO,MAAM,iBAAiB,EAAE,CAAC,CAAC,OAoD7B,CAAC;AAEN,eAAO,MAAM,yBAAyB,EAAE,CAAC,CAAC,OAMrC,CAAC;AAEN,eAAO,MAAM,uBAAuB,EAAE,CAAC,CAAC,OAMnC,CAAC;AAEN,eAAO,MAAM,mBAAmB,EAAE,CAAC,CAAC,OAM/B,CAAC;AAaN,wBAAgB,gBAAgB,CAAC,GAAG,KAAK,EAAE,MAAM,EAAE,GAAG,SAAS,CAAC,OAAO,CAAC,CAEvE;AAwBD,wBAAgB,wBAAwB,CACtC,GAAG,KAAK,EAAE,MAAM,EAAE,GACjB,SAAS,CAAC,eAAe,CAAC,CAK5B;AAsBD,wBAAgB,aAAa,CAAC,GAAG,KAAK,EAAE,MAAM,EAAE,GAAG,SAAS,CAAC,IAAI,CAAC,CAEjE"}

package/dist/v1/model.js

@@ -0,0 +1,364 @@
+// Code generated from API definition by Databricks SDK Generator. DO NOT EDIT.
+import { Temporal } from '@js-temporal/polyfill';
+import { FieldMask } from '@databricks/sdk-core/wkt';
+import { z } from 'zod';
+export const unmarshalAppSchema = z
+    .object({
+    name: z.string().optional(),
+})
+    .transform(d => ({
+    name: d.name,
+}));
+export const unmarshalExampleSchema = z
+    .object({
+    name: z.string().optional(),
+    question: z.string().optional(),
+    guidelines: z.array(z.string()).optional(),
+    example_id: z.string().optional(),
+})
+    .transform(d => ({
+    name: d.name,
+    question: d.question,
+    guidelines: d.guidelines,
+    exampleId: d.example_id,
+}));
+export const unmarshalGenieSpaceSchema = z
+    .object({
+    id: z.string().optional(),
+})
+    .transform(d => ({
+    id: d.id,
+}));
+export const unmarshalKnowledgeAssistantSchema = z
+    .object({
+    serving_endpoint_name: z.string().optional(),
+    knowledge_assistant_id: z.string().optional(),
+})
+    .transform(d => ({
+    servingEndpointName: d.serving_endpoint_name,
+    knowledgeAssistantId: d.knowledge_assistant_id,
+}));
+export const unmarshalListExamplesResponseSchema = z
+    .object({
+    examples: z.array(z.lazy(() => unmarshalExampleSchema)).optional(),
+    next_page_token: z.string().optional(),
+})
+    .transform(d => ({
+    examples: d.examples,
+    nextPageToken: d.next_page_token,
+}));
+export const unmarshalListSupervisorAgentsResponseSchema = z
+    .object({
+    supervisor_agents: z
+        .array(z.lazy(() => unmarshalSupervisorAgentSchema))
+        .optional(),
+    next_page_token: z.string().optional(),
+})
+    .transform(d => ({
+    supervisorAgents: d.supervisor_agents,
+    nextPageToken: d.next_page_token,
+}));
+export const unmarshalListToolsResponseSchema = z
+    .object({
+    tools: z.array(z.lazy(() => unmarshalToolSchema)).optional(),
+    next_page_token: z.string().optional(),
+})
+    .transform(d => ({
+    tools: d.tools,
+    nextPageToken: d.next_page_token,
+}));
+export const unmarshalSupervisorAgentSchema = z
+    .object({
+    name: z.string().optional(),
+    display_name: z.string().optional(),
+    description: z.string().optional(),
+    instructions: z.string().optional(),
+    id: z.string().optional(),
+    supervisor_agent_id: z.string().optional(),
+    creator: z.string().optional(),
+    create_time: z
+        .string()
+        .transform(s => Temporal.Instant.from(s))
+        .optional(),
+    endpoint_name: z.string().optional(),
+    experiment_id: z.string().optional(),
+})
+    .transform(d => ({
+    name: d.name,
+    displayName: d.display_name,
+    description: d.description,
+    instructions: d.instructions,
+    id: d.id,
+    supervisorAgentId: d.supervisor_agent_id,
+    creator: d.creator,
+    createTime: d.create_time,
+    endpointName: d.endpoint_name,
+    experimentId: d.experiment_id,
+}));
+export const unmarshalToolSchema = z
+    .object({
+    name: z.string().optional(),
+    id: z.string().optional(),
+    tool_type: z.string().optional(),
+    genie_space: z.lazy(() => unmarshalGenieSpaceSchema).optional(),
+    knowledge_assistant: z
+        .lazy(() => unmarshalKnowledgeAssistantSchema)
+        .optional(),
+    uc_function: z.lazy(() => unmarshalUcFunctionSchema).optional(),
+    app: z.lazy(() => unmarshalAppSchema).optional(),
+    volume: z.lazy(() => unmarshalVolumeSchema).optional(),
+    uc_connection: z.lazy(() => unmarshalUcConnectionSchema).optional(),
+    description: z.string().optional(),
+    tool_id: z.string().optional(),
+})
+    .transform(d => ({
+    name: d.name,
+    id: d.id,
+    toolType: d.tool_type,
+    spec: d.genie_space !== undefined
+        ? { $case: 'genieSpace', genieSpace: d.genie_space }
+        : d.knowledge_assistant !== undefined
+            ? {
+                $case: 'knowledgeAssistant',
+                knowledgeAssistant: d.knowledge_assistant,
+            }
+            : d.uc_function !== undefined
+                ? { $case: 'ucFunction', ucFunction: d.uc_function }
+                : d.app !== undefined
+                    ? { $case: 'app', app: d.app }
+                    : d.volume !== undefined
+                        ? { $case: 'volume', volume: d.volume }
+                        : d.uc_connection !== undefined
+                            ? {
+                                $case: 'ucConnection',
+                                ucConnection: d.uc_connection,
+                            }
+                            : undefined,
+    description: d.description,
+    toolId: d.tool_id,
+}));
+export const unmarshalUcConnectionSchema = z
+    .object({
+    name: z.string().optional(),
+})
+    .transform(d => ({
+    name: d.name,
+}));
+export const unmarshalUcFunctionSchema = z
+    .object({
+    name: z.string().optional(),
+})
+    .transform(d => ({
+    name: d.name,
+}));
+export const unmarshalVolumeSchema = z
+    .object({
+    name: z.string().optional(),
+})
+    .transform(d => ({
+    name: d.name,
+}));
+export const marshalAppSchema = z
+    .object({
+    name: z.string().optional(),
+})
+    .transform(d => ({
+    name: d.name,
+}));
+export const marshalExampleSchema = z
+    .object({
+    name: z.string().optional(),
+    question: z.string().optional(),
+    guidelines: z.array(z.string()).optional(),
+    exampleId: z.string().optional(),
+})
+    .transform(d => ({
+    name: d.name,
+    question: d.question,
+    guidelines: d.guidelines,
+    example_id: d.exampleId,
+}));
+export const marshalGenieSpaceSchema = z
+    .object({
+    id: z.string().optional(),
+})
+    .transform(d => ({
+    id: d.id,
+}));
+export const marshalKnowledgeAssistantSchema = z
+    .object({
+    servingEndpointName: z.string().optional(),
+    knowledgeAssistantId: z.string().optional(),
+})
+    .transform(d => ({
+    serving_endpoint_name: d.servingEndpointName,
+    knowledge_assistant_id: d.knowledgeAssistantId,
+}));
+export const marshalSupervisorAgentSchema = z
+    .object({
+    name: z.string().optional(),
+    displayName: z.string().optional(),
+    description: z.string().optional(),
+    instructions: z.string().optional(),
+    id: z.string().optional(),
+    supervisorAgentId: z.string().optional(),
+    creator: z.string().optional(),
+    createTime: z
+        .any()
+        .transform((d) => d.toString())
+        .optional(),
+    endpointName: z.string().optional(),
+    experimentId: z.string().optional(),
+})
+    .transform(d => ({
+    name: d.name,
+    display_name: d.displayName,
+    description: d.description,
+    instructions: d.instructions,
+    id: d.id,
+    supervisor_agent_id: d.supervisorAgentId,
+    creator: d.creator,
+    create_time: d.createTime,
+    endpoint_name: d.endpointName,
+    experiment_id: d.experimentId,
+}));
+export const marshalToolSchema = z
+    .object({
+    name: z.string().optional(),
+    id: z.string().optional(),
+    toolType: z.string().optional(),
+    spec: z
+        .discriminatedUnion('$case', [
+        z.object({
+            $case: z.literal('genieSpace'),
+            genieSpace: z.lazy(() => marshalGenieSpaceSchema),
+        }),
+        z.object({
+            $case: z.literal('knowledgeAssistant'),
+            knowledgeAssistant: z.lazy(() => marshalKnowledgeAssistantSchema),
+        }),
+        z.object({
+            $case: z.literal('ucFunction'),
+            ucFunction: z.lazy(() => marshalUcFunctionSchema),
+        }),
+        z.object({
+            $case: z.literal('app'),
+            app: z.lazy(() => marshalAppSchema),
+        }),
+        z.object({
+            $case: z.literal('volume'),
+            volume: z.lazy(() => marshalVolumeSchema),
+        }),
+        z.object({
+            $case: z.literal('ucConnection'),
+            ucConnection: z.lazy(() => marshalUcConnectionSchema),
+        }),
+    ])
+        .optional(),
+    description: z.string().optional(),
+    toolId: z.string().optional(),
+})
+    .transform(d => ({
+    name: d.name,
+    id: d.id,
+    tool_type: d.toolType,
+    ...(d.spec?.$case === 'genieSpace' && { genie_space: d.spec.genieSpace }),
+    ...(d.spec?.$case === 'knowledgeAssistant' && {
+        knowledge_assistant: d.spec.knowledgeAssistant,
+    }),
+    ...(d.spec?.$case === 'ucFunction' && { uc_function: d.spec.ucFunction }),
+    ...(d.spec?.$case === 'app' && { app: d.spec.app }),
+    ...(d.spec?.$case === 'volume' && { volume: d.spec.volume }),
+    ...(d.spec?.$case === 'ucConnection' && {
+        uc_connection: d.spec.ucConnection,
+    }),
+    description: d.description,
+    tool_id: d.toolId,
+}));
+export const marshalUcConnectionSchema = z
+    .object({
+    name: z.string().optional(),
+})
+    .transform(d => ({
+    name: d.name,
+}));
+export const marshalUcFunctionSchema = z
+    .object({
+    name: z.string().optional(),
+})
+    .transform(d => ({
+    name: d.name,
+}));
+export const marshalVolumeSchema = z
+    .object({
+    name: z.string().optional(),
+})
+    .transform(d => ({
+    name: d.name,
+}));
+const appFieldMaskSchema = {
+    name: { wire: 'name' },
+};
+const exampleFieldMaskSchema = {
+    exampleId: { wire: 'example_id' },
+    guidelines: { wire: 'guidelines' },
+    name: { wire: 'name' },
+    question: { wire: 'question' },
+};
+export function exampleFieldMask(...paths) {
+    return FieldMask.build(paths, exampleFieldMaskSchema);
+}
+const genieSpaceFieldMaskSchema = {
+    id: { wire: 'id' },
+};
+const knowledgeAssistantFieldMaskSchema = {
+    knowledgeAssistantId: { wire: 'knowledge_assistant_id' },
+    servingEndpointName: { wire: 'serving_endpoint_name' },
+};
+const supervisorAgentFieldMaskSchema = {
+    createTime: { wire: 'create_time' },
+    creator: { wire: 'creator' },
+    description: { wire: 'description' },
+    displayName: { wire: 'display_name' },
+    endpointName: { wire: 'endpoint_name' },
+    experimentId: { wire: 'experiment_id' },
+    id: { wire: 'id' },
+    instructions: { wire: 'instructions' },
+    name: { wire: 'name' },
+    supervisorAgentId: { wire: 'supervisor_agent_id' },
+};
+export function supervisorAgentFieldMask(...paths) {
+    return FieldMask.build(paths, supervisorAgentFieldMaskSchema);
+}
+const toolFieldMaskSchema = {
+    app: { wire: 'app', children: () => appFieldMaskSchema },
+    description: { wire: 'description' },
+    genieSpace: { wire: 'genie_space', children: () => genieSpaceFieldMaskSchema },
+    id: { wire: 'id' },
+    knowledgeAssistant: {
+        wire: 'knowledge_assistant',
+        children: () => knowledgeAssistantFieldMaskSchema,
+    },
+    name: { wire: 'name' },
+    toolId: { wire: 'tool_id' },
+    toolType: { wire: 'tool_type' },
+    ucConnection: {
+        wire: 'uc_connection',
+        children: () => ucConnectionFieldMaskSchema,
+    },
+    ucFunction: { wire: 'uc_function', children: () => ucFunctionFieldMaskSchema },
+    volume: { wire: 'volume', children: () => volumeFieldMaskSchema },
+};
+export function toolFieldMask(...paths) {
+    return FieldMask.build(paths, toolFieldMaskSchema);
+}
+const ucConnectionFieldMaskSchema = {
+    name: { wire: 'name' },
+};
+const ucFunctionFieldMaskSchema = {
+    name: { wire: 'name' },
+};
+const volumeFieldMaskSchema = {
+    name: { wire: 'name' },
+};
+//# sourceMappingURL=model.js.map