fix(chat): support non-streaming chat completion requests

lib/shared/src/sourcegraph-api/completions/browserClient.ts

@@ -115,6 +115,53 @@ export class SourcegraphBrowserCompletionsClient extends SourcegraphCompletionsC
             console.error(error)
         })
     }
+
+    protected async _fetchWithCallbacks(
+        params: CompletionParameters,
+        requestParams: CompletionRequestParameters,
+        cb: CompletionCallbacks,
+        signal?: AbortSignal
+    ): Promise<void> {
+        const { url, serializedParams } = await this.prepareRequest(params, requestParams)
+        const headersInstance = new Headers({
+            'Content-Type': 'application/json; charset=utf-8',
+            ...this.config.customHeaders,
+            ...requestParams.customHeaders,
+        })
+        addCustomUserAgent(headersInstance)
+        if (this.config.accessToken) {
+            headersInstance.set('Authorization', `token ${this.config.accessToken}`)
+        }
+        if (new URLSearchParams(globalThis.location.search).get('trace')) {
+            headersInstance.set('X-Sourcegraph-Should-Trace', 'true')
+        }
+        try {
+            const response = await fetch(url.toString(), {
+                method: 'POST',
+                headers: headersInstance,
+                body: JSON.stringify(serializedParams),
+                signal,
+            })
+            if (!response.ok) {
+                const errorMessage = await response.text()
+                throw new Error(
+                    errorMessage.length === 0
+                        ? `Request failed with status code ${response.status}`
+                        : errorMessage
+                )
+            }
+            const data = await response.json()
+            if (data?.completion) {
+                cb.onChange(data.completion)
+                cb.onComplete()
+            } else {
+                throw new Error('Unexpected response format')
+            }
+        } catch (error: any) {
+            cb.onError(error.message)
+            console.error(error)
+        }
+    }
 }
 
 if (isRunningInWebWorker) {

lib/shared/src/sourcegraph-api/completions/client.ts

@@ -1,6 +1,6 @@
 import type { Span } from '@opentelemetry/api'
+import { addClientInfoParams, getSerializedParams } from '../..'
 import type { ClientConfigurationWithAccessToken } from '../../configuration'
-
 import { useCustomChatClient } from '../../llm-providers'
 import { recordErrorToSpan } from '../../tracing'
 import type {
@@ -45,6 +45,8 @@ export type CompletionsClientConfig = Pick<
 export abstract class SourcegraphCompletionsClient {
     private errorEncountered = false
 
+    protected readonly isTemperatureZero = process.env.CODY_TEMPERATURE_ZERO === 'true'
+
     constructor(
         protected config: CompletionsClientConfig,
         protected logger?: CompletionLogger
@@ -88,6 +90,27 @@ export abstract class SourcegraphCompletionsClient {
         }
     }
 
+    protected async prepareRequest(
+        params: CompletionParameters,
+        requestParams: CompletionRequestParameters
+    ): Promise<{ url: URL; serializedParams: any }> {
+        const { apiVersion } = requestParams
+        const serializedParams = await getSerializedParams(params)
+        const url = new URL(this.completionsEndpoint)
+        if (apiVersion >= 1) {
+            url.searchParams.append('api-version', '' + apiVersion)
+        }
+        addClientInfoParams(url.searchParams)
+        return { url, serializedParams }
+    }
+
+    protected abstract _fetchWithCallbacks(
+        params: CompletionParameters,
+        requestParams: CompletionRequestParameters,
+        cb: CompletionCallbacks,
+        signal?: AbortSignal
+    ): Promise<void>
+
     protected abstract _streamWithCallbacks(
         params: CompletionParameters,
         requestParams: CompletionRequestParameters,
@@ -144,7 +167,11 @@ export abstract class SourcegraphCompletionsClient {
         })
 
         if (!isNonSourcegraphProvider) {
-            await this._streamWithCallbacks(params, requestParams, callbacks, signal)
+            if (params.stream === false) {
+                await this._fetchWithCallbacks(params, requestParams, callbacks, signal)
+            } else {
+                await this._streamWithCallbacks(params, requestParams, callbacks, signal)
+            }
         }
 
         for (let i = 0; ; i++) {

vscode/CHANGELOG.md

@@ -8,6 +8,7 @@ This is a log of all notable changes to Cody for VS Code. [Unreleased] changes a
 
 - The [new OpenAI models (OpenAI O1 & OpenAI O1-mini)](https://sourcegraph.com/blog/openai-o1-for-cody) are now available to selected Cody Pro users for early access. [pull/5508](https://github.com/sourcegraph/cody/pull/5508)
 - Cody Pro users can join the waitlist for the new models by clicking the "Join Waitlist" button. [pull/5508](https://github.com/sourcegraph/cody/pull/5508)
+- Chat: Support non-streaming requests.
 
 ### Fixed
 

vscode/src/completions/nodeClient.ts

@@ -29,8 +29,6 @@ import {
 } from '@sourcegraph/cody-shared'
 import { CompletionsResponseBuilder } from '@sourcegraph/cody-shared/src/sourcegraph-api/completions/CompletionsResponseBuilder'
 
-const isTemperatureZero = process.env.CODY_TEMPERATURE_ZERO === 'true'
-
 export class SourcegraphNodeCompletionsClient extends SourcegraphCompletionsClient {
     protected _streamWithCallbacks(
         params: CompletionParameters,
@@ -56,7 +54,7 @@ export class SourcegraphNodeCompletionsClient extends SourcegraphCompletionsClie
                 model: params.model,
             })
 
-            if (isTemperatureZero) {
+            if (this.isTemperatureZero) {
                 params = {
                     ...params,
                     temperature: 0,
@@ -203,17 +201,6 @@ export class SourcegraphNodeCompletionsClient extends SourcegraphCompletionsClie
                         bufferText += str
                         bufferBin = buf
 
-                        // HACK: Handles non-stream request.
-                        // TODO: Implement a function to make and process non-stream requests.
-                        if (params.stream === false) {
-                            const json = JSON.parse(bufferText)
-                            if (json?.completion) {
-                                cb.onChange(json.completion)
-                                cb.onComplete()
-                                return
-                            }
-                        }
-
                         const parseResult = parseEvents(builder, bufferText)
                         if (isError(parseResult)) {
                             logError(
@@ -286,6 +273,68 @@ export class SourcegraphNodeCompletionsClient extends SourcegraphCompletionsClie
             onAbort(signal, () => request.destroy())
         })
     }
+
+    protected async _fetchWithCallbacks(
+        params: CompletionParameters,
+        requestParams: CompletionRequestParameters,
+        cb: CompletionCallbacks,
+        signal?: AbortSignal
+    ): Promise<void> {
+        const { url, serializedParams } = await this.prepareRequest(params, requestParams)
+        const log = this.logger?.startCompletion(params, url.toString())
+        return tracer.startActiveSpan(`POST ${url.toString()}`, async span => {
+            span.setAttributes({
+                fast: params.fast,
+                maxTokensToSample: params.maxTokensToSample,
+                temperature: this.isTemperatureZero ? 0 : params.temperature,
+                topK: params.topK,
+                topP: params.topP,
+                model: params.model,
+            })
+            try {
+                const response = await fetch(url.toString(), {
+                    method: 'POST',
+                    headers: {
+                        'Content-Type': 'application/json',
+                        'Accept-Encoding': 'gzip;q=0',
+                        ...(this.config.accessToken
+                            ? { Authorization: `token ${this.config.accessToken}` }
+                            : null),
+                        ...(customUserAgent ? { 'User-Agent': customUserAgent } : null),
+                        ...this.config.customHeaders,
+                        ...requestParams.customHeaders,
+                        ...getTraceparentHeaders(),
+                    },
+                    body: JSON.stringify(serializedParams),
+                    signal,
+                })
+                if (!response.ok) {
+                    const errorMessage = await response.text()
+                    throw new NetworkError(
+                        {
+                            url: url.toString(),
+                            status: response.status,
+                            statusText: response.statusText,
+                        },
+                        errorMessage,
+                        getActiveTraceAndSpanId()?.traceId
+                    )
+                }
+                const json = await response.json()
+                if (typeof json?.completion === 'string') {
+                    cb.onChange(json.completion)
+                    cb.onComplete()
+                    return
+                }
+                throw new Error('Unexpected response format')
+            } catch (error) {
+                const errorObject = error instanceof Error ? error : new Error(`${error}`)
+                log?.onError(errorObject.message, error)
+                recordErrorToSpan(span, errorObject)
+                cb.onError(errorObject)
+            }
+        })
+    }
 }
 
 function getHeader(value: string | undefined | string[]): string | undefined {