Skip to content

Commit b2ce522

Browse files
committed
feat: modify ragConfig
1 parent 864256f commit b2ce522

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

Dockerfile

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@ ENV ACCESS_KEY_SECRET=""
2020
ENV INDEX_ID=""
2121
ENV WORK_SPACE_ID=""
2222
ENV FOLDER_PATH="/app/documents"
23+
ENV CHROMA_BASE_URL="http://tiny-engine-rag:8000"
2324
# 替换为自己的域名接口路径
2425
ENV TINY_ENGINE_URL="https://agent.opentiny.design/material-center/api/resource/download"
2526
ENTRYPOINT ["java", "-jar", "tiny-engine-app.jar", "--spring.profiles.active=alpha"]

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ public class RAGConfig {
3131
private double minScore = 0.7;
3232

3333
// Chroma 配置
34-
private String chromaBaseUrl = "http://localhost:8000";
34+
private String chromaBaseUrl = System.getenv("CHROMA_BASE_URL");
3535
private String chromaCollectionName = "tinyengine_documents";
3636
private String modelPath = "./all-MiniLM-L6-v2/model.onnx";
3737
private String tokenizerPath = "./all-MiniLM-L6-v2/tokenizer.json";

0 commit comments

Comments
 (0)