@databricks/sdk-agentkit 0.0.0-stage → 0.0.1

package/dist/v1/client.js

@@ -0,0 +1,710 @@
+import { resolveClientConfig } from "./transport.js";
+import { buildHttpRequest, executeCall, executeHttpCall, marshalRequest, parseResponse } from "./utils.js";
+import { name, version } from "../package.js";
+import { marshalAppendSessionItemsRequestSchema, marshalClearSessionItemsRequestSchema, marshalExtractMemoriesRequestSchema, marshalForkSessionRequestSchema, marshalManagedMemoryEntrySchema, marshalManagedMemoryStoreSchema, marshalPopSessionItemRequestSchema, marshalSearchManagedMemoryEntriesRequestSchema, marshalSessionSchema, marshalSessionStoreSchema, unmarshalAppendSessionItemsResponseSchema, unmarshalClearSessionItemsResponseSchema, unmarshalExtractMemoriesResponseSchema, unmarshalForkSessionResponseSchema, unmarshalListManagedMemoryEntriesResponseSchema, unmarshalListManagedMemoryStoresResponseSchema, unmarshalListSessionItemsResponseSchema, unmarshalListSessionStoresResponseSchema, unmarshalListSessionsResponseSchema, unmarshalManagedMemoryEntrySchema, unmarshalManagedMemoryStoreSchema, unmarshalPopSessionItemResponseSchema, unmarshalSearchManagedMemoryEntriesResponseSchema, unmarshalSessionSchema, unmarshalSessionStoreSchema } from "./model.js";
+import { VERSION } from "@databricks/sdk-auth";
+import { createDefault } from "@databricks/sdk-core/clientinfo";
+import { NoOpLogger } from "@databricks/sdk-core/logger";
+
+//#region src/v1/client.ts
+const PACKAGE_SEGMENT = {
+	key: "sdk-js-" + name.replace(/^@[^/]+\/sdk-/, ""),
+	value: version
+};
+var AgentKitClient = class {
+	options;
+	logger;
+	userAgent;
+	config;
+	constructor(options) {
+		this.options = options;
+		this.logger = options.logger ?? new NoOpLogger();
+		this.userAgent = createDefault().with(PACKAGE_SEGMENT).with({
+			key: "sdk-js-auth",
+			value: VERSION
+		}).with({
+			key: "auth",
+			value: options.credentials?.name() ?? "default"
+		}).toString();
+	}
+	resolveConfig() {
+		this.config ??= resolveClientConfig(this.options);
+		return this.config;
+	}
+	/** Appends items to a session. */
+	async appendSessionItems(req, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/2.0/agents/${req.parent ?? ""}/items:append`;
+		const body = marshalRequest(req, marshalAppendSessionItemsRequestSchema);
+		let resp;
+		const call = async (callSignal) => {
+			const headers = new Headers({ "Content-Type": "application/json" });
+			if (workspaceId !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			resp = parseResponse(await executeHttpCall({
+				request: buildHttpRequest("POST", url, headers, callSignal, body),
+				httpClient,
+				logger: this.logger
+			}), unmarshalAppendSessionItemsResponseSchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation completed without a result.");
+		return resp;
+	}
+	/** Clears all items from a session. */
+	async clearSessionItems(req, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/2.0/agents/${req.parent ?? ""}/items:clear`;
+		const body = marshalRequest(req, marshalClearSessionItemsRequestSchema);
+		let resp;
+		const call = async (callSignal) => {
+			const headers = new Headers({ "Content-Type": "application/json" });
+			if (workspaceId !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			resp = parseResponse(await executeHttpCall({
+				request: buildHttpRequest("POST", url, headers, callSignal, body),
+				httpClient,
+				logger: this.logger
+			}), unmarshalClearSessionItemsResponseSchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation completed without a result.");
+		return resp;
+	}
+	/**
+	* Creates a managed memory entry using exclusive-create semantics. Callers may choose the entry
+	* ID; the service generates one when it is omitted. Returns `ALREADY_EXISTS` when an entry with
+	* the same actor, session, and path already exists. Omitted `session_id` is its own uniqueness
+	* key: two omitted-session entries with the same actor and path conflict, but an omitted-session
+	* entry does not conflict with a session-scoped entry at the same actor and path.
+	*/
+	async createMemory(req, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/2.0/agents/${req.parent ?? ""}/entries`;
+		const params = new URLSearchParams();
+		if (req.managedMemoryEntryId !== void 0) params.append("managed_memory_entry_id", req.managedMemoryEntryId);
+		const query = params.toString();
+		const fullUrl = query !== "" ? `${url}?${query}` : url;
+		const body = marshalRequest(req.managedMemoryEntry, marshalManagedMemoryEntrySchema);
+		let resp;
+		const call = async (callSignal) => {
+			const headers = new Headers({ "Content-Type": "application/json" });
+			if (workspaceId !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			resp = parseResponse(await executeHttpCall({
+				request: buildHttpRequest("POST", fullUrl, headers, callSignal, body),
+				httpClient,
+				logger: this.logger
+			}), unmarshalManagedMemoryEntrySchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation completed without a result.");
+		return resp;
+	}
+	/** Creates a managed memory store in the caller's workspace. */
+	async createMemoryStore(req, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/2.0/agents/memory-stores`;
+		const params = new URLSearchParams();
+		if (req.managedMemoryStoreId !== void 0) params.append("managed_memory_store_id", req.managedMemoryStoreId);
+		const query = params.toString();
+		const fullUrl = query !== "" ? `${url}?${query}` : url;
+		const body = marshalRequest(req.managedMemoryStore, marshalManagedMemoryStoreSchema);
+		let resp;
+		const call = async (callSignal) => {
+			const headers = new Headers({ "Content-Type": "application/json" });
+			if (workspaceId !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			resp = parseResponse(await executeHttpCall({
+				request: buildHttpRequest("POST", fullUrl, headers, callSignal, body),
+				httpClient,
+				logger: this.logger
+			}), unmarshalManagedMemoryStoreSchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation completed without a result.");
+		return resp;
+	}
+	/** Creates a session within a session store. */
+	async createSession(req, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/2.0/agents/${req.parent ?? ""}/sessions`;
+		const params = new URLSearchParams();
+		if (req.sessionId !== void 0) params.append("session_id", req.sessionId);
+		const query = params.toString();
+		const fullUrl = query !== "" ? `${url}?${query}` : url;
+		const body = marshalRequest(req.session, marshalSessionSchema);
+		let resp;
+		const call = async (callSignal) => {
+			const headers = new Headers({ "Content-Type": "application/json" });
+			if (workspaceId !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			resp = parseResponse(await executeHttpCall({
+				request: buildHttpRequest("POST", fullUrl, headers, callSignal, body),
+				httpClient,
+				logger: this.logger
+			}), unmarshalSessionSchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation completed without a result.");
+		return resp;
+	}
+	/** Creates a session store. */
+	async createSessionStore(req, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/2.0/agents/session-stores`;
+		const params = new URLSearchParams();
+		if (req.sessionStoreId !== void 0) params.append("session_store_id", req.sessionStoreId);
+		const query = params.toString();
+		const fullUrl = query !== "" ? `${url}?${query}` : url;
+		const body = marshalRequest(req.sessionStore, marshalSessionStoreSchema);
+		let resp;
+		const call = async (callSignal) => {
+			const headers = new Headers({ "Content-Type": "application/json" });
+			if (workspaceId !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			resp = parseResponse(await executeHttpCall({
+				request: buildHttpRequest("POST", fullUrl, headers, callSignal, body),
+				httpClient,
+				logger: this.logger
+			}), unmarshalSessionStoreSchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation completed without a result.");
+		return resp;
+	}
+	/**
+	* Deletes a managed memory entry by resource name. Returns `NOT_FOUND` when the entry does not
+	* exist in the caller's workspace.
+	*/
+	async deleteMemory(req, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/2.0/agents/${req.name ?? ""}`;
+		const call = async (callSignal) => {
+			const headers = new Headers();
+			if (workspaceId !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			await executeHttpCall({
+				request: buildHttpRequest("DELETE", url, headers, callSignal),
+				httpClient,
+				logger: this.logger
+			});
+		};
+		await executeCall(call, options);
+	}
+	/**
+	* Deletes a managed memory store by resource name. Returns `NOT_FOUND` when the store does not
+	* exist in the caller's workspace.
+	*/
+	async deleteMemoryStore(req, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/2.0/agents/${req.name ?? ""}`;
+		const call = async (callSignal) => {
+			const headers = new Headers();
+			if (workspaceId !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			await executeHttpCall({
+				request: buildHttpRequest("DELETE", url, headers, callSignal),
+				httpClient,
+				logger: this.logger
+			});
+		};
+		await executeCall(call, options);
+	}
+	/**
+	* Deletes a session, its items, and any descendant sessions recursively. Independently retained
+	* memory is not deleted.
+	*/
+	async deleteSession(req, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/2.0/agents/${req.name ?? ""}`;
+		const call = async (callSignal) => {
+			const headers = new Headers();
+			if (workspaceId !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			await executeHttpCall({
+				request: buildHttpRequest("DELETE", url, headers, callSignal),
+				httpClient,
+				logger: this.logger
+			});
+		};
+		await executeCall(call, options);
+	}
+	/**
+	* Deletes a session store, its sessions and items, and its service-managed storage. Memory entries
+	* retained by a separate Memory Store are not deleted.
+	*/
+	async deleteSessionStore(req, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/2.0/agents/${req.name ?? ""}`;
+		const call = async (callSignal) => {
+			const headers = new Headers();
+			if (workspaceId !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			await executeHttpCall({
+				request: buildHttpRequest("DELETE", url, headers, callSignal),
+				httpClient,
+				logger: this.logger
+			});
+		};
+		await executeCall(call, options);
+	}
+	/**
+	* Synchronously extracts memories from a single session into the given memory store, returning
+	* the entries that were written.
+	*/
+	async extractMemories(req, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/2.0/agents/${req.sessionStore ?? ""}/sessions/${req.sessionId ?? ""}/extractions`;
+		const body = marshalRequest(req, marshalExtractMemoriesRequestSchema);
+		let resp;
+		const call = async (callSignal) => {
+			const headers = new Headers({ "Content-Type": "application/json" });
+			if (workspaceId !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			resp = parseResponse(await executeHttpCall({
+				request: buildHttpRequest("POST", url, headers, callSignal, body),
+				httpClient,
+				logger: this.logger
+			}), unmarshalExtractMemoriesResponseSchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation completed without a result.");
+		return resp;
+	}
+	/** Forks a session into an independent top-level copy. */
+	async forkSession(req, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/2.0/agents/${req.parent ?? ""}/sessions:fork`;
+		const body = marshalRequest(req, marshalForkSessionRequestSchema);
+		let resp;
+		const call = async (callSignal) => {
+			const headers = new Headers({ "Content-Type": "application/json" });
+			if (workspaceId !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			resp = parseResponse(await executeHttpCall({
+				request: buildHttpRequest("POST", url, headers, callSignal, body),
+				httpClient,
+				logger: this.logger
+			}), unmarshalForkSessionResponseSchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation completed without a result.");
+		return resp;
+	}
+	/**
+	* Retrieves a managed memory entry by resource name. `view` controls whether the response
+	* includes memory content. Returns `NOT_FOUND` when the entry does not exist in the caller's
+	* workspace.
+	*/
+	async getMemory(req, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/2.0/agents/${req.name ?? ""}`;
+		const params = new URLSearchParams();
+		if (req.view !== void 0) params.append("view", req.view);
+		const query = params.toString();
+		const fullUrl = query !== "" ? `${url}?${query}` : url;
+		let resp;
+		const call = async (callSignal) => {
+			const headers = new Headers();
+			if (workspaceId !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			resp = parseResponse(await executeHttpCall({
+				request: buildHttpRequest("GET", fullUrl, headers, callSignal),
+				httpClient,
+				logger: this.logger
+			}), unmarshalManagedMemoryEntrySchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation completed without a result.");
+		return resp;
+	}
+	/**
+	* Retrieves a managed memory store by resource name. Returns `NOT_FOUND` when the store does not
+	* exist in the caller's workspace.
+	*/
+	async getMemoryStore(req, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/2.0/agents/${req.name ?? ""}`;
+		let resp;
+		const call = async (callSignal) => {
+			const headers = new Headers();
+			if (workspaceId !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			resp = parseResponse(await executeHttpCall({
+				request: buildHttpRequest("GET", url, headers, callSignal),
+				httpClient,
+				logger: this.logger
+			}), unmarshalManagedMemoryStoreSchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation completed without a result.");
+		return resp;
+	}
+	/** Gets a session by resource name. */
+	async getSession(req, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/2.0/agents/${req.name ?? ""}`;
+		let resp;
+		const call = async (callSignal) => {
+			const headers = new Headers();
+			if (workspaceId !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			resp = parseResponse(await executeHttpCall({
+				request: buildHttpRequest("GET", url, headers, callSignal),
+				httpClient,
+				logger: this.logger
+			}), unmarshalSessionSchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation completed without a result.");
+		return resp;
+	}
+	/** Gets a session store by resource name. */
+	async getSessionStore(req, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/2.0/agents/${req.name ?? ""}`;
+		let resp;
+		const call = async (callSignal) => {
+			const headers = new Headers();
+			if (workspaceId !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			resp = parseResponse(await executeHttpCall({
+				request: buildHttpRequest("GET", url, headers, callSignal),
+				httpClient,
+				logger: this.logger
+			}), unmarshalSessionStoreSchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation completed without a result.");
+		return resp;
+	}
+	/**
+	* Lists managed memory entries for one actor. An exact `path` filters entries across sessions,
+	* ignoring session metadata. Otherwise, `session_id` and `path_prefix` restrict the actor
+	* partition. `view` controls whether each returned entry includes memory content.
+	*/
+	async listMemories(req, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/2.0/agents/${req.parent ?? ""}/entries`;
+		const params = new URLSearchParams();
+		if (req.pageSize !== void 0) params.append("page_size", String(req.pageSize));
+		if (req.pageToken !== void 0) params.append("page_token", req.pageToken);
+		if (req.actorId !== void 0) params.append("actor_id", req.actorId);
+		if (req.pathPrefix !== void 0) params.append("path_prefix", req.pathPrefix);
+		if (req.sessionId !== void 0) params.append("session_id", req.sessionId);
+		if (req.view !== void 0) params.append("view", req.view);
+		const query = params.toString();
+		const fullUrl = query !== "" ? `${url}?${query}` : url;
+		let resp;
+		const call = async (callSignal) => {
+			const headers = new Headers();
+			if (workspaceId !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			resp = parseResponse(await executeHttpCall({
+				request: buildHttpRequest("GET", fullUrl, headers, callSignal),
+				httpClient,
+				logger: this.logger
+			}), unmarshalListManagedMemoryEntriesResponseSchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation completed without a result.");
+		return resp;
+	}
+	async *listMemoriesIter(req, options) {
+		const pageReq = { ...req };
+		for (;;) {
+			const resp = await this.listMemories(pageReq, options);
+			for (const item of resp.managedMemoryEntries ?? []) yield item;
+			if (resp.nextPageToken === void 0 || resp.nextPageToken === "") return;
+			pageReq.pageToken = resp.nextPageToken;
+		}
+	}
+	/** Lists managed memory stores in the caller's workspace. */
+	async listMemoryStores(req, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/2.0/agents/memory-stores`;
+		const params = new URLSearchParams();
+		if (req.pageSize !== void 0) params.append("page_size", String(req.pageSize));
+		if (req.pageToken !== void 0) params.append("page_token", req.pageToken);
+		const query = params.toString();
+		const fullUrl = query !== "" ? `${url}?${query}` : url;
+		let resp;
+		const call = async (callSignal) => {
+			const headers = new Headers();
+			if (workspaceId !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			resp = parseResponse(await executeHttpCall({
+				request: buildHttpRequest("GET", fullUrl, headers, callSignal),
+				httpClient,
+				logger: this.logger
+			}), unmarshalListManagedMemoryStoresResponseSchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation completed without a result.");
+		return resp;
+	}
+	async *listMemoryStoresIter(req, options) {
+		const pageReq = { ...req };
+		for (;;) {
+			const resp = await this.listMemoryStores(pageReq, options);
+			for (const item of resp.managedMemoryStores ?? []) yield item;
+			if (resp.nextPageToken === void 0 || resp.nextPageToken === "") return;
+			pageReq.pageToken = resp.nextPageToken;
+		}
+	}
+	/** Lists items in a session. */
+	async listSessionItems(req, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/2.0/agents/${req.parent ?? ""}/items`;
+		const params = new URLSearchParams();
+		if (req.pageSize !== void 0) params.append("page_size", String(req.pageSize));
+		if (req.pageToken !== void 0) params.append("page_token", req.pageToken);
+		if (req.orderBy !== void 0) params.append("order_by", req.orderBy);
+		const query = params.toString();
+		const fullUrl = query !== "" ? `${url}?${query}` : url;
+		let resp;
+		const call = async (callSignal) => {
+			const headers = new Headers();
+			if (workspaceId !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			resp = parseResponse(await executeHttpCall({
+				request: buildHttpRequest("GET", fullUrl, headers, callSignal),
+				httpClient,
+				logger: this.logger
+			}), unmarshalListSessionItemsResponseSchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation completed without a result.");
+		return resp;
+	}
+	async *listSessionItemsIter(req, options) {
+		const pageReq = { ...req };
+		for (;;) {
+			const resp = await this.listSessionItems(pageReq, options);
+			for (const item of resp.sessionItems ?? []) yield item;
+			if (resp.nextPageToken === void 0 || resp.nextPageToken === "") return;
+			pageReq.pageToken = resp.nextPageToken;
+		}
+	}
+	/** Lists session stores. */
+	async listSessionStores(req, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/2.0/agents/session-stores`;
+		const params = new URLSearchParams();
+		if (req.pageSize !== void 0) params.append("page_size", String(req.pageSize));
+		if (req.pageToken !== void 0) params.append("page_token", req.pageToken);
+		const query = params.toString();
+		const fullUrl = query !== "" ? `${url}?${query}` : url;
+		let resp;
+		const call = async (callSignal) => {
+			const headers = new Headers();
+			if (workspaceId !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			resp = parseResponse(await executeHttpCall({
+				request: buildHttpRequest("GET", fullUrl, headers, callSignal),
+				httpClient,
+				logger: this.logger
+			}), unmarshalListSessionStoresResponseSchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation completed without a result.");
+		return resp;
+	}
+	async *listSessionStoresIter(req, options) {
+		const pageReq = { ...req };
+		for (;;) {
+			const resp = await this.listSessionStores(pageReq, options);
+			for (const item of resp.sessionStores ?? []) yield item;
+			if (resp.nextPageToken === void 0 || resp.nextPageToken === "") return;
+			pageReq.pageToken = resp.nextPageToken;
+		}
+	}
+	/** Lists sessions within a session store. */
+	async listSessions(req, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/2.0/agents/${req.parent ?? ""}/sessions`;
+		const params = new URLSearchParams();
+		if (req.pageSize !== void 0) params.append("page_size", String(req.pageSize));
+		if (req.pageToken !== void 0) params.append("page_token", req.pageToken);
+		if (req.filter !== void 0) params.append("filter", req.filter);
+		if (req.orderBy !== void 0) params.append("order_by", req.orderBy);
+		const query = params.toString();
+		const fullUrl = query !== "" ? `${url}?${query}` : url;
+		let resp;
+		const call = async (callSignal) => {
+			const headers = new Headers();
+			if (workspaceId !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			resp = parseResponse(await executeHttpCall({
+				request: buildHttpRequest("GET", fullUrl, headers, callSignal),
+				httpClient,
+				logger: this.logger
+			}), unmarshalListSessionsResponseSchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation completed without a result.");
+		return resp;
+	}
+	async *listSessionsIter(req, options) {
+		const pageReq = { ...req };
+		for (;;) {
+			const resp = await this.listSessions(pageReq, options);
+			for (const item of resp.sessions ?? []) yield item;
+			if (resp.nextPageToken === void 0 || resp.nextPageToken === "") return;
+			pageReq.pageToken = resp.nextPageToken;
+		}
+	}
+	/** Pops the newest item from a session. */
+	async popSessionItem(req, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/2.0/agents/${req.parent ?? ""}/items:pop`;
+		const body = marshalRequest(req, marshalPopSessionItemRequestSchema);
+		let resp;
+		const call = async (callSignal) => {
+			const headers = new Headers({ "Content-Type": "application/json" });
+			if (workspaceId !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			resp = parseResponse(await executeHttpCall({
+				request: buildHttpRequest("POST", url, headers, callSignal, body),
+				httpClient,
+				logger: this.logger
+			}), unmarshalPopSessionItemResponseSchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation completed without a result.");
+		return resp;
+	}
+	/**
+	* Searches managed memory entries by text query for one actor. Returns matching entries and
+	* scores ranked by relevance; `view` controls whether each returned entry includes content.
+	*/
+	async searchMemories(req, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/2.0/agents/${req.parent ?? ""}/entries:search`;
+		const body = marshalRequest(req, marshalSearchManagedMemoryEntriesRequestSchema);
+		let resp;
+		const call = async (callSignal) => {
+			const headers = new Headers({ "Content-Type": "application/json" });
+			if (workspaceId !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			resp = parseResponse(await executeHttpCall({
+				request: buildHttpRequest("POST", url, headers, callSignal, body),
+				httpClient,
+				logger: this.logger
+			}), unmarshalSearchManagedMemoryEntriesResponseSchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation completed without a result.");
+		return resp;
+	}
+	async *searchMemoriesIter(req, options) {
+		const pageReq = { ...req };
+		for (;;) {
+			const resp = await this.searchMemories(pageReq, options);
+			for (const item of resp.results ?? []) yield item;
+			if (resp.nextPageToken === void 0 || resp.nextPageToken === "") return;
+			pageReq.pageToken = resp.nextPageToken;
+		}
+	}
+	/** Updates selected mutable fields on a managed memory entry. Identity fields are immutable. */
+	async updateMemory(req, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/2.0/agents/${req.managedMemoryEntry?.name ?? ""}`;
+		const params = new URLSearchParams();
+		if (req.updateMask !== void 0) params.append("update_mask", req.updateMask.toString());
+		const query = params.toString();
+		const fullUrl = query !== "" ? `${url}?${query}` : url;
+		const body = marshalRequest(req.managedMemoryEntry, marshalManagedMemoryEntrySchema);
+		let resp;
+		const call = async (callSignal) => {
+			const headers = new Headers({ "Content-Type": "application/json" });
+			if (workspaceId !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			resp = parseResponse(await executeHttpCall({
+				request: buildHttpRequest("PATCH", fullUrl, headers, callSignal, body),
+				httpClient,
+				logger: this.logger
+			}), unmarshalManagedMemoryEntrySchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation completed without a result.");
+		return resp;
+	}
+	/** Updates a managed memory store's description and metadata. */
+	async updateMemoryStore(req, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/2.0/agents/${req.managedMemoryStore?.name ?? ""}`;
+		const params = new URLSearchParams();
+		if (req.updateMask !== void 0) params.append("update_mask", req.updateMask.toString());
+		const query = params.toString();
+		const fullUrl = query !== "" ? `${url}?${query}` : url;
+		const body = marshalRequest(req.managedMemoryStore, marshalManagedMemoryStoreSchema);
+		let resp;
+		const call = async (callSignal) => {
+			const headers = new Headers({ "Content-Type": "application/json" });
+			if (workspaceId !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			resp = parseResponse(await executeHttpCall({
+				request: buildHttpRequest("PATCH", fullUrl, headers, callSignal, body),
+				httpClient,
+				logger: this.logger
+			}), unmarshalManagedMemoryStoreSchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation completed without a result.");
+		return resp;
+	}
+	/** Updates a session's mutable fields. */
+	async updateSession(req, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/2.0/agents/${req.session?.name ?? ""}`;
+		const params = new URLSearchParams();
+		if (req.updateMask !== void 0) params.append("update_mask", req.updateMask.toString());
+		const query = params.toString();
+		const fullUrl = query !== "" ? `${url}?${query}` : url;
+		const body = marshalRequest(req.session, marshalSessionSchema);
+		let resp;
+		const call = async (callSignal) => {
+			const headers = new Headers({ "Content-Type": "application/json" });
+			if (workspaceId !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			resp = parseResponse(await executeHttpCall({
+				request: buildHttpRequest("PATCH", fullUrl, headers, callSignal, body),
+				httpClient,
+				logger: this.logger
+			}), unmarshalSessionSchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation completed without a result.");
+		return resp;
+	}
+	/** Updates a session store's description and metadata. */
+	async updateSessionStore(req, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/2.0/agents/${req.sessionStore?.name ?? ""}`;
+		const params = new URLSearchParams();
+		if (req.updateMask !== void 0) params.append("update_mask", req.updateMask.toString());
+		const query = params.toString();
+		const fullUrl = query !== "" ? `${url}?${query}` : url;
+		const body = marshalRequest(req.sessionStore, marshalSessionStoreSchema);
+		let resp;
+		const call = async (callSignal) => {
+			const headers = new Headers({ "Content-Type": "application/json" });
+			if (workspaceId !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			resp = parseResponse(await executeHttpCall({
+				request: buildHttpRequest("PATCH", fullUrl, headers, callSignal, body),
+				httpClient,
+				logger: this.logger
+			}), unmarshalSessionStoreSchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation completed without a result.");
+		return resp;
+	}
+};
+
+//#endregion
+export { AgentKitClient };
+//# sourceMappingURL=client.js.map