@databricks/ai-sdk-provider 0.2.2 → 0.3.0

package/dist/index.d.cts

@@ -1,6 +1,5 @@
-import { LanguageModelV2, ProviderV2 } from "@ai-sdk/provider";
+import { LanguageModelV3, ProviderV3 } from "@ai-sdk/provider";
 import { FetchFunction } from "@ai-sdk/provider-utils";
-import { z } from "zod/v4";
 
 //#region src/databricks-provider.d.ts
 type DatabricksLanguageModelConfig = {
@@ -10,13 +9,22 @@ type DatabricksLanguageModelConfig = {
     path: string;
   }) => string;
   fetch?: FetchFunction;
+  /**
+   * When true, marks all tool calls as dynamic and provider-executed.
+   * This indicates that tool execution is handled remotely by Databricks agents.
+   * Defaults to true.
+   */
+  useRemoteToolCalling?: boolean;
 };
-interface DatabricksProvider extends ProviderV2 {
+interface DatabricksProvider extends ProviderV3 {
   /** Agents */
-  chatAgent(modelId: string): LanguageModelV2;
-  responses(modelId: string): LanguageModelV2;
+  chatAgent(modelId: string): LanguageModelV3;
+  responses(modelId: string): LanguageModelV3;
   /** Foundation Models */
-  chatCompletions(modelId: string): LanguageModelV2;
+  chatCompletions(modelId: string): LanguageModelV3;
+  /** Required overrides from ProviderV3 */
+  textEmbeddingModel(modelId: string): never;
+  languageModel(modelId: string): never;
 }
 interface DatabricksProviderSettings {
   /** Base URL for the Databricks API calls. */
@@ -37,32 +45,16 @@ interface DatabricksProviderSettings {
     baseUrl?: string;
     path: string;
   }) => string;
+  /**
+   * When true, marks all tool calls as dynamic and provider-executed.
+   * This indicates that tool execution is handled remotely by Databricks agents.
+   * Defaults to true.
+   */
+  useRemoteToolCalling?: boolean;
 }
-declare const createDatabricksProvider: (settings: DatabricksProviderSettings) => DatabricksProvider; //#endregion
-//#region src/tools.d.ts
-
-//# sourceMappingURL=databricks-provider.d.ts.map
-declare const DATABRICKS_TOOL_CALL_ID = "databricks-tool-call";
-/**
- * The AI-SDK requires that tools used by the model are defined ahead of time.
- *
- * Since tool calls can be orchestrated by Databricks' agents we don't know the name, input, or output schemas
- * of the tools until the model is called.
- *
- * In the DatabricksProvider we transform all tool calls to fit this definition, and keep the
- * original name as part of the metadata. This allows us to parse any tool orchestrated by Databricks' agents,
- * while still being able to render the tool call and result in the UI, and pass it back to the model with the correct name.
- */
-declare const DATABRICKS_TOOL_DEFINITION: {
-  name: string;
-  description: string;
-  inputSchema: z.ZodAny;
-  outputSchema: z.ZodAny;
-};
-
+declare const createDatabricksProvider: (settings: DatabricksProviderSettings) => DatabricksProvider;
 //#endregion
 //#region src/mcp.d.ts
-//# sourceMappingURL=tools.d.ts.map
 /**
  * MCP Approval Utility Functions
  *
@@ -173,9 +165,6 @@ declare function createApprovalStatusOutput(approve: boolean): ApprovalStatusOut
  * // 'awaiting-approval' | 'approved' | 'denied'
  */
 declare function getMcpApprovalState(output: unknown): McpApprovalState;
-
 //#endregion
