From 4112fd0b6900452eb99e8428e96884aa8f8bff0b Mon Sep 17 00:00:00 2001
From: Sihan Chen <39623753+Spycsh@users.noreply.github.com>
Date: Thu, 12 Sep 2024 23:51:47 +0800
Subject: [PATCH] Fix megaservice ulimit issue under high concurrency (#797)

---
 ChatQnA/Dockerfile                | 2 ++
 ChatQnA/Dockerfile.guardrails     | 2 ++
 ChatQnA/Dockerfile.no_wrapper     | 2 ++
 ChatQnA/Dockerfile.without_rerank | 2 ++
 4 files changed, 8 insertions(+)

diff --git a/ChatQnA/Dockerfile b/ChatQnA/Dockerfile
index 3439a821e..4ece0783a 100644
--- a/ChatQnA/Dockerfile
+++ b/ChatQnA/Dockerfile
@@ -29,4 +29,6 @@ USER user
 
 WORKDIR /home/user
 
+RUN echo 'ulimit -S -n 999999' >> ~/.bashrc
+
 ENTRYPOINT ["python", "chatqna.py"]
diff --git a/ChatQnA/Dockerfile.guardrails b/ChatQnA/Dockerfile.guardrails
index d36a4de50..12f8ce957 100644
--- a/ChatQnA/Dockerfile.guardrails
+++ b/ChatQnA/Dockerfile.guardrails
@@ -29,4 +29,6 @@ USER user
 
 WORKDIR /home/user
 
+RUN echo 'ulimit -S -n 999999' >> ~/.bashrc
+
 ENTRYPOINT ["python", "chatqna_guardrails.py"]
diff --git a/ChatQnA/Dockerfile.no_wrapper b/ChatQnA/Dockerfile.no_wrapper
index c6adacaee..2cd260122 100644
--- a/ChatQnA/Dockerfile.no_wrapper
+++ b/ChatQnA/Dockerfile.no_wrapper
@@ -31,4 +31,6 @@ USER user
 
 WORKDIR /home/user
 
+RUN echo 'ulimit -S -n 999999' >> ~/.bashrc
+
 ENTRYPOINT ["python", "chatqna_no_wrapper.py"]
diff --git a/ChatQnA/Dockerfile.without_rerank b/ChatQnA/Dockerfile.without_rerank
index 7b6d0ebcd..52fcf31d9 100644
--- a/ChatQnA/Dockerfile.without_rerank
+++ b/ChatQnA/Dockerfile.without_rerank
@@ -29,4 +29,6 @@ USER user
 
 WORKDIR /home/user
 
+RUN echo 'ulimit -S -n 999999' >> ~/.bashrc
+
 ENTRYPOINT ["python", "chatqna_without_rerank.py"]