@azure/ai-projects 2.1.1 → 2.2.0

package/dist/esm/api/beta/agents/operations.js

@@ -1,15 +1,366 @@
 // Copyright (c) Microsoft Corporation.
 // Licensed under the MIT License.
-import { agentDeserializer, agentEndpointSerializer, agentCardSerializer, apiErrorResponseDeserializer, versionIndicatorUnionSerializer, agentSessionResourceDeserializer, _agentsPagedResultAgentSessionResourceDeserializer, sessionFileWriteResponseDeserializer, sessionDirectoryListResponseDeserializer, } from "../../../models/models.js";
+import { agentDeserializer, agentVersionDeserializer, agentEndpointConfigSerializer, agentCardSerializer, apiErrorResponseDeserializer, createAgentVersionFromCodeContentSerializer, versionIndicatorUnionSerializer, agentSessionResourceDeserializer, _agentsPagedResultAgentSessionResourceDeserializer, sessionFileWriteResponseDeserializer, sessionDirectoryListResponseDeserializer, optimizationJobInputsSerializer, optimizationJobDeserializer, _agentsPagedResultOptimizationJobDeserializer, agentsPagedResultOptimizationCandidateDeserializer, candidateMetadataDeserializer, candidateDeployConfigDeserializer, candidateResultsDeserializer, promoteCandidateRequestSerializer, promoteCandidateResponseDeserializer, } from "../../../models/models.js";
 import { buildPagedAsyncIterator } from "../../../static-helpers/pagingHelpers.js";
 import { getBinaryStreamResponse } from "../../../static-helpers/serialization/get-binary-stream-response.js";
 import { expandUrlTemplate } from "../../../static-helpers/urlTemplate.js";
 import { createRestError, operationOptionsToRequestParameters } from "@azure-rest/core-client";
