krstakis commited on
Commit
fd9fa1c
·
1 Parent(s): dc450dc

trying different approach without multistaging docker

Browse files
Dockerfile CHANGED
@@ -1,22 +1,33 @@
1
- FROM python:3.9 as core
2
-
3
- COPY ./core/requirements.txt ./requirements.txt
4
- RUN pip install -r requirements.txt
5
-
6
- WORKDIR /app
7
-
8
- COPY ./core .
9
- RUN python ./initialization.py
10
-
11
- FROM python:3.9
12
-
13
- COPY ./api/requirements.txt ./requirements.txt
 
 
 
 
 
 
 
 
 
 
 
 
14
  RUN pip install -r requirements.txt
15
 
16
  WORKDIR /app
17
- COPY ./api .
18
 
19
- COPY --from=core /app/engine.pickle /app/engine.pickle
20
 
21
  EXPOSE 9999
22
- ENTRYPOINT ["python", "service_manager.py"]
 
1
+ # FROM python:3.9 as core
2
+ #
3
+ # COPY ./core/requirements.txt ./requirements.txt
4
+ # RUN pip install -r requirements.txt
5
+ #
6
+ # WORKDIR /app
7
+ #
8
+ # COPY ./core .
9
+ # RUN python ./initialization.py
10
+ #
11
+ # FROM python:3.9
12
+ #
13
+ # COPY ./api/requirements.txt ./requirements.txt
14
+ # RUN pip install -r requirements.txt
15
+ #
16
+ # WORKDIR /app
17
+ # COPY ./api .
18
+ #
19
+ # COPY --from=core /app/engine.pickle /app/engine.pickle
20
+ #
21
+ # EXPOSE 9999
22
+ # ENTRYPOINT ["python", "service_manager.py"]
23
+ FROM python:3.9 as build
24
+
25
+ COPY ./prompt_search_engine/requirements.txt ./requirements.txt
26
  RUN pip install -r requirements.txt
27
 
28
  WORKDIR /app
 
29
 
30
+ COPY ./prompt_search_engine .
31
 
32
  EXPOSE 9999
33
+ ENTRYPOINT ["python", "run.py"]
api/web_server.py CHANGED
@@ -1,7 +1,7 @@
1
  import dill
2
  from fastapi import FastAPI, HTTPException
3
  from pydantic import BaseModel
4
- from search_engine import PromptSearchEngine
5
 
6
 
7
  class Query(BaseModel):
 
1
  import dill
2
  from fastapi import FastAPI, HTTPException
3
  from pydantic import BaseModel
4
+ from core.search_engine import PromptSearchEngine
5
 
6
 
7
  class Query(BaseModel):
core/initialization.py CHANGED
@@ -1,6 +1,6 @@
1
  import dill
2
  from data.dataset import PromptDataset
3
- from search_engine import PromptSearchEngine
4
 
5
 
6
  def run():
 
1
  import dill
2
  from data.dataset import PromptDataset
3
+ from core.search_engine import PromptSearchEngine
4
 
5
 
6
  def run():
core/search_engine.py CHANGED
@@ -1,7 +1,7 @@
1
  from typing import List, Sequence, Tuple
2
  import numpy as np
3
  import faiss
4
- from vectorizer import Vectorizer
5
 
6
 
7
  class PromptSearchEngine(object):
 
1
  from typing import List, Sequence, Tuple
2
  import numpy as np
3
  import faiss
4
+ from core.vectorizer import Vectorizer
5
 
6
 
7
  class PromptSearchEngine(object):
requirements.txt ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ datasets==2.20.0
2
+ dill==0.3.8
3
+ faiss_cpu==1.8.0.post1
4
+ fastapi==0.111.1
5
+ pydantic==2.8.2
6
+ sentence_transformers==3.0.1
7
+ uvicorn==0.30.3
run.py ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ import core.initialization as init
2
+ import api.service_manager as srv
3
+
4
+ init.run()
5
+ srv.run()
6
+
7
+