@celljs/ai-openai 3.6.4 → 3.6.5

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@celljs/ai-openai",
-  "version": "3.6.4",
+  "version": "3.6.5",
   "description": "OpenAI models support",
   "keywords": [
     "cell-component"
@@ -29,18 +29,18 @@
     "watch": "cell-component watch"
   },
   "dependencies": {
-    "@celljs/ai-core": "3.6.4",
-    "@celljs/core": "3.6.4",
-    "@celljs/http": "3.6.4",
+    "@celljs/ai-core": "3.6.5",
+    "@celljs/core": "3.6.5",
+    "@celljs/http": "3.6.5",
     "class-transformer": "^0.5.1",
     "rxjs": "^6.6.0",
     "tslib": "^2.8.0"
   },
   "devDependencies": {
-    "@celljs/component": "3.6.4"
+    "@celljs/component": "3.6.5"
   },
   "publishConfig": {
     "access": "public"
   },
-  "gitHead": "136649727fd8ca2c3c982dce6c5066805c88ac56"
+  "gitHead": "98e408d4578e5a0cade51e487e3d6b6a1876b5d8"
 }

package/src/common/model/openai-chat-model.ts

@@ -258,7 +258,8 @@ export class OpenAIChatModel implements ChatModel {
 
     protected async internalCall(prompt: Prompt, previousChatResponse?: ChatResponse): Promise<ChatResponse> {
         const request = this.createRequest(prompt, false);
-        const completionEntity = await this.chatApi.chat(request);
+        const httpHeaders = prompt.options?.httpHeaders;
+        const completionEntity = await this.chatApi.chat(request, httpHeaders);
         const chatCompletion = completionEntity.body;
         if (!chatCompletion) {
             this.logger.warn(`No chat completion returned for prompt: ${prompt}`);
@@ -322,8 +323,8 @@ export class OpenAIChatModel implements ChatModel {
             this.logger.warn('Audio parameters are not supported for streaming requests. Removing audio parameters.');
             throw new IllegalArgumentError('Audio parameters are not supported for streaming requests.');
         }
-
-        const response = await this.chatApi.streamingChat(request);
+        const httpHeaders = prompt.options?.httpHeaders;
+        const response = await this.chatApi.streamingChat(request, httpHeaders);
 
         // For chunked responses, only the first chunk contains the choice role.
         // Subsequent chunks contain the choices.

package/src/common/model/openai-chat-options.ts

@@ -83,6 +83,7 @@ export class OpenAIChatOptions implements FunctionCallingOptions {
      * Whether to return log probabilities of the output tokens or not. If true, returns the log probabilities
      * of each output token returned in the 'content' of 'message'.
      */
+    @Expose()
     logprobs?: boolean;
 
     /**