From babc1c44b9bf886faa931951bb9e414a5fa4786d Mon Sep 17 00:00:00 2001
From: geens <niels.geens@gmail.com>
Date: Mon, 11 Nov 2024 13:25:39 +0100
Subject: [PATCH] Renamed hf-api-token to api-token

---
 sia/__main__.py |  2 +-
 sia/config.py   | 12 ++++++------
 2 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/sia/__main__.py b/sia/__main__.py
index 42a847b..8338bcc 100644
--- a/sia/__main__.py
+++ b/sia/__main__.py
@@ -48,7 +48,7 @@ class Main:
             case "hf":
                 self._llm = HfLlmEngine(
                     model_id=self._config.model,
-                    api_token=self._config.hf_api_token
+                    api_token=self._config.api_token
                 )
             case "test":
                 self._llm = TestLLM()
diff --git a/sia/config.py b/sia/config.py
index e41a162..d911a7b 100644
--- a/sia/config.py
+++ b/sia/config.py
@@ -64,10 +64,10 @@ class Config:
             help='LLM engine (default: local, env: SIA_LLM_ENGINE)'
         )
         parser.add_argument(
-            '--hf-api-token',
+            '--api-token',
             type=str,
-            default=os.getenv('SIA_HF_API_TOKEN'),
-            help='Hugging Face access token (env: SIA_HF_API_TOKEN)'
+            default=os.getenv('SIA_API_TOKEN'),
+            help='API access token (env: SIA_API_TOKEN)'
         )
         parser.add_argument(
             '--model',
@@ -127,9 +127,9 @@ class Config:
         return self.args.llm_engine
     
     @property
-    def hf_api_token(self) -> Optional[str]:
-        """Hugging Face access token."""
-        return self.args.hf_api_token
+    def api_token(self) -> Optional[str]:
+        """API access token."""
+        return self.args.api_token
 
     @property
     def model(self) -> str: