Update llm.py

llm.py

@@ -28,9 +28,9 @@ HUGGINGFACE_LITE = {
 
 valid_LLM = [HUGGINGFACE, HUGGINGFACE_LITE]
 
-def setup_model(config: dict, callbacks=None):
+def setup_model(config: dict, API_KEY: str, callbacks=None):
     if config['type'].startswith('huggingface'):
-        llm = HuggingFaceEndpoint(**config['param'])
+        llm = HuggingFaceEndpoint(huggingfacehub_api_token=API_KEY, **config['param'])
         model = ChatHuggingFace(llm=llm, callbacks=callbacks)
     else:
         model = None