chore: update Hugging Face API token in .env.sample and handle missin…

…g token in LLM class
Khushiyant · Apr 29, 2024 · 29467f0 · 29467f0
1 parent 7307e88
commit 29467f0
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 1 deletion.
diff --git a/.env.sample b/.env.sample
@@ -1 +1 @@
-API_TOKEN = "YOUR_API_TOKEN"
+HUGGINGFACE_TOKEN = "YOUR_API_TOKEN"
diff --git a/.github/release-drafter.yml → .github/workflows/release-drafter.yml b/.github/release-drafter.yml → .github/workflows/release-drafter.yml
diff --git a/quasarpy/algorithm/llm.py b/quasarpy/algorithm/llm.py
@@ -42,6 +42,8 @@ def _generate_online(self, prompt: str, **kwargs) -> str:
                 token = os.getenv("HUGGINGFACE_TOKEN")
             except KeyError:
                 self.logger.error("Hugging Face API token not found in the environment variables.")
+                with open(".env", "w") as f:
+                    f.write("HUGGINGFACE_TOKEN=YOUR_API_TOKEN")
                 raise
 
             client = huggingface_hub.InferenceClient(token=token)