-//# sourceMappingURL=mcp.d.ts.map
-
-export { ApprovalStatusOutput, DATABRICKS_TOOL_CALL_ID, DATABRICKS_TOOL_DEFINITION, DatabricksLanguageModelConfig, DatabricksProvider, DatabricksProviderSettings, DatabricksToolMetadata, MCP_APPROVAL_REQUEST_TYPE, MCP_APPROVAL_RESPONSE_TYPE, MCP_APPROVAL_STATUS_KEY, McpApprovalState, createApprovalStatusOutput, createDatabricksProvider, extractApprovalStatus, extractApprovalStatusFromToolResult, extractDatabricksMetadata, getMcpApprovalState, isApprovalStatusOutput, isMcpApprovalRequest, isMcpApprovalResponse };
+export { ApprovalStatusOutput, DatabricksLanguageModelConfig, DatabricksProvider, DatabricksProviderSettings, DatabricksToolMetadata, MCP_APPROVAL_REQUEST_TYPE, MCP_APPROVAL_RESPONSE_TYPE, MCP_APPROVAL_STATUS_KEY, McpApprovalState, createApprovalStatusOutput, createDatabricksProvider, extractApprovalStatus, extractApprovalStatusFromToolResult, extractDatabricksMetadata, getMcpApprovalState, isApprovalStatusOutput, isMcpApprovalRequest, isMcpApprovalResponse };
 //# sourceMappingURL=index.d.cts.map

