File tree Expand file tree Collapse file tree 3 files changed +5
-2
lines changed
base/src/main/java/com/tinyengine/it/rag/config Expand file tree Collapse file tree 3 files changed +5
-2
lines changed Original file line number Diff line number Diff line change @@ -15,6 +15,7 @@ WORKDIR /app
1515COPY --from=build /app/app/target/tiny-engine-app-*.jar /app/tiny-engine-app.jar
1616COPY --from=build /app/base/target/tiny-engine-base-*.jar /app/tiny-engine-base.jar
1717COPY --from=build /app/documents /app/documents
18+ COPY --from=build /app/all-MiniLM-L6-v2 /app/all-MiniLM-L6-v2
1819# 设置环境变量
1920ENV ACCESS_KEY_ID=""
2021ENV ACCESS_KEY_SECRET=""
Original file line number Diff line number Diff 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 ;
Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments