llama-stack-client 0.0.35 → 0.1.2

package/resources/agents/steps.d.ts

@@ -1,34 +1,13 @@
 import { APIResource } from "../../resource.js";
 import * as Core from "../../core.js";
-import * as StepsAPI from "./steps.js";
 import * as AgentsAPI from "./agents.js";
 export declare class Steps extends APIResource {
-    retrieve(params: StepRetrieveParams, options?: Core.RequestOptions): Core.APIPromise<AgentsStep>;
+    retrieve(agentId: string, sessionId: string, turnId: string, stepId: string, options?: Core.RequestOptions): Core.APIPromise<StepRetrieveResponse>;
 }
-export interface AgentsStep {
+export interface StepRetrieveResponse {
     step: AgentsAPI.InferenceStep | AgentsAPI.ToolExecutionStep | AgentsAPI.ShieldCallStep | AgentsAPI.MemoryRetrievalStep;
 }
-export interface StepRetrieveParams {
-    /**
-     * Query param:
-     */
-    agent_id: string;
-    /**
-     * Query param:
-     */
-    step_id: string;
-    /**
-     * Query param:
-     */
-    turn_id: string;
-    /**
-     * Header param: JSON-encoded provider data which will be made available to the
-     * adapter servicing the API
-     */
-    'X-LlamaStack-ProviderData'?: string;
-}
 export declare namespace Steps {
-    export import AgentsStep = StepsAPI.AgentsStep;
-    export import StepRetrieveParams = StepsAPI.StepRetrieveParams;
+    export { type StepRetrieveResponse as StepRetrieveResponse };
 }
 //# sourceMappingURL=steps.d.ts.map

package/resources/agents/steps.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"steps.d.ts","sourceRoot":"","sources":["../../src/resources/agents/steps.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAC7C,OAAO,KAAK,IAAI,MAAM,YAAY,CAAC;AACnC,OAAO,KAAK,QAAQ,MAAM,SAAS,CAAC;AACpC,OAAO,KAAK,SAAS,MAAM,UAAU,CAAC;AAEtC,qBAAa,KAAM,SAAQ,WAAW;IACpC,QAAQ,CAAC,MAAM,EAAE,kBAAkB,EAAE,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC;CAajG;AAED,MAAM,WAAW,UAAU;IACzB,IAAI,EACA,SAAS,CAAC,aAAa,GACvB,SAAS,CAAC,iBAAiB,GAC3B,SAAS,CAAC,cAAc,GACxB,SAAS,CAAC,mBAAmB,CAAC;CACnC;AAED,MAAM,WAAW,kBAAkB;IACjC;;OAEG;IACH,QAAQ,EAAE,MAAM,CAAC;IAEjB;;OAEG;IACH,OAAO,EAAE,MAAM,CAAC;IAEhB;;OAEG;IACH,OAAO,EAAE,MAAM,CAAC;IAEhB;;;OAGG;IACH,2BAA2B,CAAC,EAAE,MAAM,CAAC;CACtC;AAED,yBAAiB,KAAK,CAAC;IACrB,MAAM,QAAQ,UAAU,GAAG,QAAQ,CAAC,UAAU,CAAC;IAC/C,MAAM,QAAQ,kBAAkB,GAAG,QAAQ,CAAC,kBAAkB,CAAC;CAChE"}
+{"version":3,"file":"steps.d.ts","sourceRoot":"","sources":["../../src/resources/agents/steps.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAC7C,OAAO,KAAK,IAAI,MAAM,YAAY,CAAC;AACnC,OAAO,KAAK,SAAS,MAAM,UAAU,CAAC;AAEtC,qBAAa,KAAM,SAAQ,WAAW;IACpC,QAAQ,CACN,OAAO,EAAE,MAAM,EACf,SAAS,EAAE,MAAM,EACjB,MAAM,EAAE,MAAM,EACd,MAAM,EAAE,MAAM,EACd,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAC5B,IAAI,CAAC,UAAU,CAAC,oBAAoB,CAAC;CAMzC;AAED,MAAM,WAAW,oBAAoB;IACnC,IAAI,EACA,SAAS,CAAC,aAAa,GACvB,SAAS,CAAC,iBAAiB,GAC3B,SAAS,CAAC,cAAc,GACxB,SAAS,CAAC,mBAAmB,CAAC;CACnC;AAED,MAAM,CAAC,OAAO,WAAW,KAAK,CAAC;IAC7B,OAAO,EAAE,KAAK,oBAAoB,IAAI,oBAAoB,EAAE,CAAC;CAC9D"}

package/resources/agents/steps.js

@@ -4,21 +4,9 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.Steps = void 0;
 const resource_1 = require("../../resource.js");
 class Steps extends resource_1.APIResource {
-    retrieve(params, options) {
-        const { 'X-LlamaStack-ProviderData': xLlamaStackProviderData, ...query } = params;
-        return this._client.get('/agents/step/get', {
-            query,
-            ...options,
-            headers: {
-                ...(xLlamaStackProviderData != null ?
-                    { 'X-LlamaStack-ProviderData': xLlamaStackProviderData }
-                    : undefined),
-                ...options?.headers,
-            },
-        });
+    retrieve(agentId, sessionId, turnId, stepId, options) {
+        return this._client.get(`/v1/agents/${agentId}/session/${sessionId}/turn/${turnId}/step/${stepId}`, options);
     }
 }
 exports.Steps = Steps;
-(function (Steps) {
-})(Steps = exports.Steps || (exports.Steps = {}));
 //# sourceMappingURL=steps.js.map

package/resources/agents/steps.js.map

@@ -1 +1 @@
-{"version":3,"file":"steps.js","sourceRoot":"","sources":["../../src/resources/agents/steps.ts"],"names":[],"mappings":";AAAA,sFAAsF;;;AAEtF,gDAA6C;AAK7C,MAAa,KAAM,SAAQ,sBAAW;IACpC,QAAQ,CAAC,MAA0B,EAAE,OAA6B;QAChE,MAAM,EAAE,2BAA2B,EAAE,uBAAuB,EAAE,GAAG,KAAK,EAAE,GAAG,MAAM,CAAC;QAClF,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,kBAAkB,EAAE;YAC1C,KAAK;YACL,GAAG,OAAO;YACV,OAAO,EAAE;gBACP,GAAG,CAAC,uBAAuB,IAAI,IAAI,CAAC,CAAC;oBACnC,EAAE,2BAA2B,EAAE,uBAAuB,EAAE;oBAC1D,CAAC,CAAC,SAAS,CAAC;gBACZ,GAAG,OAAO,EAAE,OAAO;aACpB;SACF,CAAC,CAAC;IACL,CAAC;CACF;AAdD,sBAcC;AAiCD,WAAiB,KAAK;AAGtB,CAAC,EAHgB,KAAK,GAAL,aAAK,KAAL,aAAK,QAGrB"}
+{"version":3,"file":"steps.js","sourceRoot":"","sources":["../../src/resources/agents/steps.ts"],"names":[],"mappings":";AAAA,sFAAsF;;;AAEtF,gDAA6C;AAI7C,MAAa,KAAM,SAAQ,sBAAW;IACpC,QAAQ,CACN,OAAe,EACf,SAAiB,EACjB,MAAc,EACd,MAAc,EACd,OAA6B;QAE7B,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CACrB,cAAc,OAAO,YAAY,SAAS,SAAS,MAAM,SAAS,MAAM,EAAE,EAC1E,OAAO,CACR,CAAC;IACJ,CAAC;CACF;AAbD,sBAaC"}

package/resources/agents/steps.mjs

@@ -1,20 +1,8 @@
 // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 import { APIResource } from "../../resource.mjs";
 export class Steps extends APIResource {
-    retrieve(params, options) {
-        const { 'X-LlamaStack-ProviderData': xLlamaStackProviderData, ...query } = params;
-        return this._client.get('/agents/step/get', {
-            query,
-            ...options,
-            headers: {
-                ...(xLlamaStackProviderData != null ?
-                    { 'X-LlamaStack-ProviderData': xLlamaStackProviderData }
-                    : undefined),
-                ...options?.headers,
-            },
-        });
+    retrieve(agentId, sessionId, turnId, stepId, options) {
+        return this._client.get(`/v1/agents/${agentId}/session/${sessionId}/turn/${turnId}/step/${stepId}`, options);
     }
 }
-(function (Steps) {
-})(Steps || (Steps = {}));
 //# sourceMappingURL=steps.mjs.map

package/resources/agents/steps.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"steps.mjs","sourceRoot":"","sources":["../../src/resources/agents/steps.ts"],"names":[],"mappings":"AAAA,sFAAsF;OAE/E,EAAE,WAAW,EAAE;AAKtB,MAAM,OAAO,KAAM,SAAQ,WAAW;IACpC,QAAQ,CAAC,MAA0B,EAAE,OAA6B;QAChE,MAAM,EAAE,2BAA2B,EAAE,uBAAuB,EAAE,GAAG,KAAK,EAAE,GAAG,MAAM,CAAC;QAClF,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,kBAAkB,EAAE;YAC1C,KAAK;YACL,GAAG,OAAO;YACV,OAAO,EAAE;gBACP,GAAG,CAAC,uBAAuB,IAAI,IAAI,CAAC,CAAC;oBACnC,EAAE,2BAA2B,EAAE,uBAAuB,EAAE;oBAC1D,CAAC,CAAC,SAAS,CAAC;gBACZ,GAAG,OAAO,EAAE,OAAO;aACpB;SACF,CAAC,CAAC;IACL,CAAC;CACF;AAiCD,WAAiB,KAAK;AAGtB,CAAC,EAHgB,KAAK,KAAL,KAAK,QAGrB"}
+{"version":3,"file":"steps.mjs","sourceRoot":"","sources":["../../src/resources/agents/steps.ts"],"names":[],"mappings":"AAAA,sFAAsF;OAE/E,EAAE,WAAW,EAAE;AAItB,MAAM,OAAO,KAAM,SAAQ,WAAW;IACpC,QAAQ,CACN,OAAe,EACf,SAAiB,EACjB,MAAc,EACd,MAAc,EACd,OAA6B;QAE7B,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CACrB,cAAc,OAAO,YAAY,SAAS,SAAS,MAAM,SAAS,MAAM,EAAE,EAC1E,OAAO,CACR,CAAC;IACJ,CAAC;CACF"}

package/resources/agents/turn.d.ts

@@ -0,0 +1,199 @@
+import { APIResource } from "../../resource.js";
+import { APIPromise } from "../../core.js";
+import * as Core from "../../core.js";
+import * as TurnAPI from "./turn.js";
+import * as Shared from "../shared.js";
+import * as AgentsAPI from "./agents.js";
+import { Stream } from "../../streaming.js";
+export declare class TurnResource extends APIResource {
+    create(agentId: string, sessionId: string, body: TurnCreateParamsNonStreaming, options?: Core.RequestOptions): APIPromise<Turn>;
+    create(agentId: string, sessionId: string, body: TurnCreateParamsStreaming, options?: Core.RequestOptions): APIPromise<Stream<AgentTurnResponseStreamChunk>>;
+    create(agentId: string, sessionId: string, body: TurnCreateParamsBase, options?: Core.RequestOptions): APIPromise<Stream<AgentTurnResponseStreamChunk> | Turn>;
+    retrieve(agentId: string, sessionId: string, turnId: string, options?: Core.RequestOptions): Core.APIPromise<Turn>;
+}
+/**
+ * streamed agent turn completion response.
+ */
+export interface AgentTurnResponseStreamChunk {
+    event: TurnResponseEvent;
+}
+/**
+ * A single turn in an interaction with an Agentic System.
+ */
+export interface Turn {
+    input_messages: Array<Shared.UserMessage | Shared.ToolResponseMessage>;
+    output_attachments: Array<Turn.OutputAttachment>;
+    /**
+     * A message containing the model's (assistant) response in a chat conversation.
+     */
+    output_message: Shared.CompletionMessage;
+    session_id: string;
+    started_at: string;
+    steps: Array<AgentsAPI.InferenceStep | AgentsAPI.ToolExecutionStep | AgentsAPI.ShieldCallStep | AgentsAPI.MemoryRetrievalStep>;
+    turn_id: string;
+    completed_at?: string;
+}
+export declare namespace Turn {
+    interface OutputAttachment {
+        /**
+         * A image content item
+         */
+        content: string | OutputAttachment.ImageContentItem | OutputAttachment.TextContentItem | Array<Shared.InterleavedContentItem> | Shared.URL;
+        mime_type: string;
+    }
+    namespace OutputAttachment {
+        /**
+         * A image content item
+         */
+        interface ImageContentItem {
+            /**
+             * Image as a base64 encoded string or an URL
+             */
+            image: ImageContentItem.Image;
+            /**
+             * Discriminator type of the content item. Always "image"
+             */
+            type: 'image';
+        }
+        namespace ImageContentItem {
+            /**
+             * Image as a base64 encoded string or an URL
+             */
+            interface Image {
+                /**
+                 * base64 encoded image data as string
+                 */
+                data?: string;
+                /**
+                 * A URL of the image or data URL in the format of data:image/{type};base64,{data}.
+                 * Note that URL could have length limits.
+                 */
+                url?: Shared.URL;
+            }
+        }
+        /**
+         * A text content item
+         */
+        interface TextContentItem {
+            /**
+             * Text content
+             */
+            text: string;
+            /**
+             * Discriminator type of the content item. Always "text"
+             */
+            type: 'text';
+        }
+    }
+}
+export interface TurnResponseEvent {
+    payload: TurnResponseEventPayload;
+}
+export type TurnResponseEventPayload = TurnResponseEventPayload.AgentTurnResponseStepStartPayload | TurnResponseEventPayload.AgentTurnResponseStepProgressPayload | TurnResponseEventPayload.AgentTurnResponseStepCompletePayload | TurnResponseEventPayload.AgentTurnResponseTurnStartPayload | TurnResponseEventPayload.AgentTurnResponseTurnCompletePayload;
+export declare namespace TurnResponseEventPayload {
+    interface AgentTurnResponseStepStartPayload {
+        event_type: 'step_start';
+        step_id: string;
+        step_type: 'inference' | 'tool_execution' | 'shield_call' | 'memory_retrieval';
+        metadata?: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+    }
+    interface AgentTurnResponseStepProgressPayload {
+        delta: Shared.ContentDelta;
+        event_type: 'step_progress';
+        step_id: string;
+        step_type: 'inference' | 'tool_execution' | 'shield_call' | 'memory_retrieval';
+    }
+    interface AgentTurnResponseStepCompletePayload {
+        event_type: 'step_complete';
+        step_details: AgentsAPI.InferenceStep | AgentsAPI.ToolExecutionStep | AgentsAPI.ShieldCallStep | AgentsAPI.MemoryRetrievalStep;
+        step_id: string;
+        step_type: 'inference' | 'tool_execution' | 'shield_call' | 'memory_retrieval';
+    }
+    interface AgentTurnResponseTurnStartPayload {
+        event_type: 'turn_start';
+        turn_id: string;
+    }
+    interface AgentTurnResponseTurnCompletePayload {
+        event_type: 'turn_complete';
+        /**
+         * A single turn in an interaction with an Agentic System.
+         */
+        turn: TurnAPI.Turn;
+    }
+}
+export type TurnCreateParams = TurnCreateParamsNonStreaming | TurnCreateParamsStreaming;
+export interface TurnCreateParamsBase {
+    messages: Array<Shared.UserMessage | Shared.ToolResponseMessage>;
+    documents?: Array<TurnCreateParams.Document>;
+    stream?: boolean;
+    toolgroups?: Array<string | TurnCreateParams.UnionMember1>;
+}
+export declare namespace TurnCreateParams {
+    interface Document {
+        /**
+         * A image content item
+         */
+        content: string | Document.ImageContentItem | Document.TextContentItem | Array<Shared.InterleavedContentItem> | Shared.URL;
+        mime_type: string;
+    }
+    namespace Document {
+        /**
+         * A image content item
+         */
+        interface ImageContentItem {
+            /**
+             * Image as a base64 encoded string or an URL
+             */
+            image: ImageContentItem.Image;
+            /**
+             * Discriminator type of the content item. Always "image"
+             */
+            type: 'image';
+        }
+        namespace ImageContentItem {
+            /**
+             * Image as a base64 encoded string or an URL
+             */
+            interface Image {
+                /**
+                 * base64 encoded image data as string
+                 */
+                data?: string;
+                /**
+                 * A URL of the image or data URL in the format of data:image/{type};base64,{data}.
+                 * Note that URL could have length limits.
+                 */
+                url?: Shared.URL;
+            }
+        }
+        /**
+         * A text content item
+         */
+        interface TextContentItem {
+            /**
+             * Text content
+             */
+            text: string;
+            /**
+             * Discriminator type of the content item. Always "text"
+             */
+            type: 'text';
+        }
+    }
+    interface UnionMember1 {
+        args: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+        name: string;
+    }
+    type TurnCreateParamsNonStreaming = TurnAPI.TurnCreateParamsNonStreaming;
+    type TurnCreateParamsStreaming = TurnAPI.TurnCreateParamsStreaming;
+}
+export interface TurnCreateParamsNonStreaming extends TurnCreateParamsBase {
+    stream?: false;
+}
+export interface TurnCreateParamsStreaming extends TurnCreateParamsBase {
+    stream: true;
+}
+export declare namespace TurnResource {
+    export { type AgentTurnResponseStreamChunk as AgentTurnResponseStreamChunk, type Turn as Turn, type TurnResponseEvent as TurnResponseEvent, type TurnResponseEventPayload as TurnResponseEventPayload, type TurnCreateParams as TurnCreateParams, type TurnCreateParamsNonStreaming as TurnCreateParamsNonStreaming, type TurnCreateParamsStreaming as TurnCreateParamsStreaming, };
+}
+//# sourceMappingURL=turn.d.ts.map

package/resources/agents/turn.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"turn.d.ts","sourceRoot":"","sources":["../../src/resources/agents/turn.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAC7C,OAAO,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AACxC,OAAO,KAAK,IAAI,MAAM,YAAY,CAAC;AACnC,OAAO,KAAK,OAAO,MAAM,QAAQ,CAAC;AAClC,OAAO,KAAK,MAAM,MAAM,WAAW,CAAC;AACpC,OAAO,KAAK,SAAS,MAAM,UAAU,CAAC;AACtC,OAAO,EAAE,MAAM,EAAE,MAAM,iBAAiB,CAAC;AAEzC,qBAAa,YAAa,SAAQ,WAAW;IAC3C,MAAM,CACJ,OAAO,EAAE,MAAM,EACf,SAAS,EAAE,MAAM,EACjB,IAAI,EAAE,4BAA4B,EAClC,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAC5B,UAAU,CAAC,IAAI,CAAC;IACnB,MAAM,CACJ,OAAO,EAAE,MAAM,EACf,SAAS,EAAE,MAAM,EACjB,IAAI,EAAE,yBAAyB,EAC/B,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAC5B,UAAU,CAAC,MAAM,CAAC,4BAA4B,CAAC,CAAC;IACnD,MAAM,CACJ,OAAO,EAAE,MAAM,EACf,SAAS,EAAE,MAAM,EACjB,IAAI,EAAE,oBAAoB,EAC1B,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAC5B,UAAU,CAAC,MAAM,CAAC,4BAA4B,CAAC,GAAG,IAAI,CAAC;IAc1D,QAAQ,CACN,OAAO,EAAE,MAAM,EACf,SAAS,EAAE,MAAM,EACjB,MAAM,EAAE,MAAM,EACd,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAC5B,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC;CAGzB;AAED;;GAEG;AACH,MAAM,WAAW,4BAA4B;IAC3C,KAAK,EAAE,iBAAiB,CAAC;CAC1B;AAED;;GAEG;AACH,MAAM,WAAW,IAAI;IACnB,cAAc,EAAE,KAAK,CAAC,MAAM,CAAC,WAAW,GAAG,MAAM,CAAC,mBAAmB,CAAC,CAAC;IAEvE,kBAAkB,EAAE,KAAK,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;IAEjD;;OAEG;IACH,cAAc,EAAE,MAAM,CAAC,iBAAiB,CAAC;IAEzC,UAAU,EAAE,MAAM,CAAC;IAEnB,UAAU,EAAE,MAAM,CAAC;IAEnB,KAAK,EAAE,KAAK,CACR,SAAS,CAAC,aAAa,GACvB,SAAS,CAAC,iBAAiB,GAC3B,SAAS,CAAC,cAAc,GACxB,SAAS,CAAC,mBAAmB,CAChC,CAAC;IAEF,OAAO,EAAE,MAAM,CAAC;IAEhB,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED,yBAAiB,IAAI,CAAC;IACpB,UAAiB,gBAAgB;QAC/B;;WAEG;QACH,OAAO,EACH,MAAM,GACN,gBAAgB,CAAC,gBAAgB,GACjC,gBAAgB,CAAC,eAAe,GAChC,KAAK,CAAC,MAAM,CAAC,sBAAsB,CAAC,GACpC,MAAM,CAAC,GAAG,CAAC;QAEf,SAAS,EAAE,MAAM,CAAC;KACnB;IAED,UAAiB,gBAAgB,CAAC;QAChC;;WAEG;QACH,UAAiB,gBAAgB;YAC/B;;eAEG;YACH,KAAK,EAAE,gBAAgB,CAAC,KAAK,CAAC;YAE9B;;eAEG;YACH,IAAI,EAAE,OAAO,CAAC;SACf;QAED,UAAiB,gBAAgB,CAAC;YAChC;;eAEG;YACH,UAAiB,KAAK;gBACpB;;mBAEG;gBACH,IAAI,CAAC,EAAE,MAAM,CAAC;gBAEd;;;mBAGG;gBACH,GAAG,CAAC,EAAE,MAAM,CAAC,GAAG,CAAC;aAClB;SACF;QAED;;WAEG;QACH,UAAiB,eAAe;YAC9B;;eAEG;YACH,IAAI,EAAE,MAAM,CAAC;YAEb;;eAEG;YACH,IAAI,EAAE,MAAM,CAAC;SACd;KACF;CACF;AAED,MAAM,WAAW,iBAAiB;IAChC,OAAO,EAAE,wBAAwB,CAAC;CACnC;AAED,MAAM,MAAM,wBAAwB,GAChC,wBAAwB,CAAC,iCAAiC,GAC1D,wBAAwB,CAAC,oCAAoC,GAC7D,wBAAwB,CAAC,oCAAoC,GAC7D,wBAAwB,CAAC,iCAAiC,GAC1D,wBAAwB,CAAC,oCAAoC,CAAC;AAElE,yBAAiB,wBAAwB,CAAC;IACxC,UAAiB,iCAAiC;QAChD,UAAU,EAAE,YAAY,CAAC;QAEzB,OAAO,EAAE,MAAM,CAAC;QAEhB,SAAS,EAAE,WAAW,GAAG,gBAAgB,GAAG,aAAa,GAAG,kBAAkB,CAAC;QAE/E,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,GAAG,MAAM,GAAG,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,OAAO,GAAG,IAAI,CAAC,CAAC;KACxF;IAED,UAAiB,oCAAoC;QACnD,KAAK,EAAE,MAAM,CAAC,YAAY,CAAC;QAE3B,UAAU,EAAE,eAAe,CAAC;QAE5B,OAAO,EAAE,MAAM,CAAC;QAEhB,SAAS,EAAE,WAAW,GAAG,gBAAgB,GAAG,aAAa,GAAG,kBAAkB,CAAC;KAChF;IAED,UAAiB,oCAAoC;QACnD,UAAU,EAAE,eAAe,CAAC;QAE5B,YAAY,EACR,SAAS,CAAC,aAAa,GACvB,SAAS,CAAC,iBAAiB,GAC3B,SAAS,CAAC,cAAc,GACxB,SAAS,CAAC,mBAAmB,CAAC;QAElC,OAAO,EAAE,MAAM,CAAC;QAEhB,SAAS,EAAE,WAAW,GAAG,gBAAgB,GAAG,aAAa,GAAG,kBAAkB,CAAC;KAChF;IAED,UAAiB,iCAAiC;QAChD,UAAU,EAAE,YAAY,CAAC;QAEzB,OAAO,EAAE,MAAM,CAAC;KACjB;IAED,UAAiB,oCAAoC;QACnD,UAAU,EAAE,eAAe,CAAC;QAE5B;;WAEG;QACH,IAAI,EAAE,OAAO,CAAC,IAAI,CAAC;KACpB;CACF;AAED,MAAM,MAAM,gBAAgB,GAAG,4BAA4B,GAAG,yBAAyB,CAAC;AAExF,MAAM,WAAW,oBAAoB;IACnC,QAAQ,EAAE,KAAK,CAAC,MAAM,CAAC,WAAW,GAAG,MAAM,CAAC,mBAAmB,CAAC,CAAC;IAEjE,SAAS,CAAC,EAAE,KAAK,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;IAE7C,MAAM,CAAC,EAAE,OAAO,CAAC;IAEjB,UAAU,CAAC,EAAE,KAAK,CAAC,MAAM,GAAG,gBAAgB,CAAC,YAAY,CAAC,CAAC;CAC5D;AAED,yBAAiB,gBAAgB,CAAC;IAChC,UAAiB,QAAQ;QACvB;;WAEG;QACH,OAAO,EACH,MAAM,GACN,QAAQ,CAAC,gBAAgB,GACzB,QAAQ,CAAC,eAAe,GACxB,KAAK,CAAC,MAAM,CAAC,sBAAsB,CAAC,GACpC,MAAM,CAAC,GAAG,CAAC;QAEf,SAAS,EAAE,MAAM,CAAC;KACnB;IAED,UAAiB,QAAQ,CAAC;QACxB;;WAEG;QACH,UAAiB,gBAAgB;YAC/B;;eAEG;YACH,KAAK,EAAE,gBAAgB,CAAC,KAAK,CAAC;YAE9B;;eAEG;YACH,IAAI,EAAE,OAAO,CAAC;SACf;QAED,UAAiB,gBAAgB,CAAC;YAChC;;eAEG;YACH,UAAiB,KAAK;gBACpB;;mBAEG;gBACH,IAAI,CAAC,EAAE,MAAM,CAAC;gBAEd;;;mBAGG;gBACH,GAAG,CAAC,EAAE,MAAM,CAAC,GAAG,CAAC;aAClB;SACF;QAED;;WAEG;QACH,UAAiB,eAAe;YAC9B;;eAEG;YACH,IAAI,EAAE,MAAM,CAAC;YAEb;;eAEG;YACH,IAAI,EAAE,MAAM,CAAC;SACd;KACF;IAED,UAAiB,YAAY;QAC3B,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,GAAG,MAAM,GAAG,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,OAAO,GAAG,IAAI,CAAC,CAAC;QAElF,IAAI,EAAE,MAAM,CAAC;KACd;IAED,KAAY,4BAA4B,GAAG,OAAO,CAAC,4BAA4B,CAAC;IAChF,KAAY,yBAAyB,GAAG,OAAO,CAAC,yBAAyB,CAAC;CAC3E;AAED,MAAM,WAAW,4BAA6B,SAAQ,oBAAoB;IACxE,MAAM,CAAC,EAAE,KAAK,CAAC;CAChB;AAED,MAAM,WAAW,yBAA0B,SAAQ,oBAAoB;IACrE,MAAM,EAAE,IAAI,CAAC;CACd;AAED,MAAM,CAAC,OAAO,WAAW,YAAY,CAAC;IACpC,OAAO,EACL,KAAK,4BAA4B,IAAI,4BAA4B,EACjE,KAAK,IAAI,IAAI,IAAI,EACjB,KAAK,iBAAiB,IAAI,iBAAiB,EAC3C,KAAK,wBAAwB,IAAI,wBAAwB,EACzD,KAAK,gBAAgB,IAAI,gBAAgB,EACzC,KAAK,4BAA4B,IAAI,4BAA4B,EACjE,KAAK,yBAAyB,IAAI,yBAAyB,GAC5D,CAAC;CACH"}

package/resources/agents/turn.js

@@ -0,0 +1,19 @@
+"use strict";
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.TurnResource = void 0;
+const resource_1 = require("../../resource.js");
+class TurnResource extends resource_1.APIResource {
+    create(agentId, sessionId, body, options) {
+        return this._client.post(`/v1/agents/${agentId}/session/${sessionId}/turn`, {
+            body,
+            ...options,
+            stream: body.stream ?? false,
+        });
+    }
+    retrieve(agentId, sessionId, turnId, options) {
+        return this._client.get(`/v1/agents/${agentId}/session/${sessionId}/turn/${turnId}`, options);
+    }
+}
+exports.TurnResource = TurnResource;
+//# sourceMappingURL=turn.js.map

package/resources/agents/turn.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"turn.js","sourceRoot":"","sources":["../../src/resources/agents/turn.ts"],"names":[],"mappings":";AAAA,sFAAsF;;;AAEtF,gDAA6C;AAQ7C,MAAa,YAAa,SAAQ,sBAAW;IAmB3C,MAAM,CACJ,OAAe,EACf,SAAiB,EACjB,IAAsB,EACtB,OAA6B;QAE7B,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,cAAc,OAAO,YAAY,SAAS,OAAO,EAAE;YAC1E,IAAI;YACJ,GAAG,OAAO;YACV,MAAM,EAAE,IAAI,CAAC,MAAM,IAAI,KAAK;SAC7B,CAAwE,CAAC;IAC5E,CAAC;IAED,QAAQ,CACN,OAAe,EACf,SAAiB,EACjB,MAAc,EACd,OAA6B;QAE7B,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,OAAO,YAAY,SAAS,SAAS,MAAM,EAAE,EAAE,OAAO,CAAC,CAAC;IAChG,CAAC;CACF;AAxCD,oCAwCC"}

package/resources/agents/turn.mjs

@@ -0,0 +1,15 @@
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+import { APIResource } from "../../resource.mjs";
+export class TurnResource extends APIResource {
+    create(agentId, sessionId, body, options) {
+        return this._client.post(`/v1/agents/${agentId}/session/${sessionId}/turn`, {
+            body,
+            ...options,
+            stream: body.stream ?? false,
+        });
+    }
+    retrieve(agentId, sessionId, turnId, options) {
+        return this._client.get(`/v1/agents/${agentId}/session/${sessionId}/turn/${turnId}`, options);
+    }
+}
+//# sourceMappingURL=turn.mjs.map

package/resources/agents/turn.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"turn.mjs","sourceRoot":"","sources":["../../src/resources/agents/turn.ts"],"names":[],"mappings":"AAAA,sFAAsF;OAE/E,EAAE,WAAW,EAAE;AAQtB,MAAM,OAAO,YAAa,SAAQ,WAAW;IAmB3C,MAAM,CACJ,OAAe,EACf,SAAiB,EACjB,IAAsB,EACtB,OAA6B;QAE7B,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,cAAc,OAAO,YAAY,SAAS,OAAO,EAAE;YAC1E,IAAI;YACJ,GAAG,OAAO;YACV,MAAM,EAAE,IAAI,CAAC,MAAM,IAAI,KAAK;SAC7B,CAAwE,CAAC;IAC5E,CAAC;IAED,QAAQ,CACN,OAAe,EACf,SAAiB,EACjB,MAAc,EACd,OAA6B;QAE7B,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,OAAO,YAAY,SAAS,SAAS,MAAM,EAAE,EAAE,OAAO,CAAC,CAAC;IAChG,CAAC;CACF"}

package/resources/batch-inference.d.ts

@@ -1,99 +1,66 @@
 import { APIResource } from "../resource.js";
 import * as Core from "../core.js";
-import * as BatchInferenceAPI from "./batch-inference.js";
 import * as Shared from "./shared.js";
-import * as AgentsAPI from "./agents/agents.js";
 export declare class BatchInference extends APIResource {
-    chatCompletion(params: BatchInferenceChatCompletionParams, options?: Core.RequestOptions): Core.APIPromise<BatchChatCompletion>;
-    completion(params: BatchInferenceCompletionParams, options?: Core.RequestOptions): Core.APIPromise<Shared.BatchCompletion>;
+    chatCompletion(body: BatchInferenceChatCompletionParams, options?: Core.RequestOptions): Core.APIPromise<BatchInferenceChatCompletionResponse>;
+    completion(body: BatchInferenceCompletionParams, options?: Core.RequestOptions): Core.APIPromise<Shared.BatchCompletion>;
 }
-export interface BatchChatCompletion {
-    completion_message_batch: Array<Shared.CompletionMessage>;
+export interface BatchInferenceChatCompletionResponse {
+    batch: Array<Shared.ChatCompletionResponse>;
 }
 export interface BatchInferenceChatCompletionParams {
-    /**
-     * Body param:
-     */
-    messages_batch: Array<Array<Shared.UserMessage | Shared.SystemMessage | Shared.ToolResponseMessage | Shared.CompletionMessage>>;
-    /**
-     * Body param:
-     */
+    messages_batch: Array<Array<Shared.Message>>;
     model: string;
-    /**
-     * Body param:
-     */
     logprobs?: BatchInferenceChatCompletionParams.Logprobs;
     /**
-     * Body param:
+     * Configuration for JSON schema-guided response generation.
      */
+    response_format?: Shared.ResponseFormat;
     sampling_params?: Shared.SamplingParams;
     /**
-     * Body param:
+     * Whether tool use is required or automatic. This is a hint to the model which may
+     * not be followed. It depends on the Instruction Following capabilities of the
+     * model.
      */
     tool_choice?: 'auto' | 'required';
     /**
-     * Body param: `json` -- Refers to the json format for calling tools. The json
-     * format takes the form like { "type": "function", "function" : { "name":
-     * "function_name", "description": "function_description", "parameters": {...} } }
-     *
-     * `function_tag` -- This is an example of how you could define your own user
-     * defined format for making tool calls. The function_tag format looks like this,
-     * <function=function_name>(parameters)</function>
-     *
-     * The detailed prompts for each of these formats are added to llama cli
-     */
-    tool_prompt_format?: 'json' | 'function_tag';
-    /**
-     * Body param:
+     * Prompt format for calling custom / zero shot tools.
      */
+    tool_prompt_format?: 'json' | 'function_tag' | 'python_list';
     tools?: Array<BatchInferenceChatCompletionParams.Tool>;
-    /**
-     * Header param: JSON-encoded provider data which will be made available to the
-     * adapter servicing the API
-     */
-    'X-LlamaStack-ProviderData'?: string;
 }
 export declare namespace BatchInferenceChatCompletionParams {
     interface Logprobs {
+        /**
+         * How many tokens (for each position) to return log probabilities for.
+         */
         top_k?: number;
     }
     interface Tool {
         tool_name: 'brave_search' | 'wolfram_alpha' | 'photogen' | 'code_interpreter' | (string & {});
         description?: string;
-        parameters?: Record<string, AgentsAPI.ToolParamDefinition>;
+        parameters?: Record<string, Shared.ToolParamDefinition>;
     }
 }
 export interface BatchInferenceCompletionParams {
-    /**
-     * Body param:
-     */
-    content_batch: Array<string | Array<string>>;
-    /**
-     * Body param:
-     */
+    content_batch: Array<Shared.InterleavedContent>;
     model: string;
-    /**
-     * Body param:
-     */
     logprobs?: BatchInferenceCompletionParams.Logprobs;
     /**
-     * Body param:
+     * Configuration for JSON schema-guided response generation.
      */
+    response_format?: Shared.ResponseFormat;
     sampling_params?: Shared.SamplingParams;
-    /**
-     * Header param: JSON-encoded provider data which will be made available to the
-     * adapter servicing the API
-     */
-    'X-LlamaStack-ProviderData'?: string;
 }
 export declare namespace BatchInferenceCompletionParams {
     interface Logprobs {
+        /**
+         * How many tokens (for each position) to return log probabilities for.
+         */
         top_k?: number;
     }
 }
 export declare namespace BatchInference {
-    export import BatchChatCompletion = BatchInferenceAPI.BatchChatCompletion;
-    export import BatchInferenceChatCompletionParams = BatchInferenceAPI.BatchInferenceChatCompletionParams;
-    export import BatchInferenceCompletionParams = BatchInferenceAPI.BatchInferenceCompletionParams;
+    export { type BatchInferenceChatCompletionResponse as BatchInferenceChatCompletionResponse, type BatchInferenceChatCompletionParams as BatchInferenceChatCompletionParams, type BatchInferenceCompletionParams as BatchInferenceCompletionParams, };
 }
 //# sourceMappingURL=batch-inference.d.ts.map

package/resources/batch-inference.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"batch-inference.d.ts","sourceRoot":"","sources":["../src/resources/batch-inference.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAC1C,OAAO,KAAK,IAAI,MAAM,SAAS,CAAC;AAChC,OAAO,KAAK,iBAAiB,MAAM,mBAAmB,CAAC;AACvD,OAAO,KAAK,MAAM,MAAM,UAAU,CAAC;AACnC,OAAO,KAAK,SAAS,MAAM,iBAAiB,CAAC;AAE7C,qBAAa,cAAe,SAAQ,WAAW;IAC7C,cAAc,CACZ,MAAM,EAAE,kCAAkC,EAC1C,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAC5B,IAAI,CAAC,UAAU,CAAC,mBAAmB,CAAC;IAcvC,UAAU,CACR,MAAM,EAAE,8BAA8B,EACtC,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAC5B,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,eAAe,CAAC;CAa3C;AAED,MAAM,WAAW,mBAAmB;IAClC,wBAAwB,EAAE,KAAK,CAAC,MAAM,CAAC,iBAAiB,CAAC,CAAC;CAC3D;AAED,MAAM,WAAW,kCAAkC;IACjD;;OAEG;IACH,cAAc,EAAE,KAAK,CACnB,KAAK,CAAC,MAAM,CAAC,WAAW,GAAG,MAAM,CAAC,aAAa,GAAG,MAAM,CAAC,mBAAmB,GAAG,MAAM,CAAC,iBAAiB,CAAC,CACzG,CAAC;IAEF;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IAEd;;OAEG;IACH,QAAQ,CAAC,EAAE,kCAAkC,CAAC,QAAQ,CAAC;IAEvD;;OAEG;IACH,eAAe,CAAC,EAAE,MAAM,CAAC,cAAc,CAAC;IAExC;;OAEG;IACH,WAAW,CAAC,EAAE,MAAM,GAAG,UAAU,CAAC;IAElC;;;;;;;;;;OAUG;IACH,kBAAkB,CAAC,EAAE,MAAM,GAAG,cAAc,CAAC;IAE7C;;OAEG;IACH,KAAK,CAAC,EAAE,KAAK,CAAC,kCAAkC,CAAC,IAAI,CAAC,CAAC;IAEvD;;;OAGG;IACH,2BAA2B,CAAC,EAAE,MAAM,CAAC;CACtC;AAED,yBAAiB,kCAAkC,CAAC;IAClD,UAAiB,QAAQ;QACvB,KAAK,CAAC,EAAE,MAAM,CAAC;KAChB;IAED,UAAiB,IAAI;QACnB,SAAS,EAAE,cAAc,GAAG,eAAe,GAAG,UAAU,GAAG,kBAAkB,GAAG,CAAC,MAAM,GAAG,EAAE,CAAC,CAAC;QAE9F,WAAW,CAAC,EAAE,MAAM,CAAC;QAErB,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,mBAAmB,CAAC,CAAC;KAC5D;CACF;AAED,MAAM,WAAW,8BAA8B;IAC7C;;OAEG;IACH,aAAa,EAAE,KAAK,CAAC,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;IAE7C;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IAEd;;OAEG;IACH,QAAQ,CAAC,EAAE,8BAA8B,CAAC,QAAQ,CAAC;IAEnD;;OAEG;IACH,eAAe,CAAC,EAAE,MAAM,CAAC,cAAc,CAAC;IAExC;;;OAGG;IACH,2BAA2B,CAAC,EAAE,MAAM,CAAC;CACtC;AAED,yBAAiB,8BAA8B,CAAC;IAC9C,UAAiB,QAAQ;QACvB,KAAK,CAAC,EAAE,MAAM,CAAC;KAChB;CACF;AAED,yBAAiB,cAAc,CAAC;IAC9B,MAAM,QAAQ,mBAAmB,GAAG,iBAAiB,CAAC,mBAAmB,CAAC;IAC1E,MAAM,QAAQ,kCAAkC,GAAG,iBAAiB,CAAC,kCAAkC,CAAC;IACxG,MAAM,QAAQ,8BAA8B,GAAG,iBAAiB,CAAC,8BAA8B,CAAC;CACjG"}
+{"version":3,"file":"batch-inference.d.ts","sourceRoot":"","sources":["../src/resources/batch-inference.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAC1C,OAAO,KAAK,IAAI,MAAM,SAAS,CAAC;AAChC,OAAO,KAAK,MAAM,MAAM,UAAU,CAAC;AAEnC,qBAAa,cAAe,SAAQ,WAAW;IAC7C,cAAc,CACZ,IAAI,EAAE,kCAAkC,EACxC,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAC5B,IAAI,CAAC,UAAU,CAAC,oCAAoC,CAAC;IAIxD,UAAU,CACR,IAAI,EAAE,8BAA8B,EACpC,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAC5B,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,eAAe,CAAC;CAG3C;AAED,MAAM,WAAW,oCAAoC;IACnD,KAAK,EAAE,KAAK,CAAC,MAAM,CAAC,sBAAsB,CAAC,CAAC;CAC7C;AAED,MAAM,WAAW,kCAAkC;IACjD,cAAc,EAAE,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC;IAE7C,KAAK,EAAE,MAAM,CAAC;IAEd,QAAQ,CAAC,EAAE,kCAAkC,CAAC,QAAQ,CAAC;IAEvD;;OAEG;IACH,eAAe,CAAC,EAAE,MAAM,CAAC,cAAc,CAAC;IAExC,eAAe,CAAC,EAAE,MAAM,CAAC,cAAc,CAAC;IAExC;;;;OAIG;IACH,WAAW,CAAC,EAAE,MAAM,GAAG,UAAU,CAAC;IAElC;;OAEG;IACH,kBAAkB,CAAC,EAAE,MAAM,GAAG,cAAc,GAAG,aAAa,CAAC;IAE7D,KAAK,CAAC,EAAE,KAAK,CAAC,kCAAkC,CAAC,IAAI,CAAC,CAAC;CACxD;AAED,yBAAiB,kCAAkC,CAAC;IAClD,UAAiB,QAAQ;QACvB;;WAEG;QACH,KAAK,CAAC,EAAE,MAAM,CAAC;KAChB;IAED,UAAiB,IAAI;QACnB,SAAS,EAAE,cAAc,GAAG,eAAe,GAAG,UAAU,GAAG,kBAAkB,GAAG,CAAC,MAAM,GAAG,EAAE,CAAC,CAAC;QAE9F,WAAW,CAAC,EAAE,MAAM,CAAC;QAErB,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,mBAAmB,CAAC,CAAC;KACzD;CACF;AAED,MAAM,WAAW,8BAA8B;IAC7C,aAAa,EAAE,KAAK,CAAC,MAAM,CAAC,kBAAkB,CAAC,CAAC;IAEhD,KAAK,EAAE,MAAM,CAAC;IAEd,QAAQ,CAAC,EAAE,8BAA8B,CAAC,QAAQ,CAAC;IAEnD;;OAEG;IACH,eAAe,CAAC,EAAE,MAAM,CAAC,cAAc,CAAC;IAExC,eAAe,CAAC,EAAE,MAAM,CAAC,cAAc,CAAC;CACzC;AAED,yBAAiB,8BAA8B,CAAC;IAC9C,UAAiB,QAAQ;QACvB;;WAEG;QACH,KAAK,CAAC,EAAE,MAAM,CAAC;KAChB;CACF;AAED,MAAM,CAAC,OAAO,WAAW,cAAc,CAAC;IACtC,OAAO,EACL,KAAK,oCAAoC,IAAI,oCAAoC,EACjF,KAAK,kCAAkC,IAAI,kCAAkC,EAC7E,KAAK,8BAA8B,IAAI,8BAA8B,GACtE,CAAC;CACH"}

package/resources/batch-inference.js

@@ -4,34 +4,12 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.BatchInference = void 0;
 const resource_1 = require("../resource.js");
 class BatchInference extends resource_1.APIResource {
-    chatCompletion(params, options) {
-        const { 'X-LlamaStack-ProviderData': xLlamaStackProviderData, ...body } = params;
-        return this._client.post('/batch_inference/chat_completion', {
-            body,
-            ...options,
-            headers: {
-                ...(xLlamaStackProviderData != null ?
-                    { 'X-LlamaStack-ProviderData': xLlamaStackProviderData }
-                    : undefined),
-                ...options?.headers,
-            },
-        });
+    chatCompletion(body, options) {
+        return this._client.post('/v1/batch-inference/chat-completion', { body, ...options });
     }
-    completion(params, options) {
-        const { 'X-LlamaStack-ProviderData': xLlamaStackProviderData, ...body } = params;
-        return this._client.post('/batch_inference/completion', {
-            body,
-            ...options,
-            headers: {
-                ...(xLlamaStackProviderData != null ?
-                    { 'X-LlamaStack-ProviderData': xLlamaStackProviderData }
-                    : undefined),
-                ...options?.headers,
-            },
-        });
+    completion(body, options) {
+        return this._client.post('/v1/batch-inference/completion', { body, ...options });
     }
 }
 exports.BatchInference = BatchInference;
-(function (BatchInference) {
-})(BatchInference = exports.BatchInference || (exports.BatchInference = {}));
 //# sourceMappingURL=batch-inference.js.map

package/resources/batch-inference.js.map

@@ -1 +1 @@
-{"version":3,"file":"batch-inference.js","sourceRoot":"","sources":["../src/resources/batch-inference.ts"],"names":[],"mappings":";AAAA,sFAAsF;;;AAEtF,6CAA0C;AAM1C,MAAa,cAAe,SAAQ,sBAAW;IAC7C,cAAc,CACZ,MAA0C,EAC1C,OAA6B;QAE7B,MAAM,EAAE,2BAA2B,EAAE,uBAAuB,EAAE,GAAG,IAAI,EAAE,GAAG,MAAM,CAAC;QACjF,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,kCAAkC,EAAE;YAC3D,IAAI;YACJ,GAAG,OAAO;YACV,OAAO,EAAE;gBACP,GAAG,CAAC,uBAAuB,IAAI,IAAI,CAAC,CAAC;oBACnC,EAAE,2BAA2B,EAAE,uBAAuB,EAAE;oBAC1D,CAAC,CAAC,SAAS,CAAC;gBACZ,GAAG,OAAO,EAAE,OAAO;aACpB;SACF,CAAC,CAAC;IACL,CAAC;IAED,UAAU,CACR,MAAsC,EACtC,OAA6B;QAE7B,MAAM,EAAE,2BAA2B,EAAE,uBAAuB,EAAE,GAAG,IAAI,EAAE,GAAG,MAAM,CAAC;QACjF,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,6BAA6B,EAAE;YACtD,IAAI;YACJ,GAAG,OAAO;YACV,OAAO,EAAE;gBACP,GAAG,CAAC,uBAAuB,IAAI,IAAI,CAAC,CAAC;oBACnC,EAAE,2BAA2B,EAAE,uBAAuB,EAAE;oBAC1D,CAAC,CAAC,SAAS,CAAC;gBACZ,GAAG,OAAO,EAAE,OAAO;aACpB;SACF,CAAC,CAAC;IACL,CAAC;CACF;AAlCD,wCAkCC;AA2GD,WAAiB,cAAc;AAI/B,CAAC,EAJgB,cAAc,GAAd,sBAAc,KAAd,sBAAc,QAI9B"}
+{"version":3,"file":"batch-inference.js","sourceRoot":"","sources":["../src/resources/batch-inference.ts"],"names":[],"mappings":";AAAA,sFAAsF;;;AAEtF,6CAA0C;AAI1C,MAAa,cAAe,SAAQ,sBAAW;IAC7C,cAAc,CACZ,IAAwC,EACxC,OAA6B;QAE7B,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,qCAAqC,EAAE,EAAE,IAAI,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC;IACxF,CAAC;IAED,UAAU,CACR,IAAoC,EACpC,OAA6B;QAE7B,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,gCAAgC,EAAE,EAAE,IAAI,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC;IACnF,CAAC;CACF;AAdD,wCAcC"}

package/resources/batch-inference.mjs

@@ -1,33 +1,11 @@
 // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 import { APIResource } from "../resource.mjs";
 export class BatchInference extends APIResource {
-    chatCompletion(params, options) {
-        const { 'X-LlamaStack-ProviderData': xLlamaStackProviderData, ...body } = params;
-        return this._client.post('/batch_inference/chat_completion', {
-            body,
-            ...options,
-            headers: {
-                ...(xLlamaStackProviderData != null ?
-                    { 'X-LlamaStack-ProviderData': xLlamaStackProviderData }
-                    : undefined),
-                ...options?.headers,
-            },
-        });
+    chatCompletion(body, options) {
+        return this._client.post('/v1/batch-inference/chat-completion', { body, ...options });
     }
-    completion(params, options) {
-        const { 'X-LlamaStack-ProviderData': xLlamaStackProviderData, ...body } = params;
-        return this._client.post('/batch_inference/completion', {
-            body,
-            ...options,
-            headers: {
-                ...(xLlamaStackProviderData != null ?
-                    { 'X-LlamaStack-ProviderData': xLlamaStackProviderData }
-                    : undefined),
-                ...options?.headers,
-            },
-        });
+    completion(body, options) {
+        return this._client.post('/v1/batch-inference/completion', { body, ...options });
     }
 }
-(function (BatchInference) {
-})(BatchInference || (BatchInference = {}));
 //# sourceMappingURL=batch-inference.mjs.map

package/resources/batch-inference.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"batch-inference.mjs","sourceRoot":"","sources":["../src/resources/batch-inference.ts"],"names":[],"mappings":"AAAA,sFAAsF;OAE/E,EAAE,WAAW,EAAE;AAMtB,MAAM,OAAO,cAAe,SAAQ,WAAW;IAC7C,cAAc,CACZ,MAA0C,EAC1C,OAA6B;QAE7B,MAAM,EAAE,2BAA2B,EAAE,uBAAuB,EAAE,GAAG,IAAI,EAAE,GAAG,MAAM,CAAC;QACjF,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,kCAAkC,EAAE;YAC3D,IAAI;YACJ,GAAG,OAAO;YACV,OAAO,EAAE;gBACP,GAAG,CAAC,uBAAuB,IAAI,IAAI,CAAC,CAAC;oBACnC,EAAE,2BAA2B,EAAE,uBAAuB,EAAE;oBAC1D,CAAC,CAAC,SAAS,CAAC;gBACZ,GAAG,OAAO,EAAE,OAAO;aACpB;SACF,CAAC,CAAC;IACL,CAAC;IAED,UAAU,CACR,MAAsC,EACtC,OAA6B;QAE7B,MAAM,EAAE,2BAA2B,EAAE,uBAAuB,EAAE,GAAG,IAAI,EAAE,GAAG,MAAM,CAAC;QACjF,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,6BAA6B,EAAE;YACtD,IAAI;YACJ,GAAG,OAAO;YACV,OAAO,EAAE;gBACP,GAAG,CAAC,uBAAuB,IAAI,IAAI,CAAC,CAAC;oBACnC,EAAE,2BAA2B,EAAE,uBAAuB,EAAE;oBAC1D,CAAC,CAAC,SAAS,CAAC;gBACZ,GAAG,OAAO,EAAE,OAAO;aACpB;SACF,CAAC,CAAC;IACL,CAAC;CACF;AA2GD,WAAiB,cAAc;AAI/B,CAAC,EAJgB,cAAc,KAAd,cAAc,QAI9B"}
+{"version":3,"file":"batch-inference.mjs","sourceRoot":"","sources":["../src/resources/batch-inference.ts"],"names":[],"mappings":"AAAA,sFAAsF;OAE/E,EAAE,WAAW,EAAE;AAItB,MAAM,OAAO,cAAe,SAAQ,WAAW;IAC7C,cAAc,CACZ,IAAwC,EACxC,OAA6B;QAE7B,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,qCAAqC,EAAE,EAAE,IAAI,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC;IACxF,CAAC;IAED,UAAU,CACR,IAAoC,EACpC,OAA6B;QAE7B,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,gCAAgC,EAAE,EAAE,IAAI,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC;IACnF,CAAC;CACF"}

package/resources/datasetio.d.ts

@@ -0,0 +1,25 @@
+import { APIResource } from "../resource.js";
+import * as Core from "../core.js";
+export declare class Datasetio extends APIResource {
+    appendRows(body: DatasetioAppendRowsParams, options?: Core.RequestOptions): Core.APIPromise<void>;
+    getRowsPaginated(query: DatasetioGetRowsPaginatedParams, options?: Core.RequestOptions): Core.APIPromise<PaginatedRowsResult>;
+}
+export interface PaginatedRowsResult {
+    rows: Array<Record<string, boolean | number | string | Array<unknown> | unknown | null>>;
+    total_count: number;
+    next_page_token?: string;
+}
+export interface DatasetioAppendRowsParams {
+    dataset_id: string;
+    rows: Array<Record<string, boolean | number | string | Array<unknown> | unknown | null>>;
+}
+export interface DatasetioGetRowsPaginatedParams {
+    dataset_id: string;
+    rows_in_page: number;
+    filter_condition?: string;
+    page_token?: string;
+}
+export declare namespace Datasetio {
+    export { type PaginatedRowsResult as PaginatedRowsResult, type DatasetioAppendRowsParams as DatasetioAppendRowsParams, type DatasetioGetRowsPaginatedParams as DatasetioGetRowsPaginatedParams, };
+}
+//# sourceMappingURL=datasetio.d.ts.map

package/resources/datasetio.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"datasetio.d.ts","sourceRoot":"","sources":["../src/resources/datasetio.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAC1C,OAAO,KAAK,IAAI,MAAM,SAAS,CAAC;AAEhC,qBAAa,SAAU,SAAQ,WAAW;IACxC,UAAU,CAAC,IAAI,EAAE,yBAAyB,EAAE,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC;IAQjG,gBAAgB,CACd,KAAK,EAAE,+BAA+B,EACtC,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAC5B,IAAI,CAAC,UAAU,CAAC,mBAAmB,CAAC;CAGxC;AAED,MAAM,WAAW,mBAAmB;IAClC,IAAI,EAAE,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,GAAG,MAAM,GAAG,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,OAAO,GAAG,IAAI,CAAC,CAAC,CAAC;IAEzF,WAAW,EAAE,MAAM,CAAC;IAEpB,eAAe,CAAC,EAAE,MAAM,CAAC;CAC1B;AAED,MAAM,WAAW,yBAAyB;IACxC,UAAU,EAAE,MAAM,CAAC;IAEnB,IAAI,EAAE,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,GAAG,MAAM,GAAG,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,OAAO,GAAG,IAAI,CAAC,CAAC,CAAC;CAC1F;AAED,MAAM,WAAW,+BAA+B;IAC9C,UAAU,EAAE,MAAM,CAAC;IAEnB,YAAY,EAAE,MAAM,CAAC;IAErB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAE1B,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAED,MAAM,CAAC,OAAO,WAAW,SAAS,CAAC;IACjC,OAAO,EACL,KAAK,mBAAmB,IAAI,mBAAmB,EAC/C,KAAK,yBAAyB,IAAI,yBAAyB,EAC3D,KAAK,+BAA+B,IAAI,+BAA+B,GACxE,CAAC;CACH"}