package/dist/index.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.cts","names":[],"sources":["../src/databricks-provider.ts","../src/tools.ts","../src/mcp.ts"],"sourcesContent":null,"mappings":";;;;;KAMY,6BAAA;;iBAEK;EAFL,GAAA,EAAA,CAAA,OAAA,EAAA;IAA6B,IAAA,EAAA,MAAA;EAAA,CAAA,EAExB,GAAA,MAAA;EAAM,KAEb,CAAA,EAAA,aAAA;AAAa,CAAA;AAGN,UAAA,kBAAA,SAA2B,UAAR,CAAA;EAAA;EAAA,SAEN,CAAA,OAAA,EAAA,MAAA,CAAA,EAAA,eAAA;EAAe,SACf,CAAA,OAAA,EAAA,MAAA,CAAA,EAAA,eAAA;EAAe;EAGM,eANP,CAAA,OAAA,EAAA,MAAA,CAAA,EAMR,eANQ;AAAU;AASrC,UAAA,0BAAA,CAA0B;EAAA;EAAA,OAI/B,EAAA,MAAA;EAAM;EAQK,OAAA,CAAA,EARX,MAQW,CAAA,MAAA,EAAA,MAAA,CAAA;EAQV;EAiDZ,QAAA,CAAA,EAAA,MAAA;EAAA;;AAAA;;UAzDS;;AC/BV;;;;;;;cDuCa,qCACD,+BACT;;;;cCzCU,uBAAA;;;;ADGb;;;;AAIuB;AAGvB;;AAE8B,cCAjB,0BDAiB,EAAA;EAAe,IACf,EAAA,MAAA;EAAe,WAGT,EAAA,MAAA;EAAe,WANP,ECQ5B,CAAA,CAAA,MDR4B;EAAU,YAAA,UAAA;AAStD,CAAA;;;;;;;;;;AAhBA;;AAEiB,cEIJ,uBAAA,GFJI,oBAAA;;AAEM,cEKV,yBAAA,GFLU,sBAAA;AAGvB;AAAoC,cEKvB,0BAAA,GFLuB,uBAAA;;AAGN,KESlB,oBAAA,GFTkB;EAAe,CEU1C,uBAAuB,CFPU,EAAA,OAAA;CAAe;AANG;AASrC,KEQL,gBAAA,GFRK,mBAA0B,GAAA,UAAA,GAAA,QAAA;;AAI/B,UEOK,sBAAA,CFPL;EAAM,IAQR,CAAA,EAAA,MAAA;EAAa,QAAA,CAAA,EAAA,MAAA;EAQV,MAAA,CAAA,EAAA,MAAA;EAiDZ,WAAA,CAAA,EAAA,MAAA;EAAA,aAhDW,CAAA,EAAA,MAAA;EAA0B,iBACnC,CAAA,EAAA,MAAA;EA+CF,OAAA,CAAA,EAAA,OAAA;;;;ACxFD;;;;;;;iBCqDgB,sBAAA,6BAAmD;;;;ADzCnE;;;;;;iBC2DgB,oBAAA,WACJ,yBAAyB;;;;iBAQrB,qBAAA,WACJ,yBAAyB;;;;;AAxErC;;AAGA;;AAGa,iBAmFG,yBAAA,CAnFuB,IAAA,EAAA;yBAoFd;AA7EzB,CAAA,CAAA,EA8EI,sBA9EQ,GAAoB,SAC7B;;AAIH;;AAGA;;;;;;;;iBAwFgB,qBAAA;AAjEhB;;;;;;;iBA+EgB,mCAAA;;;AA7DhB,CAAA,CAAA,EAAgB,OAAA,GAAA,SAAA;;;;AAC2B;;;;AAQ3C;;AACY,iBAkFI,0BAAA,CAlFJ,OAAA,EAAA,OAAA,CAAA,EAkFkD,oBAlFlD;;AAA+B;;;;;;;;;AAiB3C;;;AAEI,iBAoFY,mBAAA,CApFZ,MAAA,EAAA,OAAA,CAAA,EAoFkD,gBApFlD;;;AAAsB"}
+{"version":3,"file":"index.d.cts","names":[],"sources":["../src/databricks-provider.ts","../src/mcp.ts"],"mappings":";;;;KAMY,6BAAA;EACV,QAAA;EACA,OAAA,QAAe,MAAA;EACf,GAAA,GAAM,OAAA;IAAW,IAAA;EAAA;EACjB,KAAA,GAAQ,aAAA;EAFR;;;;;EAQA,oBAAA;AAAA;AAAA,UAGe,kBAAA,SAA2B,UAAA;EAHtB;EAKpB,SAAA,CAAU,OAAA,WAAkB,eAAA;EAC5B,SAAA,CAAU,OAAA,WAAkB,eAAA;EAHM;EAMlC,eAAA,CAAgB,OAAA,WAAkB,eAAA;EAJN;EAO5B,kBAAA,CAAmB,OAAA;EACnB,aAAA,CAAc,OAAA;AAAA;AAAA,UAGC,0BAAA;EAbqC;EAepD,OAAA;EAbA;EAeA,OAAA,GAAU,MAAA;EAfkB;EAiB5B,QAAA;EAhBU;;;;EAsBV,KAAA,GAAQ,aAAA;EAhBR;;;EAqBA,SAAA,IAAa,OAAA;IAAW,OAAA;IAAkB,IAAA;EAAA;EAjBD;;;;;EAwBzC,oBAAA;AAAA;AAAA,cAGW,wBAAA,GACX,QAAA,EAAU,0BAAA,KACT,kBAAA;;;;;;;AAvDH;;;cCMa,uBAAA;;cAGA,yBAAA;;cAGA,0BAAA;;KAOD,oBAAA;EAAA,CACT,uBAAA;AAAA;;KAIS,gBAAA;;UAGK,sBAAA;EACf,IAAA;EACA,QAAA;EACA,MAAA;EACA,WAAA;EACA,aAAA;EACA,iBAAA;EACA,OAAA;EACA,MAAA;AAAA;;;;;;;;;iBAec,sBAAA,CAAuB,MAAA,YAAkB,MAAA,IAAU,oBAAA;;;;;;;ADxBnE;;;iBC0CgB,oBAAA,CACd,QAAA,EAAU,sBAAA,GAAyB,MAAA;;;;iBAQrB,qBAAA,CACd,QAAA,EAAU,sBAAA,GAAyB,MAAA;;;;;;;;;iBAiBrB,yBAAA,CAA0B,IAAA;EACxC,oBAAA,GAAuB,MAAA;AAAA,IACrB,sBAAA;;;;;;;;;;;AA3FJ;iBA6GgB,qBAAA,CAAsB,MAAA;;;;AA1GtC;;;;iBAwHgB,mCAAA,CAAoC,MAAA;EAClD,IAAA;EACA,KAAA;AAAA;;;AAhHF;;;;;AAKA;;iBAwIgB,0BAAA,CAA2B,OAAA,YAAmB,oBAAA;;;AArI9D;;;;;;;;;;;iBA0JgB,mBAAA,CAAoB,MAAA,YAAkB,gBAAA"}

