diff --git a/.env.example b/.env.example index d498e7c45..6d986ae91 100644 --- a/.env.example +++ b/.env.example @@ -170,3 +170,15 @@ INIT_EMBEDDING_MODEL_ID=builtin:nomic-embed-text:768 # Rerank模型的API密钥,如果需要身份验证,可以设置 # 支持第三方模型服务的API密钥 # INIT_RERANK_MODEL_API_KEY=your_rerank_model_api_key + +# VLM_MODEL_NAME 使用的多模态模型名称 +# 用于解析图片数据 +# VLM_MODEL_NAME=your_vlm_model_name + +# VLM_MODEL_BASE_URL 使用的多模态模型访问地址 +# 支持第三方模型服务的URL +# VLM_MODEL_BASE_URL=your_vlm_model_base_url + +# VLM_MODEL_API_KEY 使用的多模态模型API密钥 +# 支持第三方模型服务的API密钥 +# VLM_MODEL_API_KEY=your_vlm_model_api_key \ No newline at end of file diff --git a/docker-compose.yml b/docker-compose.yml index 84e8e71fd..9521ae6f3 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -100,6 +100,7 @@ services: - COS_ENABLE_OLD_DOMAIN=${COS_ENABLE_OLD_DOMAIN} - VLM_MODEL_BASE_URL=${VLM_MODEL_BASE_URL} - VLM_MODEL_NAME=${VLM_MODEL_NAME} + - VLM_MODEL_API_KEY=${VLM_MODEL_API_KEY} networks: - WeKnora-network restart: unless-stopped diff --git a/services/docreader/src/parser/caption.py b/services/docreader/src/parser/caption.py index 7c9cae0f2..6d7f8b3dc 100644 --- a/services/docreader/src/parser/caption.py +++ b/services/docreader/src/parser/caption.py @@ -177,6 +177,7 @@ class Caption: return self.completion_url = os.getenv("VLM_MODEL_BASE_URL") + "/v1/chat/completions" self.model = os.getenv("VLM_MODEL_NAME") + self.api_key = os.getenv("VLM_MODEL_API_KEY") logger.info( f"Service configured with model: {self.model}, endpoint: {self.completion_url}" ) @@ -218,6 +219,8 @@ class Caption: "Cache-Control": "no-cache", "Connection": "keep-alive", } + if self.api_key: + headers["Authorization"] = f"Bearer {self.api_key}" try: logger.info(f"Sending request to Caption API with model: {self.model}")