From e752a4462cf0e8982166849d0fa20034bc9e0e0a Mon Sep 17 00:00:00 2001 From: Hardik Shah Date: Mon, 16 Jun 2025 11:58:05 -0700 Subject: [PATCH] Sync updates from stainless branch: hardikjshah/dev --- src/index.ts | 8 ++++---- src/resource.ts | 2 +- src/resources/datasets.ts | 5 +++++ src/resources/embeddings.ts | 15 +++++++++------ src/resources/index.ts | 2 +- 5 files changed, 20 insertions(+), 12 deletions(-) diff --git a/src/index.ts b/src/index.ts index 0727bcd..49b9135 100644 --- a/src/index.ts +++ b/src/index.ts @@ -32,7 +32,7 @@ import { Datasets, ListDatasetsResponse, } from './resources/datasets'; -import { EmbeddingCreateParams, Embeddings, EmbeddingsResponse } from './resources/embeddings'; +import { CreateEmbeddingsResponse, EmbeddingCreateParams, Embeddings } from './resources/embeddings'; import { DeleteFileResponse, File, @@ -45,7 +45,7 @@ import { import { ChatCompletionResponseStreamChunk, CompletionResponse, - EmbeddingsResponse as InferenceAPIEmbeddingsResponse, + EmbeddingsResponse, Inference, InferenceBatchChatCompletionParams, InferenceBatchChatCompletionResponse, @@ -492,7 +492,7 @@ export declare namespace LlamaStackClient { Inference as Inference, type ChatCompletionResponseStreamChunk as ChatCompletionResponseStreamChunk, type CompletionResponse as CompletionResponse, - type InferenceAPIEmbeddingsResponse as EmbeddingsResponse, + type EmbeddingsResponse as EmbeddingsResponse, type TokenLogProbs as TokenLogProbs, type InferenceBatchChatCompletionResponse as InferenceBatchChatCompletionResponse, type InferenceBatchChatCompletionParams as InferenceBatchChatCompletionParams, @@ -508,7 +508,7 @@ export declare namespace LlamaStackClient { export { Embeddings as Embeddings, - type EmbeddingsResponse as EmbeddingsResponse, + type CreateEmbeddingsResponse as CreateEmbeddingsResponse, type EmbeddingCreateParams as EmbeddingCreateParams, }; diff --git a/src/resource.ts b/src/resource.ts index a433822..eafa8c0 100644 --- a/src/resource.ts +++ b/src/resource.ts @@ -2,7 +2,7 @@ import type { LlamaStackClient } from './index'; -export class APIResource { +export abstract class APIResource { protected _client: LlamaStackClient; constructor(client: LlamaStackClient) { diff --git a/src/resources/datasets.ts b/src/resources/datasets.ts index 313e0a1..631562e 100644 --- a/src/resources/datasets.ts +++ b/src/resources/datasets.ts @@ -195,6 +195,11 @@ export interface DatasetIterrowsResponse { * Whether there are more items available after this set */ has_more: boolean; + + /** + * The URL for accessing this list + */ + url?: string; } export interface DatasetRegisterResponse { diff --git a/src/resources/embeddings.ts b/src/resources/embeddings.ts index d04331b..89758af 100644 --- a/src/resources/embeddings.ts +++ b/src/resources/embeddings.ts @@ -8,7 +8,10 @@ export class Embeddings extends APIResource { * Generate OpenAI-compatible embeddings for the given input using the specified * model. */ - create(body: EmbeddingCreateParams, options?: Core.RequestOptions): Core.APIPromise { + create( + body: EmbeddingCreateParams, + options?: Core.RequestOptions, + ): Core.APIPromise { return this._client.post('/v1/openai/v1/embeddings', { body, ...options }); } } @@ -16,11 +19,11 @@ export class Embeddings extends APIResource { /** * Response from an OpenAI-compatible embeddings request. */ -export interface EmbeddingsResponse { +export interface CreateEmbeddingsResponse { /** * List of embedding data objects */ - data: Array; + data: Array; /** * The model that was used to generate the embeddings @@ -35,10 +38,10 @@ export interface EmbeddingsResponse { /** * Usage information */ - usage: EmbeddingsResponse.Usage; + usage: CreateEmbeddingsResponse.Usage; } -export namespace EmbeddingsResponse { +export namespace CreateEmbeddingsResponse { /** * A single embedding data object from an OpenAI-compatible embeddings response. */ @@ -110,7 +113,7 @@ export interface EmbeddingCreateParams { export declare namespace Embeddings { export { - type EmbeddingsResponse as EmbeddingsResponse, + type CreateEmbeddingsResponse as CreateEmbeddingsResponse, type EmbeddingCreateParams as EmbeddingCreateParams, }; } diff --git a/src/resources/index.ts b/src/resources/index.ts index 902b74e..376eab8 100644 --- a/src/resources/index.ts +++ b/src/resources/index.ts @@ -36,7 +36,7 @@ export { type DatasetIterrowsParams, type DatasetRegisterParams, } from './datasets'; -export { Embeddings, type EmbeddingsResponse, type EmbeddingCreateParams } from './embeddings'; +export { Embeddings, type CreateEmbeddingsResponse, type EmbeddingCreateParams } from './embeddings'; export { Eval, type BenchmarkConfig,