feat(structured outputs): support accessing raw responses (#1058)

openai · Sep 11, 2024 · af17697 · af17697
1 parent f9b7eac
commit af17697
Showing 1 changed file with 11 additions and 11 deletions.
diff --git a/src/resources/beta/chat/completions.ts b/src/resources/beta/chat/completions.ts
@@ -59,21 +59,21 @@ export interface ParsedChatCompletion<ParsedT> extends ChatCompletion {
 export type ChatCompletionParseParams = ChatCompletionCreateParamsNonStreaming;
 
 export class Completions extends APIResource {
-  async parse<Params extends ChatCompletionParseParams, ParsedT = ExtractParsedContentFromParams<Params>>(
+  parse<Params extends ChatCompletionParseParams, ParsedT = ExtractParsedContentFromParams<Params>>(
     body: Params,
     options?: Core.RequestOptions,
-  ): Promise<ParsedChatCompletion<ParsedT>> {
+  ): Core.APIPromise<ParsedChatCompletion<ParsedT>> {
     validateInputTools(body.tools);
 
-    const completion = await this._client.chat.completions.create(body, {
-      ...options,
-      headers: {
-        ...options?.headers,
-        'X-Stainless-Helper-Method': 'beta.chat.completions.parse',
-      },
-    });
-
-    return parseChatCompletion(completion, body);
+    return this._client.chat.completions
+      .create(body, {
+        ...options,
+        headers: {
+          ...options?.headers,
+          'X-Stainless-Helper-Method': 'beta.chat.completions.parse',
+        },
+      })
+      ._thenUnwrap((completion) => parseChatCompletion(completion, body));
   }
 
   /**