Update ChatGoogleGenerativeAI serialization

langchain-ai · Aug 2, 2024 · 5036783 · 5036783
1 parent 668588e
commit 5036783
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/libs/langchain-google-genai/src/chat_models.ts b/libs/langchain-google-genai/src/chat_models.ts
@@ -217,6 +217,14 @@ export class ChatGoogleGenerativeAI
     };
   }
 
+  lc_namespace = ["langchain", "chat_models", "google_genai"];
+
+  get lc_aliases() {
+    return {
+      apiKey: "google_api_key",
+    };
+  }
+
   modelName = "gemini-pro";
 
   model = "gemini-pro";