secret mapping corrected

digiteinfotech · Dec 13, 2024 · 9feba66 · 9feba66
1 parent fc6a5e9
commit 9feba66
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/kairon/nlu/classifiers/llm.py b/kairon/nlu/classifiers/llm.py
@@ -48,6 +48,7 @@ def __init__(
         data: Optional[Dict[Text, Any]] = None,
     ) -> None:
         """Construct a new intent classifiers using the OpenAI Completion framework."""
+        self.secret = {}
         self.component_config = config
         self._model_storage = model_storage
         self._resource = resource
@@ -82,7 +83,7 @@ def load_api_key(self, bot_id: Text):
         if bot_id:
             from kairon.shared.admin.processor import Sysadmin
             llm_secret = Sysadmin.get_llm_secret("openai", bot_id)
-            self.secret = llm_secret.get('api_key')
+            self.secret = llm_secret
         elif os.environ.get("LLM_API_KEY"):
             self.secret = {'api_key': os.environ.get("LLM_API_KEY")}
         else: