Added API key init

NeonGeckoCom · Nov 25, 2023 · f208f1c · f208f1c
1 parent 2c6dff1
commit f208f1c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/neon_llm_palm2/palm2.py b/neon_llm_palm2/palm2.py
@@ -77,6 +77,7 @@ def _system_prompt(self) -> str:
         return self.role
 
     def warmup(self):
+        os.environ["GOOGLE_APPLICATION_CREDENTIALS"] = self.api_key_path
         self.model
         self.embedding