Skip to content

Commit 8bc8564

Browse files
committed
fix: modeify dockerfile
1 parent 592f754 commit 8bc8564

File tree

3 files changed

+5
-2
lines changed

3 files changed

+5
-2
lines changed

Dockerfile

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@ WORKDIR /app
1515
COPY --from=build /app/app/target/tiny-engine-app-*.jar /app/tiny-engine-app.jar
1616
COPY --from=build /app/base/target/tiny-engine-base-*.jar /app/tiny-engine-base.jar
1717
COPY --from=build /app/documents /app/documents
18+
COPY --from=build /app/all-MiniLM-L6-v2 /app/all-MiniLM-L6-v2
1819
# 设置环境变量
1920
ENV ACCESS_KEY_ID=""
2021
ENV ACCESS_KEY_SECRET=""

base/src/main/java/com/tinyengine/it/rag/config/RAGConfig.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -33,8 +33,8 @@ public class RAGConfig {
3333
// Chroma 配置
3434
private String chromaBaseUrl = System.getenv("CHROMA_BASE_URL");
3535
private String chromaCollectionName = "tinyengine_documents";
36-
private String modelPath = "./all-MiniLM-L6-v2/model.onnx";
37-
private String tokenizerPath = "./all-MiniLM-L6-v2/tokenizer.json";
36+
private String modelPath = System.getenv("MODEL_PATH");
37+
private String tokenizerPath = System.getenv("TOKENIZER_PATH");
3838

3939
// 连接配置
4040
private int timeoutSeconds = 30;

docker-compose.yml

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,8 @@ services:
4141
- tiny-engine-rag
4242
environment:
4343
- CHROMA_BASE_URL=http://tiny-engine-rag:8000
44+
- MODEL_PATH=/app/all-MiniLM-L6-v2/model.onnx
45+
- TOKENIZER_PATH=/app/all-MiniLM-L6-v2/tokenizer.json
4446

4547
tiny-engine:
4648
image: tiny-engine

0 commit comments

Comments
 (0)