Merge branch 'feat/huggingface-embedding-support' into deploy/dev

langgenius · Sep 20, 2023 · deb4632 · deb4632
2 parents 2694d4c + 10870a7
commit deb4632
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/api/core/third_party/langchain/embeddings/huggingface_hub_embedding.py b/api/core/third_party/langchain/embeddings/huggingface_hub_embedding.py
@@ -29,7 +29,7 @@ def validate_environment(cls, values: Dict) -> Dict:
             values, "huggingfacehub_api_token", "HUGGINGFACEHUB_API_TOKEN"
         )
 
-        values['client'] = InferenceClient(values['huggingfacehub_api_token'])
+        values['client'] = InferenceClient(token=values['huggingfacehub_api_token'])
 
         return values