package/dist/{index.d.ts → index.d.mts}

@@ -1,6 +1,5 @@
 import { FetchFunction } from "@ai-sdk/provider-utils";
-import { z } from "zod/v4";
-import { LanguageModelV2, ProviderV2 } from "@ai-sdk/provider";
+import { LanguageModelV3, ProviderV3 } from "@ai-sdk/provider";
 
 //#region src/databricks-provider.d.ts
 type DatabricksLanguageModelConfig = {
@@ -10,13 +9,22 @@ type DatabricksLanguageModelConfig = {
     path: string;
   }) => string;
   fetch?: FetchFunction;
+  /**
+   * When true, marks all tool calls as dynamic and provider-executed.
+   * This indicates that tool execution is handled remotely by Databricks agents.
+   * Defaults to true.
+   */
+  useRemoteToolCalling?: boolean;
 };
-interface DatabricksProvider extends ProviderV2 {
+interface DatabricksProvider extends ProviderV3 {
   /** Agents */
-  chatAgent(modelId: string): LanguageModelV2;
-  responses(modelId: string): LanguageModelV2;
+  chatAgent(modelId: string): LanguageModelV3;
+  responses(modelId: string): LanguageModelV3;
   /** Foundation Models */
-  chatCompletions(modelId: string): LanguageModelV2;
+  chatCompletions(modelId: string): LanguageModelV3;
+  /** Required overrides from ProviderV3 */
+  textEmbeddingModel(modelId: string): never;
+  languageModel(modelId: string): never;
 }
 interface DatabricksProviderSettings {
   /** Base URL for the Databricks API calls. */
@@ -37,32 +45,16 @@ interface DatabricksProviderSettings {
     baseUrl?: string;
     path: string;
   }) => string;
+  /**
+   * When true, marks all tool calls as dynamic and provider-executed.
+   * This indicates that tool execution is handled remotely by Databricks agents.
+   * Defaults to true.
+   */
+  useRemoteToolCalling?: boolean;
 }
-declare const createDatabricksProvider: (settings: DatabricksProviderSettings) => DatabricksProvider; //#endregion
-//#region src/tools.d.ts
-
-//# sourceMappingURL=databricks-provider.d.ts.map
-declare const DATABRICKS_TOOL_CALL_ID = "databricks-tool-call";
-/**
- * The AI-SDK requires that tools used by the model are defined ahead of time.
- *
- * Since tool calls can be orchestrated by Databricks' agents we don't know the name, input, or output schemas
- * of the tools until the model is called.
- *
- * In the DatabricksProvider we transform all tool calls to fit this definition, and keep the
- * original name as part of the metadata. This allows us to parse any tool orchestrated by Databricks' agents,
- * while still being able to render the tool call and result in the UI, and pass it back to the model with the correct name.
- */
-declare const DATABRICKS_TOOL_DEFINITION: {
-  name: string;
-  description: string;
-  inputSchema: z.ZodAny;
-  outputSchema: z.ZodAny;
-};
-
+declare const createDatabricksProvider: (settings: DatabricksProviderSettings) => DatabricksProvider;
 //#endregion
 //#region src/mcp.d.ts
-//# sourceMappingURL=tools.d.ts.map
 /**
  * MCP Approval Utility Functions
  *
@@ -173,9 +165,6 @@ declare function createApprovalStatusOutput(approve: boolean): ApprovalStatusOut
  * // 'awaiting-approval' | 'approved' | 'denied'
  */
 declare function getMcpApprovalState(output: unknown): McpApprovalState;
-
 //#endregion