-export function _deleteSessionFileSend(context, agentName, sessionId, path, options = { requestOptions: {} }) {
-    const foundryFeatures = "HostedAgents=V1Preview";
-    const path_1 = expandUrlTemplate("/agents/{agent_name}/endpoint/sessions/{session_id}/files{?path,recursive,api-version}", {
+export function _promoteCandidateSend(context, jobId, candidateId, candidateRequest, options = { requestOptions: {} }) {
+    const foundryFeatures = "HostedAgents=V1Preview,AgentEndpoints=V1Preview";
+    const path = expandUrlTemplate("/agent_optimization_jobs/{jobId}/candidates/{candidateId}:promote{?api-version}", {
+        jobId: jobId,
+        candidateId: candidateId,
+        "api-version": context.apiVersion,
+    }, {
+        allowReserved: options?.requestOptions?.skipUrlEncoding,
+    });
+    return context.path(path).post({
+        ...operationOptionsToRequestParameters(options),
+        contentType: "application/json",
+        headers: {
+            "foundry-features": foundryFeatures,
+            accept: "application/json",
+            ...options.requestOptions?.headers,
+        },
+        body: promoteCandidateRequestSerializer(candidateRequest),
+    });
+}
+export async function _promoteCandidateDeserialize(result) {
+    const expectedStatuses = ["200"];
+    if (!expectedStatuses.includes(result.status)) {
+        const error = createRestError(result);
+        error.details = apiErrorResponseDeserializer(result.body);
+        throw error;
+    }
+    return promoteCandidateResponseDeserializer(result.body);
+}
+/** Promotes a candidate, recording the deployment timestamp and target agent version. */
+export async function promoteCandidate(context, jobId, candidateId, candidateRequest, options = { requestOptions: {} }) {
+    const result = await _promoteCandidateSend(context, jobId, candidateId, candidateRequest, options);
+    return _promoteCandidateDeserialize(result);
+}
+export function _getCandidateFileSend(context, jobId, candidateId, path, options = { requestOptions: {} }) {
+    const path_1 = expandUrlTemplate("/agent_optimization_jobs/{jobId}/candidates/{candidateId}/files{?path,api-version}", {
+        jobId: jobId,
+        candidateId: candidateId,
+        path: path,
+        "api-version": context.apiVersion,
+    }, {
+        allowReserved: options?.requestOptions?.skipUrlEncoding,
+    });
+    return context.path(path_1).get({
+        ...operationOptionsToRequestParameters(options),
+        headers: {
+            "foundry-features": "HostedAgents=V1Preview,AgentEndpoints=V1Preview",
+            accept: "application/octet-stream",
+            ...options.requestOptions?.headers,
+        },
+    });
+}
+export async function _getCandidateFileDeserialize(result) {
+    const expectedStatuses = ["200"];
+    if (!expectedStatuses.includes(result.status)) {
+        const error = createRestError(result);
+        error.details = apiErrorResponseDeserializer(result.body);
+        throw error;
+    }
+    return { blobBody: result.blobBody, readableStreamBody: result.readableStreamBody };
+}
+/** Stream a specific file from the candidate's blob directory. */
+export async function getCandidateFile(context, jobId, candidateId, path, options = { requestOptions: {} }) {
+    const result = await _getCandidateFileSend(context, jobId, candidateId, path, options);
+    return _getCandidateFileDeserialize(result);
+}
+export function _getOptimizationCandidateResultsSend(context, jobId, candidateId, options = { requestOptions: {} }) {
+    const path = expandUrlTemplate("/agent_optimization_jobs/{jobId}/candidates/{candidateId}/results{?api-version}", {
+        jobId: jobId,
+        candidateId: candidateId,
+        "api-version": context.apiVersion,
+    }, {
+        allowReserved: options?.requestOptions?.skipUrlEncoding,
+    });
+    return context.path(path).get({
+        ...operationOptionsToRequestParameters(options),
+        headers: {
+            "foundry-features": "HostedAgents=V1Preview,AgentEndpoints=V1Preview",
+            accept: "application/json",
+            ...options.requestOptions?.headers,
+        },
+    });
+}
+export async function _getOptimizationCandidateResultsDeserialize(result) {
+    const expectedStatuses = ["200"];
+    if (!expectedStatuses.includes(result.status)) {
+        const error = createRestError(result);
+        error.details = apiErrorResponseDeserializer(result.body);
+        throw error;
+    }
+    return candidateResultsDeserializer(result.body);
+}
+/** Get full per-task evaluation results for a candidate. */
+export async function getOptimizationCandidateResults(context, jobId, candidateId, options = { requestOptions: {} }) {
+    const result = await _getOptimizationCandidateResultsSend(context, jobId, candidateId, options);
+    return _getOptimizationCandidateResultsDeserialize(result);
+}
+export function _getOptimizationCandidateConfigSend(context, jobId, candidateId, options = { requestOptions: {} }) {
+    const path = expandUrlTemplate("/agent_optimization_jobs/{jobId}/candidates/{candidateId}/config{?api-version}", {
+        jobId: jobId,
+        candidateId: candidateId,
+        "api-version": context.apiVersion,
+    }, {
+        allowReserved: options?.requestOptions?.skipUrlEncoding,
+    });
+    return context.path(path).get({
+        ...operationOptionsToRequestParameters(options),
+        headers: {
+            "foundry-features": "HostedAgents=V1Preview,AgentEndpoints=V1Preview",
+            accept: "application/json",
+            ...options.requestOptions?.headers,
+        },
+    });
+}
+export async function _getOptimizationCandidateConfigDeserialize(result) {
+    const expectedStatuses = ["200"];
+    if (!expectedStatuses.includes(result.status)) {
+        const error = createRestError(result);
+        error.details = apiErrorResponseDeserializer(result.body);
+        throw error;
+    }
+    return candidateDeployConfigDeserializer(result.body);
+}
+/** Get the candidate's deploy config JSON. Used to compose `agents.create_version(...)` from a candidate. */
+export async function getOptimizationCandidateConfig(context, jobId, candidateId, options = { requestOptions: {} }) {
+    const result = await _getOptimizationCandidateConfigSend(context, jobId, candidateId, options);
+    return _getOptimizationCandidateConfigDeserialize(result);
+}
+export function _getOptimizationCandidateSend(context, jobId, candidateId, options = { requestOptions: {} }) {
+    const path = expandUrlTemplate("/agent_optimization_jobs/{jobId}/candidates/{candidateId}{?api-version}", {
+        jobId: jobId,
+        candidateId: candidateId,
+        "api-version": context.apiVersion,
+    }, {
+        allowReserved: options?.requestOptions?.skipUrlEncoding,
+    });
+    return context.path(path).get({
+        ...operationOptionsToRequestParameters(options),
+        headers: {
+            "foundry-features": "HostedAgents=V1Preview,AgentEndpoints=V1Preview",
+            accept: "application/json",
+            ...options.requestOptions?.headers,
+        },
+    });
+}
+export async function _getOptimizationCandidateDeserialize(result) {
+    const expectedStatuses = ["200"];
+    if (!expectedStatuses.includes(result.status)) {
+        const error = createRestError(result);
+        error.details = apiErrorResponseDeserializer(result.body);
+        throw error;
+    }
+    return candidateMetadataDeserializer(result.body);
+}
+/** Get a single candidate's metadata, manifest, and promotion info. */
+export async function getOptimizationCandidate(context, jobId, candidateId, options = { requestOptions: {} }) {
+    const result = await _getOptimizationCandidateSend(context, jobId, candidateId, options);
+    return _getOptimizationCandidateDeserialize(result);
+}
+export function _listOptimizationCandidatesSend(context, jobId, options = { requestOptions: {} }) {
+    const path = expandUrlTemplate("/agent_optimization_jobs/{jobId}/candidates{?limit,order,after,before,api-version}", {
+        jobId: jobId,
+        limit: options?.limit,
+        order: options?.order,
+        after: options?.after,
+        before: options?.before,
+        "api-version": context.apiVersion,
+    }, {
+        allowReserved: options?.requestOptions?.skipUrlEncoding,
+    });
+    return context.path(path).get({
+        ...operationOptionsToRequestParameters(options),
+        headers: {
+            "foundry-features": "HostedAgents=V1Preview,AgentEndpoints=V1Preview",
+            accept: "application/json",
+            ...options.requestOptions?.headers,
+        },
+    });
+}
+export async function _listOptimizationCandidatesDeserialize(result) {
+    const expectedStatuses = ["200"];
+    if (!expectedStatuses.includes(result.status)) {
+        const error = createRestError(result);
+        error.details = apiErrorResponseDeserializer(result.body);
+        throw error;
+    }
+    return agentsPagedResultOptimizationCandidateDeserializer(result.body);
+}
+/** List candidates produced by a job. */
+export async function listOptimizationCandidates(context, jobId, options = { requestOptions: {} }) {
+    const result = await _listOptimizationCandidatesSend(context, jobId, options);
+    return _listOptimizationCandidatesDeserialize(result);
+}
+export function _deleteOptimizationJobSend(context, jobId, options = { requestOptions: {} }) {
+    const foundryFeatures = "HostedAgents=V1Preview,AgentEndpoints=V1Preview";
+    const path = expandUrlTemplate("/agent_optimization_jobs/{jobId}{?force,api-version}", {
+        jobId: jobId,
+        force: options?.force,
+        "api-version": context.apiVersion,
+    }, {
+        allowReserved: options?.requestOptions?.skipUrlEncoding,
+    });
+    return context.path(path).delete({
+        ...operationOptionsToRequestParameters(options),
+        headers: {
+            "foundry-features": foundryFeatures,
+            ...options.requestOptions?.headers,
+        },
+    });
+}
+export async function _deleteOptimizationJobDeserialize(result) {
+    const expectedStatuses = ["204"];
+    if (!expectedStatuses.includes(result.status)) {
+        const error = createRestError(result);
+        error.details = apiErrorResponseDeserializer(result.body);
+        throw error;
+    }
+    return;
+}
+/** Delete the job and its candidate artifacts. Cancels first if non-terminal. */
+export async function deleteOptimizationJob(context, jobId, options = { requestOptions: {} }) {
+    const result = await _deleteOptimizationJobSend(context, jobId, options);
+    return _deleteOptimizationJobDeserialize(result);
+}
+export function _cancelOptimizationJobSend(context, jobId, options = { requestOptions: {} }) {
+    const path = expandUrlTemplate("/agent_optimization_jobs/{jobId}:cancel{?api-version}", {
+        jobId: jobId,
+        "api-version": context.apiVersion,
+    }, {
+        allowReserved: options?.requestOptions?.skipUrlEncoding,
+    });
+    return context.path(path).post({
+        ...operationOptionsToRequestParameters(options),
+        headers: {
+            "foundry-features": "HostedAgents=V1Preview,AgentEndpoints=V1Preview",
+            accept: "application/json",
+            ...options.requestOptions?.headers,
+        },
+    });
+}
+export async function _cancelOptimizationJobDeserialize(result) {
+    const expectedStatuses = ["200"];
+    if (!expectedStatuses.includes(result.status)) {
+        const error = createRestError(result);
+        error.details = apiErrorResponseDeserializer(result.body);
+        throw error;
+    }
+    return optimizationJobDeserializer(result.body);
+}
+/** Request cancellation. Idempotent on terminal states. */
+export async function cancelOptimizationJob(context, jobId, options = { requestOptions: {} }) {
+    const result = await _cancelOptimizationJobSend(context, jobId, options);
+    return _cancelOptimizationJobDeserialize(result);
+}
+export function _listOptimizationJobsSend(context, options = { requestOptions: {} }) {
+    const path = expandUrlTemplate("/agent_optimization_jobs{?limit,order,after,before,status,agent_name,api-version}", {
+        limit: options?.limit,
+        order: options?.order,
+        after: options?.after,
+        before: options?.before,
+        status: options?.status,
+        agent_name: options?.agentName,
+        "api-version": context.apiVersion,
+    }, {
+        allowReserved: options?.requestOptions?.skipUrlEncoding,
+    });
+    return context.path(path).get({
+        ...operationOptionsToRequestParameters(options),
+        headers: {
+            "foundry-features": "HostedAgents=V1Preview,AgentEndpoints=V1Preview",
+            accept: "application/json",
+            ...options.requestOptions?.headers,
+        },
+    });
+}
+export async function _listOptimizationJobsDeserialize(result) {
+    const expectedStatuses = ["200"];
+    if (!expectedStatuses.includes(result.status)) {
+        const error = createRestError(result);
+        error.details = apiErrorResponseDeserializer(result.body);
+        throw error;
+    }
+    return _agentsPagedResultOptimizationJobDeserializer(result.body);
+}
+/** List optimization jobs. Supports cursor pagination and optional status / agent_name filters. */
+export function listOptimizationJobs(context, options = { requestOptions: {} }) {
+    return buildPagedAsyncIterator(context, () => _listOptimizationJobsSend(context, options), _listOptimizationJobsDeserialize, ["200"], { itemName: "data", apiVersion: context.apiVersion });
+}
+export function _getOptimizationJobSend(context, jobId, options = { requestOptions: {} }) {
+    const path = expandUrlTemplate("/agent_optimization_jobs/{jobId}{?api-version}", {
+        jobId: jobId,
+        "api-version": context.apiVersion,
+    }, {
+        allowReserved: options?.requestOptions?.skipUrlEncoding,
+    });
+    return context.path(path).get({
+        ...operationOptionsToRequestParameters(options),
+        headers: {
+            "foundry-features": "HostedAgents=V1Preview,AgentEndpoints=V1Preview",
+            accept: "application/json",
+            ...options.requestOptions?.headers,
+        },
+    });
+}
+export async function _getOptimizationJobDeserialize(result) {
+    const expectedStatuses = ["200", "202"];
+    if (!expectedStatuses.includes(result.status)) {
+        const error = createRestError(result);
+        error.details = apiErrorResponseDeserializer(result.body);
+        throw error;
+    }
+    return optimizationJobDeserializer(result.body);
+}
+/** Get an optimization job by id. Returns 202 while in progress, 200 when terminal. */
+export async function getOptimizationJob(context, jobId, options = { requestOptions: {} }) {
+    const result = await _getOptimizationJobSend(context, jobId, options);
+    return _getOptimizationJobDeserialize(result);
+}
+export function _createOptimizationJobSend(context, inputs, options = { requestOptions: {} }) {
+    const foundryFeatures = "HostedAgents=V1Preview,AgentEndpoints=V1Preview";
+    const path = expandUrlTemplate("/agent_optimization_jobs{?api-version}", {
+        "api-version": context.apiVersion,
+    }, {
+        allowReserved: options?.requestOptions?.skipUrlEncoding,
+    });
+    return context.path(path).post({
+        ...operationOptionsToRequestParameters(options),
+        contentType: "application/json",
+        headers: {
+            "foundry-features": foundryFeatures,
+            ...(options?.operationId !== undefined ? { "operation-id": options?.operationId } : {}),
+            accept: "application/json",
+            ...options.requestOptions?.headers,
+        },
+        body: optimizationJobInputsSerializer(inputs),
+    });
+}
+export async function _createOptimizationJobDeserialize(result) {
+    const expectedStatuses = ["201"];
+    if (!expectedStatuses.includes(result.status)) {
+        const error = createRestError(result);
+        error.details = apiErrorResponseDeserializer(result.body);
+        throw error;
+    }
+    return optimizationJobDeserializer(result.body);
+}
+/** Create an optimization job. Returns 201 with the queued job. Honours `Operation-Id` for idempotent retry. */
+export async function createOptimizationJob(context, inputs, options = { requestOptions: {} }) {
+    const result = await _createOptimizationJobSend(context, inputs, options);
+    return _createOptimizationJobDeserialize(result);
+}
+export function _deleteSessionFileSend(context, agentName, agentSessionId, path, options = { requestOptions: {} }) {
+    const foundryFeatures = "HostedAgents=V1Preview,AgentEndpoints=V1Preview";
+    const path_1 = expandUrlTemplate("/agents/{agent_name}/endpoint/sessions/{agent_session_id}/files{?path,recursive,api-version}", {
         agent_name: agentName,
-        session_id: sessionId,
+        agent_session_id: agentSessionId,
         path: path,
         recursive: options?.recursive,
         "api-version": context.apiVersion,
@@ -20,6 +371,9 @@ export function _deleteSessionFileSend(context, agentName, sessionId, path, opti
         ...operationOptionsToRequestParameters(options),
         headers: {
             "foundry-features": foundryFeatures,
+            ...(options?.userIsolationKey !== undefined
+                ? { "x-ms-user-isolation-key": options?.userIsolationKey }
+                : {}),
             ...options.requestOptions?.headers,
         },
     });
@@ -37,24 +391,31 @@ export async function _deleteSessionFileDeserialize(result) {
  * Delete a file or directory from the session sandbox.
  * If `recursive` is false (default) and the target is a non-empty directory, the API returns 409 Conflict.
  */
-export async function deleteSessionFile(context, agentName, sessionId, path, options = { requestOptions: {} }) {
-    const result = await _deleteSessionFileSend(context, agentName, sessionId, path, options);
+export async function deleteSessionFile(context, agentName, agentSessionId, path, options = { requestOptions: {} }) {
+    const result = await _deleteSessionFileSend(context, agentName, agentSessionId, path, options);
     return _deleteSessionFileDeserialize(result);
 }
-export function _listSessionFilesSend(context, agentName, sessionId, path, options = { requestOptions: {} }) {
-    const foundryFeatures = "HostedAgents=V1Preview";
-    const path_1 = expandUrlTemplate("/agents/{agent_name}/endpoint/sessions/{session_id}/files{?path,api-version}", {
+export function _listSessionFilesSend(context, agentName, agentSessionId, options = { requestOptions: {} }) {
+    const foundryFeatures = "HostedAgents=V1Preview,AgentEndpoints=V1Preview";
+    const path = expandUrlTemplate("/agents/{agent_name}/endpoint/sessions/{agent_session_id}/files{?path,limit,order,after,before,api-version}", {
         agent_name: agentName,
-        session_id: sessionId,
-        path: path,
+        agent_session_id: agentSessionId,
+        path: options?.path,
+        limit: options?.limit,
+        order: options?.order,
+        after: options?.after,
+        before: options?.before,
         "api-version": context.apiVersion,
     }, {
         allowReserved: options?.requestOptions?.skipUrlEncoding,
     });
-    return context.path(path_1).get({
+    return context.path(path).get({
         ...operationOptionsToRequestParameters(options),
         headers: {
             "foundry-features": foundryFeatures,
+            ...(options?.userIsolationKey !== undefined
+                ? { "x-ms-user-isolation-key": options?.userIsolationKey }
+                : {}),
             accept: "application/json",
             ...options.requestOptions?.headers,
         },
@@ -72,16 +433,21 @@ export async function _listSessionFilesDeserialize(result) {
 /**
  * List files and directories at a given path in the session sandbox.
  * Returns only the immediate children of the specified directory (non-recursive).
+ * If path is not provided, lists the session home directory.
  */
-export async function listSessionFiles(context, agentName, sessionId, path, options = { requestOptions: {} }) {
-    const result = await _listSessionFilesSend(context, agentName, sessionId, path, options);
-    return _listSessionFilesDeserialize(result);
+export function listSessionFiles(context, agentName, agentSessionId, options = { requestOptions: {} }) {
+    return buildPagedAsyncIterator(context, () => _listSessionFilesSend(context, agentName, agentSessionId, options), _listSessionFilesDeserialize, ["200"], {
+        itemName: "entries",
+        apiVersion: context.apiVersion,
+        cursorFieldName: "last_id",
+        hasMoreFieldName: "has_more",
+    });
 }
-export function _downloadSessionFileSend(context, agentName, sessionId, path, options = { requestOptions: {} }) {
+export function _downloadSessionFileSend(context, agentName, agentSessionId, path, options = { requestOptions: {} }) {
     const foundryFeatures = "HostedAgents=V1Preview,AgentEndpoints=V1Preview";
-    const path_1 = expandUrlTemplate("/agents/{agent_name}/endpoint/sessions/{session_id}/files/content{?path,api-version}", {
+    const path_1 = expandUrlTemplate("/agents/{agent_name}/endpoint/sessions/{agent_session_id}/files/content{?path,api-version}", {
         agent_name: agentName,
-        session_id: sessionId,
+        agent_session_id: agentSessionId,
         path: path,
         "api-version": context.apiVersion,
     }, {
@@ -91,6 +457,9 @@ export function _downloadSessionFileSend(context, agentName, sessionId, path, op
         ...operationOptionsToRequestParameters(options),
         headers: {
             "foundry-features": foundryFeatures,
+            ...(options?.userIsolationKey !== undefined
+                ? { "x-ms-user-isolation-key": options?.userIsolationKey }
+                : {}),
             accept: "application/octet-stream",
             ...options.requestOptions?.headers,
         },
@@ -106,16 +475,16 @@ export async function _downloadSessionFileDeserialize(result) {
     return { blobBody: result.blobBody, readableStreamBody: result.readableStreamBody };
 }
 /** Download a file from the session sandbox as a binary stream. */
-export async function downloadSessionFile(context, agentName, sessionId, path, options = { requestOptions: {} }) {
-    const streamableMethod = _downloadSessionFileSend(context, agentName, sessionId, path, options);
+export async function downloadSessionFile(context, agentName, agentSessionId, path, options = { requestOptions: {} }) {
+    const streamableMethod = _downloadSessionFileSend(context, agentName, agentSessionId, path, options);
     const result = await getBinaryStreamResponse(streamableMethod);
     return _downloadSessionFileDeserialize(result);
 }
-export function _uploadSessionFileSend(context, agentName, sessionId, path, content, options = { requestOptions: {} }) {
+export function _uploadSessionFileSend(context, agentName, agentSessionId, path, content, options = { requestOptions: {} }) {
     const foundryFeatures = "HostedAgents=V1Preview,AgentEndpoints=V1Preview";
-    const path_1 = expandUrlTemplate("/agents/{agent_name}/endpoint/sessions/{session_id}/files/content{?path,api-version}", {
+    const path_1 = expandUrlTemplate("/agents/{agent_name}/endpoint/sessions/{agent_session_id}/files/content{?path,api-version}", {
         agent_name: agentName,
-        session_id: sessionId,
+        agent_session_id: agentSessionId,
         path: path,
         "api-version": context.apiVersion,
     }, {
@@ -126,6 +495,9 @@ export function _uploadSessionFileSend(context, agentName, sessionId, path, cont
         contentType: "application/octet-stream",
         headers: {
             "foundry-features": foundryFeatures,
+            ...(options?.userIsolationKey !== undefined
+                ? { "x-ms-user-isolation-key": options?.userIsolationKey }
+                : {}),
             accept: "application/json",
             ...options.requestOptions?.headers,
         },
@@ -145,10 +517,69 @@ export async function _uploadSessionFileDeserialize(result) {
  * Upload a file to the session sandbox via binary stream.
  * Maximum file size is 50 MB. Uploads exceeding this limit return 413 Payload Too Large.
  */
-export async function uploadSessionFile(context, agentName, sessionId, path, content, options = { requestOptions: {} }) {
-    const result = await _uploadSessionFileSend(context, agentName, sessionId, path, content, options);
+export async function uploadSessionFile(context, agentName, agentSessionId, path, content, options = { requestOptions: {} }) {
+    const result = await _uploadSessionFileSend(context, agentName, agentSessionId, path, content, options);
     return _uploadSessionFileDeserialize(result);
 }
+export function _getSessionLogStreamSend(context, agentName, agentVersion, sessionId, options = { requestOptions: {} }) {
+    const foundryFeatures = "HostedAgents=V1Preview,AgentEndpoints=V1Preview";
+    const path = expandUrlTemplate("/agents/{agent_name}/versions/{agent_version}/sessions/{session_id}:logstream{?api-version}", {
+        agent_name: agentName,
+        agent_version: agentVersion,
+        session_id: sessionId,
+        "api-version": context.apiVersion,
+    }, {
+        allowReserved: options?.requestOptions?.skipUrlEncoding,
+    });
+    return context.path(path).get({
+        ...operationOptionsToRequestParameters(options),
+        headers: {
+            "foundry-features": foundryFeatures,
+            accept: "text/event-stream",
+            ...options.requestOptions?.headers,
+        },
+    });
+}
+export async function _getSessionLogStreamDeserialize(result) {
+    const expectedStatuses = ["200"];
+    if (!expectedStatuses.includes(result.status)) {
+        throw createRestError(result);
+    }
+    return { blobBody: result.blobBody, readableStreamBody: result.readableStreamBody };
+}
+/**
+ * Streams console logs (stdout / stderr) for a specific hosted agent session
+ * as a Server-Sent Events (SSE) stream.
+ *
+ * Each SSE frame contains:
+ * - `event`: always `"log"`
+ * - `data`: a plain-text log line (currently JSON-formatted, but the schema
+ * is not contractual and may include additional keys or change format
+ * over time — clients should treat it as an opaque string)
+ *
+ * Example SSE frames:
+ * ```
+ * event: log
+ * data: {"timestamp":"2026-03-10T09:33:17.121Z","stream":"stdout","message":"Starting FoundryCBAgent server on port 8088"}
+ *
+ * event: log
+ * data: {"timestamp":"2026-03-10T09:33:17.130Z","stream":"stderr","message":"INFO: Application startup complete."}
+ *
+ * event: log
+ * data: {"timestamp":"2026-03-10T09:34:52.714Z","stream":"status","message":"Successfully connected to container"}
+ *
+ * event: log
+ * data: {"timestamp":"2026-03-10T09:35:52.714Z","stream":"status","message":"No logs since last 60 seconds"}
+ * ```
+ *
+ * The stream remains open until the client disconnects or the server
+ * terminates the connection. Clients should handle reconnection as needed.
+ */
+export async function getSessionLogStream(context, agentName, agentVersion, sessionId, options = { requestOptions: {} }) {
+    const streamableMethod = _getSessionLogStreamSend(context, agentName, agentVersion, sessionId, options);
+    const result = await getBinaryStreamResponse(streamableMethod);
+    return _getSessionLogStreamDeserialize(result);
+}
 export function _listSessionsSend(context, agentName, options = { requestOptions: {} }) {
     const foundryFeatures = "HostedAgents=V1Preview,AgentEndpoints=V1Preview";
     const path = expandUrlTemplate("/agents/{agent_name}/endpoint/sessions{?limit,order,after,before,api-version}", {
@@ -165,6 +596,9 @@ export function _listSessionsSend(context, agentName, options = { requestOptions
         ...operationOptionsToRequestParameters(options),
         headers: {
             "foundry-features": foundryFeatures,
+            ...(options?.userIsolationKey !== undefined
+                ? { "x-ms-user-isolation-key": options?.userIsolationKey }
+                : {}),
             accept: "application/json",
             ...options.requestOptions?.headers,
         },
@@ -191,7 +625,41 @@ export function listSessions(context, agentName, options = { requestOptions: {}
         hasMoreFieldName: "has_more",
     });
 }
-export function _deleteSessionSend(context, agentName, sessionId, isolationKey, options = { requestOptions: {} }) {
+export function _stopSessionSend(context, agentName, sessionId, options = { requestOptions: {} }) {
+    const foundryFeatures = "HostedAgents=V1Preview,AgentEndpoints=V1Preview";
+    const path = expandUrlTemplate("/agents/{agent_name}/endpoint/sessions/{session_id}:stop{?api-version}", {
+        agent_name: agentName,
+        session_id: sessionId,
+        "api-version": context.apiVersion,
+    }, {
+        allowReserved: options?.requestOptions?.skipUrlEncoding,
+    });
+    return context.path(path).post({
+        ...operationOptionsToRequestParameters(options),
+        headers: {
+            "foundry-features": foundryFeatures,
+            ...options.requestOptions?.headers,
+        },
+    });
+}
+export async function _stopSessionDeserialize(result) {
+    const expectedStatuses = ["204"];
+    if (!expectedStatuses.includes(result.status)) {
+        const error = createRestError(result);
+        error.details = apiErrorResponseDeserializer(result.body);
+        throw error;
+    }
+    return;
+}
+/**
+ * Stops a session.
+ * Returns 204 No Content when the stop succeeds.
+ */
+export async function stopSession(context, agentName, sessionId, options = { requestOptions: {} }) {
+    const result = await _stopSessionSend(context, agentName, sessionId, options);
+    return _stopSessionDeserialize(result);
+}
+export function _deleteSessionSend(context, agentName, sessionId, options = { requestOptions: {} }) {
     const foundryFeatures = "HostedAgents=V1Preview,AgentEndpoints=V1Preview";
     const path = expandUrlTemplate("/agents/{agent_name}/endpoint/sessions/{session_id}{?api-version}", {
         agent_name: agentName,
@@ -204,7 +672,9 @@ export function _deleteSessionSend(context, agentName, sessionId, isolationKey,
         ...operationOptionsToRequestParameters(options),
         headers: {
             "foundry-features": foundryFeatures,
-            "x-session-isolation-key": isolationKey,
+            ...(options?.userIsolationKey !== undefined
+                ? { "x-ms-user-isolation-key": options?.userIsolationKey }
+                : {}),
             ...options.requestOptions?.headers,
         },
     });
@@ -222,8 +692,8 @@ export async function _deleteSessionDeserialize(result) {
  * Deletes a session synchronously.
  * Returns 204 No Content when the session is deleted or does not exist.
  */
-export async function deleteSession(context, agentName, sessionId, isolationKey, options = { requestOptions: {} }) {
-    const result = await _deleteSessionSend(context, agentName, sessionId, isolationKey, options);
+export async function deleteSession(context, agentName, sessionId, options = { requestOptions: {} }) {
+    const result = await _deleteSessionSend(context, agentName, sessionId, options);
     return _deleteSessionDeserialize(result);
 }
 export function _getSessionSend(context, agentName, sessionId, options = { requestOptions: {} }) {
@@ -239,6 +709,9 @@ export function _getSessionSend(context, agentName, sessionId, options = { reque
         ...operationOptionsToRequestParameters(options),
         headers: {
             "foundry-features": foundryFeatures,
+            ...(options?.userIsolationKey !== undefined
+                ? { "x-ms-user-isolation-key": options?.userIsolationKey }
+                : {}),
             accept: "application/json",
             ...options.requestOptions?.headers,
         },
@@ -258,7 +731,7 @@ export async function getSession(context, agentName, sessionId, options = { requ
     const result = await _getSessionSend(context, agentName, sessionId, options);
     return _getSessionDeserialize(result);
 }
-export function _createSessionSend(context, agentName, isolationKey, versionIndicator, options = { requestOptions: {} }) {
+export function _createSessionSend(context, agentName, versionIndicator, options = { requestOptions: {} }) {
     const foundryFeatures = "HostedAgents=V1Preview,AgentEndpoints=V1Preview";
     const path = expandUrlTemplate("/agents/{agent_name}/endpoint/sessions{?api-version}", {
         agent_name: agentName,
@@ -271,7 +744,9 @@ export function _createSessionSend(context, agentName, isolationKey, versionIndi
         contentType: "application/json",
         headers: {
             "foundry-features": foundryFeatures,
-            "x-session-isolation-key": isolationKey,
+            ...(options?.userIsolationKey !== undefined
+                ? { "x-ms-user-isolation-key": options?.userIsolationKey }
+                : {}),
             accept: "application/json",
             ...options.requestOptions?.headers,
         },
@@ -295,10 +770,87 @@ export async function _createSessionDeserialize(result) {
  * The endpoint resolves the backing agent version from `version_indicator` and
  * enforces session ownership using the provided isolation key for session-mutating operations.
  */
-export async function createSession(context, agentName, isolationKey, versionIndicator, options = { requestOptions: {} }) {
-    const result = await _createSessionSend(context, agentName, isolationKey, versionIndicator, options);
+export async function createSession(context, agentName, versionIndicator, options = { requestOptions: {} }) {
+    const result = await _createSessionSend(context, agentName, versionIndicator, options);
     return _createSessionDeserialize(result);
 }
+export function _downloadAgentCodeSend(context, agentName, options = { requestOptions: {} }) {
+    const foundryFeatures = "HostedAgents=V1Preview,CodeAgents=V1Preview";
+    const path = expandUrlTemplate("/agents/{agent_name}/code:download{?agent_version,api-version}", {
+        agent_name: agentName,
+        agent_version: options?.agentVersion,
+        "api-version": context.apiVersion,
+    }, {
+        allowReserved: options?.requestOptions?.skipUrlEncoding,
+    });
+    return context.path(path).get({
+        ...operationOptionsToRequestParameters(options),
+        headers: {
+            "foundry-features": foundryFeatures,
+            accept: "application/zip",
+            ...options.requestOptions?.headers,
+        },
+    });
+}
+export async function _downloadAgentCodeDeserialize(result) {
+    const expectedStatuses = ["200"];
+    if (!expectedStatuses.includes(result.status)) {
+        const error = createRestError(result);
+        if (result.body?.error) {
+            error.details = apiErrorResponseDeserializer(result.body);
+        }
+        throw error;
+    }
+    return { blobBody: result.blobBody, readableStreamBody: result.readableStreamBody };
+}
+/**
+ * Download the code zip for a code-based hosted agent.
+ * Returns the previously-uploaded zip (`application/zip`).
+ *
+ * If `agent_version` is supplied, returns that version's code zip; otherwise
+ * returns the latest version's code zip.
+ *
+ * The SHA-256 digest of the returned bytes matches the `content_hash` on the
+ * resolved version's `code_configuration`.
+ */
+export async function downloadAgentCode(context, agentName, options = { requestOptions: {} }) {
+    const streamableMethod = _downloadAgentCodeSend(context, agentName, options);
+    const result = await getBinaryStreamResponse(streamableMethod);
+    return _downloadAgentCodeDeserialize(result);
+}
+export function _createAgentVersionFromCodeSend(context, agentName, codeZipSha256, content, options = { requestOptions: {} }) {
+    const foundryFeatures = "HostedAgents=V1Preview,CodeAgents=V1Preview";
+    const path = expandUrlTemplate("/agents/{agent_name}/versions{?api-version}", {
+        agent_name: agentName,
+        "api-version": context.apiVersion,
+    }, {
+        allowReserved: options?.requestOptions?.skipUrlEncoding,
+    });
+    return context.path(path).post({
+        ...operationOptionsToRequestParameters(options),
+        contentType: "multipart/form-data",
+        headers: {
+            "foundry-features": foundryFeatures,
+            "x-ms-code-zip-sha256": codeZipSha256,
+            accept: "application/json",
+            ...options.requestOptions?.headers,
+        },
+        body: createAgentVersionFromCodeContentSerializer(content),
+    });
+}
+export async function _createAgentVersionFromCodeDeserialize(result) {
+    const expectedStatuses = ["200"];
+    if (!expectedStatuses.includes(result.status)) {
+        const error = createRestError(result);
+        error.details = apiErrorResponseDeserializer(result.body);
+        throw error;
+    }
+    return agentVersionDeserializer(result.body);
+}
+export async function createVersionFromCode(context, agentName, codeZipSha256, content, options = { requestOptions: {} }) {
+    const result = await _createAgentVersionFromCodeSend(context, agentName, codeZipSha256, content, options);
+    return _createAgentVersionFromCodeDeserialize(result);
+}
 export function _patchAgentObjectSend(context, agentName, options = { requestOptions: {} }) {
     const foundryFeatures = "HostedAgents=V1Preview,AgentEndpoints=V1Preview";
     const path = expandUrlTemplate("/agents/{agent_name}{?api-version}", {
@@ -318,7 +870,7 @@ export function _patchAgentObjectSend(context, agentName, options = { requestOpt
         body: {
             agent_endpoint: !options?.agentEndpoint
                 ? options?.agentEndpoint
-                : agentEndpointSerializer(options?.agentEndpoint),
+                : agentEndpointConfigSerializer(options?.agentEndpoint),
             agent_card: !options?.agentCard
                 ? options?.agentCard
                 : agentCardSerializer(options?.agentCard),
@@ -335,7 +887,7 @@ export async function _patchAgentObjectDeserialize(result) {
     return agentDeserializer(result.body);
 }
 /** Updates an agent endpoint. */
-export async function patchAgentObject(context, agentName, options = { requestOptions: {} }) {
+export async function updateAgentObject(context, agentName, options = { requestOptions: {} }) {
     const result = await _patchAgentObjectSend(context, agentName, options);
     return _patchAgentObjectDeserialize(result);
 }