llama-stack-client 0.4.4 → 0.4.5

package/tests/api-resources/beta/datasets.test.ts

@@ -49,7 +49,10 @@ describe('resource datasets', () => {
   });
 
   test('appendrows: only required params', async () => {
-    const responsePromise = client.beta.datasets.appendrows('dataset_id', { rows: [{ foo: 'bar' }] });
+    const responsePromise = client.beta.datasets.appendrows('dataset_id', {
+      body_dataset_id: 'dataset_id',
+      rows: [{ foo: 'bar' }],
+    });
     const rawResponse = await responsePromise.asResponse();
     expect(rawResponse).toBeInstanceOf(Response);
     const response = await responsePromise;
@@ -60,7 +63,10 @@ describe('resource datasets', () => {
   });
 
   test('appendrows: required and optional params', async () => {
-    const response = await client.beta.datasets.appendrows('dataset_id', { rows: [{ foo: 'bar' }] });
+    const response = await client.beta.datasets.appendrows('dataset_id', {
+      body_dataset_id: 'dataset_id',
+      rows: [{ foo: 'bar' }],
+    });
   });
 
   test('iterrows', async () => {
@@ -95,7 +101,7 @@ describe('resource datasets', () => {
   test('register: only required params', async () => {
     const responsePromise = client.beta.datasets.register({
       purpose: 'post-training/messages',
-      source: { uri: 'uri' },
+      source: { uri: 'uri', type: 'uri' },
     });
     const rawResponse = await responsePromise.asResponse();
     expect(rawResponse).toBeInstanceOf(Response);

package/tests/api-resources/chat/completions.test.ts

@@ -28,19 +28,27 @@ describe('resource completions', () => {
 
   test('create: required and optional params', async () => {
     const response = await client.chat.completions.create({
-      messages: [{ content: 'string', name: 'name', role: 'user' }],
+      messages: [
+        {
+          content: 'string',
+          name: 'name',
+          role: 'user',
+        },
+      ],
       model: 'model',
-      frequency_penalty: 0,
+      frequency_penalty: -2,
       function_call: 'string',
       functions: [{ foo: 'bar' }],
       logit_bias: { foo: 0 },
       logprobs: true,
-      max_completion_tokens: 0,
-      max_tokens: 0,
-      n: 0,
+      max_completion_tokens: 1,
+      max_tokens: 1,
+      n: 1,
       parallel_tool_calls: true,
-      presence_penalty: 0,
+      presence_penalty: -2,
+      reasoning_effort: 'none',
       response_format: { type: 'text' },
+      safety_identifier: 'safety_identifier',
       seed: 0,
       stop: 'string',
       stream: false,
@@ -94,7 +102,12 @@ describe('resource completions', () => {
     // ensure the request options are being passed correctly by passing an invalid HTTP method in order to cause an error
     await expect(
       client.chat.completions.list(
-        { after: 'after', limit: 0, model: 'model', order: 'asc' },
+        {
+          after: 'after',
+          limit: 0,
+          model: 'model',
+          order: 'asc',
+        },
         { path: '/_stainless_unknown_path' },
       ),
     ).rejects.toThrow(LlamaStackClient.NotFoundError);

package/tests/api-resources/completions.test.ts

@@ -27,14 +27,14 @@ describe('resource completions', () => {
     const response = await client.completions.create({
       model: 'model',
       prompt: 'string',
-      best_of: 0,
+      best_of: 1,
       echo: true,
-      frequency_penalty: 0,
+      frequency_penalty: -2,
       logit_bias: { foo: 0 },
       logprobs: true,
-      max_tokens: 0,
-      n: 0,
-      presence_penalty: 0,
+      max_tokens: 1,
+      n: 1,
+      presence_penalty: -2,
       seed: 0,
       stop: 'string',
       stream: false,

package/tests/api-resources/conversations/items.test.ts

@@ -14,7 +14,13 @@ const client = new LlamaStackClient({ baseURL: process.env['TEST_API_BASE_URL']
 describe('resource items', () => {
   test('create: only required params', async () => {
     const responsePromise = client.conversations.items.create('conversation_id', {
-      items: [{ content: 'string', role: 'system', type: 'message' }],
+      items: [
+        {
+          content: 'string',
+          role: 'system',
+          type: 'message',
+        },
+      ],
     });
     const rawResponse = await responsePromise.asResponse();
     expect(rawResponse).toBeInstanceOf(Response);
@@ -27,7 +33,15 @@ describe('resource items', () => {
 
   test('create: required and optional params', async () => {
     const response = await client.conversations.items.create('conversation_id', {
-      items: [{ content: 'string', role: 'system', id: 'id', status: 'status', type: 'message' }],
+      items: [
+        {
+          content: 'string',
+          role: 'system',
+          id: 'id',
+          status: 'status',
+          type: 'message',
+        },
+      ],
     });
   });
 
@@ -54,7 +68,12 @@ describe('resource items', () => {
     await expect(
       client.conversations.items.list(
         'conversation_id',
-        { after: 'after', include: ['web_search_call.action.sources'], limit: 0, order: 'asc' },
+        {
+          after: 'after',
+          include: ['web_search_call.action.sources'],
+          limit: 0,
+          order: 'asc',
+        },
         { path: '/_stainless_unknown_path' },
       ),
     ).rejects.toThrow(LlamaStackClient.NotFoundError);

package/tests/api-resources/embeddings.test.ts

@@ -27,8 +27,8 @@ describe('resource embeddings', () => {
     const response = await client.embeddings.create({
       input: 'string',
       model: 'model',
-      dimensions: 0,
-      encoding_format: 'encoding_format',
+      dimensions: 1,
+      encoding_format: 'float',
       user: 'user',
     });
   });

package/tests/api-resources/files.test.ts

@@ -74,7 +74,12 @@ describe('resource files', () => {
     // ensure the request options are being passed correctly by passing an invalid HTTP method in order to cause an error
     await expect(
       client.files.list(
-        { after: 'after', limit: 0, order: 'asc', purpose: 'assistants' },
+        {
+          after: 'after',
+          limit: 0,
+          order: 'asc',
+          purpose: 'assistants',
+        },
         { path: '/_stainless_unknown_path' },
       ),
     ).rejects.toThrow(LlamaStackClient.NotFoundError);

package/tests/api-resources/responses/input-items.test.ts

@@ -35,7 +35,13 @@ describe('resource inputItems', () => {
     await expect(
       client.responses.inputItems.list(
         'response_id',
-        { after: 'after', before: 'before', include: ['string'], limit: 0, order: 'asc' },
+        {
+          after: 'after',
+          before: 'before',
+          include: ['web_search_call.action.sources'],
+          limit: 0,
+          order: 'asc',
+        },
         { path: '/_stainless_unknown_path' },
       ),
     ).rejects.toThrow(LlamaStackClient.NotFoundError);

package/tests/api-resources/responses/responses.test.ts

@@ -28,14 +28,22 @@ describe('resource responses', () => {
       input: 'string',
       model: 'model',
       conversation: 'conversation',
-      include: ['string'],
+      guardrails: ['string'],
+      include: ['web_search_call.action.sources'],
       instructions: 'instructions',
-      max_infer_iters: 0,
-      max_tool_calls: 0,
+      max_infer_iters: 1,
+      max_output_tokens: 16,
+      max_tool_calls: 1,
       metadata: { foo: 'string' },
       parallel_tool_calls: true,
       previous_response_id: 'previous_response_id',
-      prompt: { id: 'id', variables: { foo: { text: 'text', type: 'input_text' } }, version: 'version' },
+      prompt: {
+        id: 'id',
+        variables: { foo: { text: 'text', type: 'input_text' } },
+        version: 'version',
+      },
+      reasoning: { effort: 'none' },
+      safety_identifier: 'safety_identifier',
       store: true,
       stream: false,
       temperature: 0,
@@ -48,6 +56,7 @@ describe('resource responses', () => {
           type: 'text',
         },
       },
+      tool_choice: 'auto',
       tools: [{ search_context_size: 'S?oC"high', type: 'web_search' }],
     });
   });
@@ -92,7 +101,12 @@ describe('resource responses', () => {
     // ensure the request options are being passed correctly by passing an invalid HTTP method in order to cause an error
     await expect(
       client.responses.list(
-        { after: 'after', limit: 0, model: 'model', order: 'asc' },
+        {
+          after: 'after',
+          limit: 0,
+          model: 'model',
+          order: 'asc',
+        },
         { path: '/_stainless_unknown_path' },
       ),
     ).rejects.toThrow(LlamaStackClient.NotFoundError);

package/tests/api-resources/safety.test.ts

@@ -15,8 +15,7 @@ describe('resource safety', () => {
   test('runShield: only required params', async () => {
     const responsePromise = client.safety.runShield({
       messages: [{ content: 'string', role: 'user' }],
-      params: { foo: 'bar' },
-      shield_id: 'shield_id',
+      shield_id: 'x',
     });
     const rawResponse = await responsePromise.asResponse();
     expect(rawResponse).toBeInstanceOf(Response);
@@ -29,9 +28,14 @@ describe('resource safety', () => {
 
   test('runShield: required and optional params', async () => {
     const response = await client.safety.runShield({
-      messages: [{ content: 'string', name: 'name', role: 'user' }],
-      params: { foo: 'bar' },
-      shield_id: 'shield_id',
+      messages: [
+        {
+          content: 'string',
+          name: 'name',
+          role: 'user',
+        },
+      ],
+      shield_id: 'x',
     });
   });
 });

package/tests/api-resources/tool-runtime.test.ts

@@ -13,7 +13,10 @@ const client = new LlamaStackClient({ baseURL: process.env['TEST_API_BASE_URL']
 
 describe('resource toolRuntime', () => {
   test('invokeTool: only required params', async () => {
-    const responsePromise = client.toolRuntime.invokeTool({ kwargs: { foo: 'bar' }, tool_name: 'tool_name' });
+    const responsePromise = client.toolRuntime.invokeTool({
+      kwargs: { foo: 'bar' },
+      tool_name: 'tool_name',
+    });
     const rawResponse = await responsePromise.asResponse();
     expect(rawResponse).toBeInstanceOf(Response);
     const response = await responsePromise;
@@ -53,7 +56,11 @@ describe('resource toolRuntime', () => {
     // ensure the request options are being passed correctly by passing an invalid HTTP method in order to cause an error
     await expect(
       client.toolRuntime.listTools(
-        { authorization: 'authorization', mcp_endpoint: { uri: 'uri' }, tool_group_id: 'tool_group_id' },
+        {
+          authorization: 'authorization',
+          mcp_endpoint: { uri: 'uri' },
+          tool_group_id: 'tool_group_id',
+        },
         { path: '/_stainless_unknown_path' },
       ),
     ).rejects.toThrow(LlamaStackClient.NotFoundError);

package/tests/api-resources/vector-io.test.ts

@@ -14,7 +14,16 @@ const client = new LlamaStackClient({ baseURL: process.env['TEST_API_BASE_URL']
 describe('resource vectorIo', () => {
   test('insert: only required params', async () => {
     const responsePromise = client.vectorIo.insert({
-      chunks: [{ chunk_id: 'chunk_id', content: 'string' }],
+      chunks: [
+        {
+          chunk_id: 'chunk_id',
+          chunk_metadata: {},
+          content: 'string',
+          embedding: [0],
+          embedding_dimension: 0,
+          embedding_model: 'embedding_model',
+        },
+      ],
       vector_store_id: 'vector_store_id',
     });
     const rawResponse = await responsePromise.asResponse();
@@ -31,10 +40,7 @@ describe('resource vectorIo', () => {
       chunks: [
         {
           chunk_id: 'chunk_id',
-          content: 'string',
           chunk_metadata: {
-            chunk_embedding_dimension: 0,
-            chunk_embedding_model: 'chunk_embedding_model',
             chunk_id: 'chunk_id',
             chunk_tokenizer: 'chunk_tokenizer',
             chunk_window: 'chunk_window',
@@ -45,7 +51,10 @@ describe('resource vectorIo', () => {
             source: 'source',
             updated_timestamp: 0,
           },
+          content: 'string',
           embedding: [0],
+          embedding_dimension: 0,
+          embedding_model: 'embedding_model',
           metadata: { foo: 'bar' },
         },
       ],

package/tests/api-resources/vector-stores/file-batches.test.ts

@@ -99,7 +99,13 @@ describe('resource fileBatches', () => {
       client.vectorStores.fileBatches.listFiles(
         'vector_store_id',
         'batch_id',
-        { after: 'after', before: 'before', filter: 'filter', limit: 0, order: 'order' },
+        {
+          after: 'after',
+          before: 'before',
+          filter: 'filter',
+          limit: 0,
+          order: 'order',
+        },
         { path: '/_stainless_unknown_path' },
       ),
     ).rejects.toThrow(LlamaStackClient.NotFoundError);

package/tests/api-resources/vector-stores/files.test.ts

@@ -91,7 +91,13 @@ describe('resource files', () => {
     await expect(
       client.vectorStores.files.list(
         'vector_store_id',
-        { after: 'after', before: 'before', filter: 'completed', limit: 0, order: 'order' },
+        {
+          after: 'after',
+          before: 'before',
+          filter: 'completed',
+          limit: 0,
+          order: 'order',
+        },
         { path: '/_stainless_unknown_path' },
       ),
     ).rejects.toThrow(LlamaStackClient.NotFoundError);

package/tests/api-resources/vector-stores/vector-stores.test.ts

@@ -74,7 +74,12 @@ describe('resource vectorStores', () => {
     // ensure the request options are being passed correctly by passing an invalid HTTP method in order to cause an error
     await expect(
       client.vectorStores.list(
-        { after: 'after', before: 'before', limit: 0, order: 'order' },
+        {
+          after: 'after',
+          before: 'before',
+          limit: 0,
+          order: 'order',
+        },
         { path: '/_stainless_unknown_path' },
       ),
     ).rejects.toThrow(LlamaStackClient.NotFoundError);
@@ -114,7 +119,14 @@ describe('resource vectorStores', () => {
       query: 'string',
       filters: { foo: 'bar' },
       max_num_results: 0,
-      ranking_options: { ranker: 'ranker', score_threshold: 0 },
+      ranking_options: {
+        alpha: 0,
+        impact_factor: 0,
+        model: 'model',
+        ranker: 'ranker',
+        score_threshold: 0,
+        weights: { foo: 0 },
+      },
       rewrite_query: true,
       search_mode: 'search_mode',
     });