groq[patch]: set api key attr (#6864)

Co-authored-by: jacoblee93 <jacoblee93@gmail.com>
langchain-ai · Sep 23, 2024 · 6647485 · 6647485
1 parent 8fe60ef
commit 6647485
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/libs/langchain-groq/src/chat_models.ts b/libs/langchain-groq/src/chat_models.ts
@@ -660,6 +660,8 @@ export class ChatGroq extends BaseChatModel<
 
   streaming = false;
 
+  apiKey?: string;
+
   static lc_name() {
     return "ChatGroq";
   }
@@ -690,6 +692,7 @@ export class ChatGroq extends BaseChatModel<
       apiKey,
       dangerouslyAllowBrowser: true,
     });
+    this.apiKey = apiKey;
     this.temperature = fields?.temperature ?? this.temperature;
     this.modelName = fields?.model ?? fields?.modelName ?? this.model;
     this.model = this.modelName;

diff --git a/libs/langchain-groq/src/tests/chat_models.test.ts b/libs/langchain-groq/src/tests/chat_models.test.ts
@@ -10,3 +10,11 @@ test("Serialization", () => {
     `{"lc":1,"type":"constructor","id":["langchain","chat_models","groq","ChatGroq"],"kwargs":{"api_key":{"lc":1,"type":"secret","id":["GROQ_API_KEY"]}}}`
   );
 });
+
+test("Serialization with no params", () => {
+  process.env.GROQ_API_KEY = "foo";
+  const model = new ChatGroq();
+  expect(JSON.stringify(model)).toEqual(
+    `{"lc":1,"type":"constructor","id":["langchain","chat_models","groq","ChatGroq"],"kwargs":{"api_key":{"lc":1,"type":"secret","id":["GROQ_API_KEY"]}}}`
+  );
+});