RYDE-WORK/aimingmed-ai
1
0
Fork 0
mirror of https://github.com/aimingmed/aimingmed-ai.git synced 2026-02-04 14:19:49 +08:00
Code Issues 6 Packages Projects Releases Wiki Activity

update .env default

Browse Source
This commit is contained in:
leehk 2025-03-07 13:01:39 +08:00
parent 8a8337cc5c
commit 1e242de51d
3 changed files with 17 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
app/docker-compose.yml
View File

@ -3,8 +3,7 @@ version: "3.9"
services:
streamlit:
build: ./streamlit
platform: linux/amd64
ports:
- "8501:8501"
volumes:
- ./llmops/src/rag_cot/chroma_db:/app/llmops/src/rag_cot/chroma_db
- ./llmops/src/rag_cot_evaluation/chroma_db:/app/llmops/src/rag_cot_evaluation/chroma_db

14
app/streamlit/Chatbot.py
View File

@ -16,13 +16,13 @@ torch.classes.__path__ = [os.path.join(torch.__path__[0], torch.classes.__file__
os.environ["TOKENIZERS_PARALLELISM"] = "false"
GEMINI_API_KEY = config("GOOGLE_API_KEY", cast=str)
DEEKSEEK_API_KEY = config("DEEKSEEK_API_KEY", cast=str)
MOONSHOT_API_KEY = config("MOONSHOT_API_KEY", cast=str)
CHAT_MODEL_PROVIDER = config("CHAT_MODEL_PROVIDER", cast=str)
INPUT_CHROMADB_LOCAL = config("INPUT_CHROMADB_LOCAL", cast=str)
EMBEDDING_MODEL = config("EMBEDDING_MODEL", cast=str)
COLLECTION_NAME = config("COLLECTION_NAME", cast=str)
GEMINI_API_KEY = config("GOOGLE_API_KEY", cast=str, default="123456")
DEEKSEEK_API_KEY = config("DEEKSEEK_API_KEY", cast=str, default="123456")
MOONSHOT_API_KEY = config("MOONSHOT_API_KEY", cast=str, default="123456")
CHAT_MODEL_PROVIDER = config("CHAT_MODEL_PROVIDER", cast=str, default="gemini")
INPUT_CHROMADB_LOCAL = config("INPUT_CHROMADB_LOCAL", cast=str, default="../llmops/src/rag_cot_evaluation/chroma_db")
EMBEDDING_MODEL = config("EMBEDDING_MODEL", cast=str, default="paraphrase-multilingual-mpnet-base-v2")
COLLECTION_NAME = config("COLLECTION_NAME", cast=str, default="rag_experiment")
st.title("💬 RAG AI for Medical Guideline")
st.caption(f"🚀 A RAG AI for Medical Guideline powered by {CHAT_MODEL_PROVIDER}")

10
app/streamlit/Dockerfile
View File

@ -2,6 +2,12 @@ FROM python:3.11-slim
WORKDIR /app/streamlit
RUN apt-get update && apt-get install -y \
build-essential \
curl \
software-properties-common \
&& rm -rf /var/lib/apt/lists/*
COPY requirements.txt ./
RUN pip install --no-cache-dir -r requirements.txt
@ -11,4 +17,6 @@ COPY .env .
EXPOSE 8501
ENTRYPOINT ["streamlit", "run", "Chatbot.py"]
HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health
ENTRYPOINT ["streamlit", "run", "Chatbot.py", "--server.port=8501", "--server.address=0.0.0.0"]