diff --git a/Dockerfile b/Dockerfile index 9130e60c1..49bb29445 100644 --- a/Dockerfile +++ b/Dockerfile @@ -16,9 +16,6 @@ ARG DEBUG # Set environment variable based on the build argument ENV DEBUG=${DEBUG} -# if you located in China, you can use aliyun mirror to speed up -#RUN sed -i 's@deb.debian.org@mirrors.ustc.edu.cn@g' /etc/apt/sources.list.d/debian.sources - # Install system dependencies RUN apt-get update && apt-get install -y \ gcc \ diff --git a/cognee/infrastructure/databases/vector/embeddings/LiteLLMEmbeddingEngine.py b/cognee/infrastructure/databases/vector/embeddings/LiteLLMEmbeddingEngine.py index 7a1306bf8..558b11538 100644 --- a/cognee/infrastructure/databases/vector/embeddings/LiteLLMEmbeddingEngine.py +++ b/cognee/infrastructure/databases/vector/embeddings/LiteLLMEmbeddingEngine.py @@ -134,9 +134,7 @@ class LiteLLMEmbeddingEngine(EmbeddingEngine): litellm.aembedding( model=self.model, input=text, - api_key=self.api_key - if self.api_key and self.api_key.strip() != "" - else "EMPTY", + api_key=self.api_key, api_base=self.endpoint, api_version=self.api_version, ),