From 978ef781e48ff73a3029bd08a17ca4f98fe1bf74 Mon Sep 17 00:00:00 2001
From: swk777 <swk777@gmail.com>
Date: Mon, 14 Aug 2023 15:13:29 +0800
Subject: [PATCH] Fix OpenAIEmbedding Fails to Retrieve Session from Parameters
 When Not Using Azure

---
 packages/core/src/Embedding.ts | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

diff --git a/packages/core/src/Embedding.ts b/packages/core/src/Embedding.ts
index 89d8fbc27..d2d07c098 100644
--- a/packages/core/src/Embedding.ts
+++ b/packages/core/src/Embedding.ts
@@ -259,11 +259,13 @@ export class OpenAIEmbedding extends BaseEmbedding {
         });
     } else {
       this.apiKey = init?.apiKey ?? undefined;
-      this.session = getOpenAISession({
-        apiKey: this.apiKey,
-        maxRetries: this.maxRetries,
-        timeout: this.timeout,
-      });
+      this.session =
+        init?.session ??
+        getOpenAISession({
+          apiKey: this.apiKey,
+          maxRetries: this.maxRetries,
+          timeout: this.timeout,
+        });
     }
   }
 
-- 
GitLab