Updated reqs
Browse files- app.py +1 -1
- climateqa/engine/retriever.py +1 -1
- requirements.txt +2 -1
- setup.py +1 -0
- test +32 -0
app.py
CHANGED
@@ -144,7 +144,7 @@ async def chat(query,history,audience,sources,reports):
|
|
144 |
reports = []
|
145 |
|
146 |
llm = get_llm(max_tokens = 1024,temperature = 0.0)
|
147 |
-
retriever = ClimateQARetriever(vectorstore=vectorstore,sources = sources,reports = reports,k_summary = 3,k_total = 10,threshold=0.
|
148 |
rag_chain = make_rag_chain(retriever,llm)
|
149 |
|
150 |
source_string = ""
|
|
|
144 |
reports = []
|
145 |
|
146 |
llm = get_llm(max_tokens = 1024,temperature = 0.0)
|
147 |
+
retriever = ClimateQARetriever(vectorstore=vectorstore,sources = sources,reports = reports,k_summary = 3,k_total = 10,threshold=0.7)
|
148 |
rag_chain = make_rag_chain(retriever,llm)
|
149 |
|
150 |
source_string = ""
|
climateqa/engine/retriever.py
CHANGED
@@ -13,7 +13,7 @@ class ClimateQARetriever(BaseRetriever):
|
|
13 |
vectorstore:VectorStore
|
14 |
sources:list = ["IPCC","IPBES"]
|
15 |
reports:list = []
|
16 |
-
threshold:float = 0.
|
17 |
k_summary:int = 3
|
18 |
k_total:int = 10
|
19 |
namespace:str = "vectors"
|
|
|
13 |
vectorstore:VectorStore
|
14 |
sources:list = ["IPCC","IPBES"]
|
15 |
reports:list = []
|
16 |
+
threshold:float = 0.6
|
17 |
k_summary:int = 3
|
18 |
k_total:int = 10
|
19 |
namespace:str = "vectors"
|
requirements.txt
CHANGED
@@ -5,4 +5,5 @@ azure-storage-blob
|
|
5 |
python-dotenv==1.0.0
|
6 |
langchain==0.0.333
|
7 |
pinecone-client==2.2.4
|
8 |
-
sentence-transformers
|
|
|
|
5 |
python-dotenv==1.0.0
|
6 |
langchain==0.0.333
|
7 |
pinecone-client==2.2.4
|
8 |
+
sentence-transformers
|
9 |
+
huggingface-hub
|
setup.py
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
print("yoooooo")
|
test
ADDED
@@ -0,0 +1,32 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
FROM python:3.10
|
2 |
+
|
3 |
+
WORKDIR /src
|
4 |
+
|
5 |
+
COPY requirements.txt .
|
6 |
+
|
7 |
+
RUN pip install --no-cache-dir -r requirements.txt
|
8 |
+
|
9 |
+
# Set up a new user named "user" with user ID 1000
|
10 |
+
RUN useradd -m -u 1000 user
|
11 |
+
# Switch to the "user" user
|
12 |
+
USER user
|
13 |
+
# Set home to the user's home directory
|
14 |
+
ENV HOME=/home/user \
|
15 |
+
PATH=/home/user/.local/bin:$PATH \
|
16 |
+
PYTHONPATH=$HOME/app \
|
17 |
+
PYTHONUNBUFFERED=1 \
|
18 |
+
GRADIO_ALLOW_FLAGGING=never \
|
19 |
+
GRADIO_NUM_PORTS=1 \
|
20 |
+
GRADIO_SERVER_NAME=0.0.0.0 \
|
21 |
+
GRADIO_THEME=huggingface \
|
22 |
+
SYSTEM=spaces
|
23 |
+
|
24 |
+
# Set the working directory to the user's home directory
|
25 |
+
WORKDIR $HOME/app
|
26 |
+
|
27 |
+
# Copy the current directory contents into the container at $HOME/app setting the owner to the user
|
28 |
+
COPY --chown=user . $HOME/app
|
29 |
+
|
30 |
+
CMD ["python","setup.py"]
|
31 |
+
|
32 |
+
CMD ["python", "app.py"]
|