From cdbca7ec6236854098ca9da339e35cce1dc19ace Mon Sep 17 00:00:00 2001
From: Yuzhong Zhang <141388234+BetterAndBetterII@users.noreply.github.com>
Date: Mon, 3 Mar 2025 13:55:09 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96dockerfile=EF=BC=8C=E5=A2=9E?=
 =?UTF-8?q?=E5=8A=A0docker-compose=EF=BC=8Casync=20openai?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 Dockerfile                          | 4 +++-
 app/services/openai_chat_service.py | 6 +++---
 docker-compose.yml                  | 9 +++++++++
 3 files changed, 15 insertions(+), 4 deletions(-)
 create mode 100644 docker-compose.yml

diff --git a/Dockerfile b/Dockerfile
index 7f7b119..a1c6213 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -3,16 +3,18 @@ FROM python:3.10-slim
 WORKDIR /app
 
 # 复制所需文件到容器中
-COPY ./app /app/app
 COPY ./requirements.txt /app
 
 RUN pip install --no-cache-dir -r requirements.txt
+COPY ./app /app/app
 ENV API_KEYS='["your_api_key_1"]'
 ENV ALLOWED_TOKENS='["your_token_1"]'
 ENV BASE_URL=https://generativelanguage.googleapis.com/v1beta
 ENV TOOLS_CODE_EXECUTION_ENABLED=true
 ENV MODEL_SEARCH='["gemini-2.0-flash-exp"]'
 
+COPY ./api_keys.txt /app
+
 # Expose port
 EXPOSE 8000
 
diff --git a/app/services/openai_chat_service.py b/app/services/openai_chat_service.py
index 8294756..eefc1b5 100644
--- a/app/services/openai_chat_service.py
+++ b/app/services/openai_chat_service.py
@@ -137,13 +137,13 @@ class OpenAIChatService:
 
         if request.stream:
             return self._handle_stream_completion(request.model, payload, api_key)
-        return self._handle_normal_completion(request.model, payload, api_key)
+        return await self._handle_normal_completion(request.model, payload, api_key)
 
-    def _handle_normal_completion(
+    async def _handle_normal_completion(
             self, model: str, payload: Dict[str, Any], api_key: str
     ) -> Dict[str, Any]:
         """处理普通聊天完成"""
-        response = self.api_client.generate_content(payload, model, api_key)
+        response = await self.api_client.generate_content(payload, model, api_key)
         return self.response_handler.handle_response(
             response, model, stream=False, finish_reason="stop"
         )
diff --git a/docker-compose.yml b/docker-compose.yml
new file mode 100644
index 0000000..965b3fa
--- /dev/null
+++ b/docker-compose.yml
@@ -0,0 +1,9 @@
+version: '3'
+
+services:
+  gemini-balance:
+    build: .
+    ports:
+      - "8000:8000"
+    env_file:
+      - .env