-//# sourceMappingURL=mcp.d.ts.map
-
-export { ApprovalStatusOutput, DATABRICKS_TOOL_CALL_ID, DATABRICKS_TOOL_DEFINITION, DatabricksLanguageModelConfig, DatabricksProvider, DatabricksProviderSettings, DatabricksToolMetadata, MCP_APPROVAL_REQUEST_TYPE, MCP_APPROVAL_RESPONSE_TYPE, MCP_APPROVAL_STATUS_KEY, McpApprovalState, createApprovalStatusOutput, createDatabricksProvider, extractApprovalStatus, extractApprovalStatusFromToolResult, extractDatabricksMetadata, getMcpApprovalState, isApprovalStatusOutput, isMcpApprovalRequest, isMcpApprovalResponse };
-//# sourceMappingURL=index.d.ts.map
+export { ApprovalStatusOutput, DatabricksLanguageModelConfig, DatabricksProvider, DatabricksProviderSettings, DatabricksToolMetadata, MCP_APPROVAL_REQUEST_TYPE, MCP_APPROVAL_RESPONSE_TYPE, MCP_APPROVAL_STATUS_KEY, McpApprovalState, createApprovalStatusOutput, createDatabricksProvider, extractApprovalStatus, extractApprovalStatusFromToolResult, extractDatabricksMetadata, getMcpApprovalState, isApprovalStatusOutput, isMcpApprovalRequest, isMcpApprovalResponse };
+//# sourceMappingURL=index.d.mts.map

package/dist/index.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.mts","names":[],"sources":["../src/databricks-provider.ts","../src/mcp.ts"],"mappings":";;;;KAMY,6BAAA;EACV,QAAA;EACA,OAAA,QAAe,MAAA;EACf,GAAA,GAAM,OAAA;IAAW,IAAA;EAAA;EACjB,KAAA,GAAQ,aAAA;EAFR;;;;;EAQA,oBAAA;AAAA;AAAA,UAGe,kBAAA,SAA2B,UAAA;EAHtB;EAKpB,SAAA,CAAU,OAAA,WAAkB,eAAA;EAC5B,SAAA,CAAU,OAAA,WAAkB,eAAA;EAHM;EAMlC,eAAA,CAAgB,OAAA,WAAkB,eAAA;EAJN;EAO5B,kBAAA,CAAmB,OAAA;EACnB,aAAA,CAAc,OAAA;AAAA;AAAA,UAGC,0BAAA;EAbqC;EAepD,OAAA;EAbA;EAeA,OAAA,GAAU,MAAA;EAfkB;EAiB5B,QAAA;EAhBU;;;;EAsBV,KAAA,GAAQ,aAAA;EAhBR;;;EAqBA,SAAA,IAAa,OAAA;IAAW,OAAA;IAAkB,IAAA;EAAA;EAjBD;;;;;EAwBzC,oBAAA;AAAA;AAAA,cAGW,wBAAA,GACX,QAAA,EAAU,0BAAA,KACT,kBAAA;;;;;;;AAvDH;;;cCMa,uBAAA;;cAGA,yBAAA;;cAGA,0BAAA;;KAOD,oBAAA;EAAA,CACT,uBAAA;AAAA;;KAIS,gBAAA;;UAGK,sBAAA;EACf,IAAA;EACA,QAAA;EACA,MAAA;EACA,WAAA;EACA,aAAA;EACA,iBAAA;EACA,OAAA;EACA,MAAA;AAAA;;;;;;;;;iBAec,sBAAA,CAAuB,MAAA,YAAkB,MAAA,IAAU,oBAAA;;;;;;;ADxBnE;;;iBC0CgB,oBAAA,CACd,QAAA,EAAU,sBAAA,GAAyB,MAAA;;;;iBAQrB,qBAAA,CACd,QAAA,EAAU,sBAAA,GAAyB,MAAA;;;;;;;;;iBAiBrB,yBAAA,CAA0B,IAAA;EACxC,oBAAA,GAAuB,MAAA;AAAA,IACrB,sBAAA;;;;;;;;;;;AA3FJ;iBA6GgB,qBAAA,CAAsB,MAAA;;;;AA1GtC;;;;iBAwHgB,mCAAA,CAAoC,MAAA;EAClD,IAAA;EACA,KAAA;AAAA;;;AAhHF;;;;;AAKA;;iBAwIgB,0BAAA,CAA2B,OAAA,YAAmB,oBAAA;;;AArI9D;;;;;;;;;;;iBA0JgB,mBAAA,CAAoB,MAAA,YAAkB,gBAAA"}