Skip to content

Commit

Permalink
Update dataprep default mosec embedding model in config.py (opea-proj…
Browse files Browse the repository at this point in the history
…ect#435)

Signed-off-by: Tian, Feng <[email protected]>
  • Loading branch information
ftian1 authored Aug 9, 2024
1 parent cafcf1b commit 8f0f2b0
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion comps/dataprep/milvus/config.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@
MILVUS_PORT = int(os.getenv("MILVUS_PORT", 19530))
COLLECTION_NAME = os.getenv("COLLECTION_NAME", "rag_milvus")

MOSEC_EMBEDDING_MODEL = "/root/bce-embedding-base_v1"
MOSEC_EMBEDDING_MODEL = os.environ.get("MOSEC_EMBEDDING_MODEL", "/root/bce-embedding-base_v1")
MOSEC_EMBEDDING_ENDPOINT = os.environ.get("MOSEC_EMBEDDING_ENDPOINT", "")
os.environ["OPENAI_API_BASE"] = MOSEC_EMBEDDING_ENDPOINT
os.environ["OPENAI_API_KEY"] = "Dummy key"

0 comments on commit 8f0f2b0

Please sign in to comment.