Sync updates from stainless branch: main

examples/agents.test.ts

@@ -60,7 +60,6 @@ describe('RAG Integration Tests', () => {
         for await (const chunk of turnResponse) {
             if (chunk.event.payload.event_type === 'turn_complete') {
                 expect(chunk.event.payload.turn.output_message).toBeTruthy();
-            break;
             }
         }
     }

src/core.ts

@@ -62,8 +62,8 @@ async function defaultParseResponse<T>(props: APIResponseProps): Promise<T> {
   }
 
   const contentType = response.headers.get('content-type');
-  const isJSON =
-    contentType?.includes('application/json') || contentType?.includes('application/vnd.api+json');
+  const mediaType = contentType?.split(';')[0]?.trim();
+  const isJSON = mediaType?.includes('application/json') || mediaType?.endsWith('+json');
   if (isJSON) {
     const json = await response.json();
 

src/resources/agents.ts

@@ -0,0 +1,3 @@
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+export * from './agents/index';

src/resources/agents/turn.ts

@@ -482,10 +482,9 @@ export type TurnResumeParams = TurnResumeParamsNonStreaming | TurnResumeParamsSt
 
 export interface TurnResumeParamsBase {
   /**
-   * The tool call responses to resume the turn with. NOTE: ToolResponseMessage will
-   * be deprecated. Use ToolResponse.
+   * The tool call responses to resume the turn with.
    */
-  tool_responses: Array<AgentsAPI.ToolResponse> | Array<Shared.ToolResponseMessage>;
+  tool_responses: Array<AgentsAPI.ToolResponse>;
 
   /**
    * Whether to stream the response.

src/resources/eval.ts

@@ -0,0 +1,3 @@
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+export * from './eval/index';

src/resources/inference.ts

@@ -113,19 +113,9 @@ export namespace ChatCompletionResponseStreamChunk {
   export interface Metric {
     metric: string;
 
-    span_id: string;
-
-    timestamp: string;
-
-    trace_id: string;
-
-    type: 'metric';
-
-    unit: string;
-
     value: number;
 
-    attributes?: Record<string, string | number | boolean | null>;
+    unit?: string;
   }
 }
 
@@ -147,6 +137,18 @@ export interface CompletionResponse {
    * Optional log probabilities for generated tokens
    */
   logprobs?: Array<TokenLogProbs>;
+
+  metrics?: Array<CompletionResponse.Metric>;
+}
+
+export namespace CompletionResponse {
+  export interface Metric {
+    metric: string;
+
+    value: number;
+
+    unit?: string;
+  }
 }
 
 /**

src/resources/inspect.ts

@@ -20,6 +20,8 @@ export interface HealthInfo {
 export interface ProviderInfo {
   api: string;
 
+  config: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+
   provider_id: string;
 
   provider_type: string;

src/resources/post-training.ts

@@ -0,0 +1,3 @@
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+export * from './post-training/index';

src/resources/providers.ts

@@ -5,9 +5,13 @@ import * as Core from '../core';
 import * as InspectAPI from './inspect';
 
 export class Providers extends APIResource {
+  retrieve(providerId: string, options?: Core.RequestOptions): Core.APIPromise<InspectAPI.ProviderInfo> {
+    return this._client.get(`/v1/providers/${providerId}`, options);
+  }
+
   list(options?: Core.RequestOptions): Core.APIPromise<ProviderListResponse> {
     return (
-      this._client.get('/v1/inspect/providers', options) as Core.APIPromise<{ data: ProviderListResponse }>
+      this._client.get('/v1/providers', options) as Core.APIPromise<{ data: ProviderListResponse }>
     )._thenUnwrap((obj) => obj.data);
   }
 }

src/resources/shared.ts

@@ -110,19 +110,9 @@ export namespace ChatCompletionResponse {
   export interface Metric {
     metric: string;
 
-    span_id: string;
-
-    timestamp: string;
-
-    trace_id: string;
-
-    type: 'metric';
-
-    unit: string;
-
     value: number;
 
-    attributes?: Record<string, string | number | boolean | null>;
+    unit?: string;
   }
 }
 
@@ -663,11 +653,6 @@ export interface ToolResponseMessage {
    * Must be "tool" to identify this as a tool response
    */
   role: 'tool';
-
-  /**
-   * Name of the tool that was called
-   */
-  tool_name: 'brave_search' | 'wolfram_alpha' | 'photogen' | 'code_interpreter' | (string & {});
 }
 
 /**

src/resources/tool-runtime.ts

@@ -0,0 +1,3 @@
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+export * from './tool-runtime/index';

tests/api-resources/providers.test.ts

@@ -6,6 +6,24 @@ import { Response } from 'node-fetch';
 const client = new LlamaStackClient({ baseURL: process.env['TEST_API_BASE_URL'] ?? 'http://127.0.0.1:4010' });
 
 describe('resource providers', () => {
+  test('retrieve', async () => {
+    const responsePromise = client.providers.retrieve('provider_id');
+    const rawResponse = await responsePromise.asResponse();
+    expect(rawResponse).toBeInstanceOf(Response);
+    const response = await responsePromise;
+    expect(response).not.toBeInstanceOf(Response);
+    const dataAndResponse = await responsePromise.withResponse();
+    expect(dataAndResponse.data).toBe(response);
+    expect(dataAndResponse.response).toBe(rawResponse);
+  });
+
+  test('retrieve: request options instead of params are passed correctly', async () => {
+    // ensure the request options are being passed correctly by passing an invalid HTTP method in order to cause an error
+    await expect(
+      client.providers.retrieve('provider_id', { path: '/_stainless_unknown_path' }),
+    ).rejects.toThrow(LlamaStackClient.NotFoundError);
+  });
+
   test('list', async () => {
     const responsePromise = client.providers.list();
     const rawResponse = await responsePromise.asResponse();