Spaces:
Sleeping
Sleeping
Vela
commited on
Commit
·
411f074
1
Parent(s):
f0cdbe5
modified docker file
Browse files- Dockerfile +21 -9
- chroma.log +0 -20
- requirements.txt +1 -3
- src/backend/models/__pycache__/embedding_model.cpython-313.pyc +0 -0
- src/backend/models/embedding_model.py +1 -3
Dockerfile
CHANGED
@@ -2,22 +2,34 @@ FROM python:3.11
|
|
2 |
|
3 |
WORKDIR /app
|
4 |
|
5 |
-
|
|
|
6 |
|
7 |
-
|
8 |
-
RUN pip install -
|
|
|
9 |
|
10 |
-
|
|
|
11 |
|
12 |
-
#
|
13 |
RUN mkdir -p /app/logs && chmod -R 777 /app/logs
|
14 |
|
15 |
-
# Install
|
16 |
-
RUN apt-get update && apt-get install -y tmux
|
|
|
|
|
|
|
17 |
|
18 |
# Set Python path
|
19 |
ENV PYTHONPATH="/app/src"
|
20 |
|
21 |
-
#
|
|
|
|
|
|
|
|
|
|
|
22 |
|
23 |
-
|
|
|
|
2 |
|
3 |
WORKDIR /app
|
4 |
|
5 |
+
# Copy requirements first for better caching
|
6 |
+
COPY requirements.txt /app/
|
7 |
|
8 |
+
# Install dependencies and upgrade pip
|
9 |
+
RUN pip install --no-cache-dir --upgrade pip \
|
10 |
+
&& pip install --no-cache-dir -r requirements.txt
|
11 |
|
12 |
+
# Copy remaining application code
|
13 |
+
COPY . /app
|
14 |
|
15 |
+
# Create logs directory with proper permissions
|
16 |
RUN mkdir -p /app/logs && chmod -R 777 /app/logs
|
17 |
|
18 |
+
# Install additional dependencies
|
19 |
+
RUN apt-get update && apt-get install -y tmux curl
|
20 |
+
|
21 |
+
# Ensure the Hugging Face cache is set correctly
|
22 |
+
ENV TRANSFORMERS_CACHE="/app/.cache/huggingface"
|
23 |
|
24 |
# Set Python path
|
25 |
ENV PYTHONPATH="/app/src"
|
26 |
|
27 |
+
# Add a script to manage process control
|
28 |
+
COPY wait-for-it.sh /usr/local/bin/wait-for-it
|
29 |
+
RUN chmod +x /usr/local/bin/wait-for-it
|
30 |
+
|
31 |
+
# Expose ports for both FastAPI (8000) and Streamlit (7860)
|
32 |
+
EXPOSE 8000 7860
|
33 |
|
34 |
+
# Combined startup with better control
|
35 |
+
CMD ["sh", "-c", "tmux new-session -d -s backend 'uvicorn src.backend.main:app --host 0.0.0.0 --port 8000' && /usr/local/bin/wait-for-it localhost:8000 --timeout=30 && streamlit run src/frontend/home.py --server.port 7860 --server.address 0.0.0.0"]
|
chroma.log
DELETED
@@ -1,20 +0,0 @@
|
|
1 |
-
[33mWARNING[0m: [10-03-2025 20:07:32] chroma_server_nofile is not supported on Windows. chroma_server_nofile will not be set.
|
2 |
-
[33mWARNING[0m: [10-03-2025 20:07:51] chroma_server_nofile is not supported on Windows. chroma_server_nofile will not be set.
|
3 |
-
[32mINFO[0m: [10-03-2025 20:07:52] Anonymized telemetry enabled. See https://docs.trychroma.com/telemetry for more information.
|
4 |
-
[36mDEBUG[0m: [10-03-2025 20:07:52] Starting component System
|
5 |
-
[36mDEBUG[0m: [10-03-2025 20:07:52] Starting component OpenTelemetryClient
|
6 |
-
[36mDEBUG[0m: [10-03-2025 20:07:52] Starting component SqliteDB
|
7 |
-
[36mDEBUG[0m: [10-03-2025 20:07:52] Starting component SimpleQuotaEnforcer
|
8 |
-
[36mDEBUG[0m: [10-03-2025 20:07:52] Starting component Posthog
|
9 |
-
[36mDEBUG[0m: [10-03-2025 20:07:52] Starting component SimpleRateLimitEnforcer
|
10 |
-
[36mDEBUG[0m: [10-03-2025 20:07:52] Starting component LocalSegmentManager
|
11 |
-
[36mDEBUG[0m: [10-03-2025 20:07:52] Starting component LocalExecutor
|
12 |
-
[36mDEBUG[0m: [10-03-2025 20:07:52] Starting component SegmentAPI
|
13 |
-
[36mDEBUG[0m: [10-03-2025 20:07:52] Starting component SimpleAsyncRateLimitEnforcer
|
14 |
-
[32mINFO[0m: [10-03-2025 20:07:52] Started server process [[36m17592[0m]
|
15 |
-
[32mINFO[0m: [10-03-2025 20:07:52] Waiting for application startup.
|
16 |
-
[32mINFO[0m: [10-03-2025 20:07:52] Application startup complete.
|
17 |
-
[32mINFO[0m: [10-03-2025 20:07:52] Uvicorn running on [1mhttp://localhost:8000[0m (Press CTRL+C to quit)
|
18 |
-
[32mINFO[0m: [10-03-2025 20:16:05] ::1:57148 - "GET /api/v2/auth/identity HTTP/1.1" 200
|
19 |
-
[32mINFO[0m: [10-03-2025 20:16:05] ::1:57149 - "GET /api/v2/tenants/default_tenant HTTP/1.1" 200
|
20 |
-
[32mINFO[0m: [10-03-2025 20:16:05] ::1:57149 - "GET /api/v2/tenants/default_tenant/databases/default_database HTTP/1.1" 200
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
requirements.txt
CHANGED
@@ -8,6 +8,4 @@ torch
|
|
8 |
torchvision
|
9 |
torchaudio
|
10 |
sentence_transformers
|
11 |
-
groq
|
12 |
-
chromadb
|
13 |
-
uvicorn
|
|
|
8 |
torchvision
|
9 |
torchaudio
|
10 |
sentence_transformers
|
11 |
+
groq
|
|
|
|
src/backend/models/__pycache__/embedding_model.cpython-313.pyc
CHANGED
Binary files a/src/backend/models/__pycache__/embedding_model.cpython-313.pyc and b/src/backend/models/__pycache__/embedding_model.cpython-313.pyc differ
|
|
src/backend/models/embedding_model.py
CHANGED
@@ -1,11 +1,9 @@
|
|
1 |
from sentence_transformers import SentenceTransformer
|
2 |
from utils import logger
|
3 |
-
from fastapi import UploadFile
|
4 |
-
from PIL import Image
|
5 |
|
6 |
logger = logger.get_logger()
|
7 |
|
8 |
-
model = SentenceTransformer("
|
9 |
|
10 |
def get_text_embedding(search_query: str):
|
11 |
try:
|
|
|
1 |
from sentence_transformers import SentenceTransformer
|
2 |
from utils import logger
|
|
|
|
|
3 |
|
4 |
logger = logger.get_logger()
|
5 |
|
6 |
+
model = SentenceTransformer("all-MiniLM-L6-v2")
|
7 |
|
8 |
def get_text_embedding(search_query: str):
|
9 |
try:
|