lydgs commited on
Commit
ff017f1
·
verified ·
1 Parent(s): ccc2dd0

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +22 -2
Dockerfile CHANGED
@@ -1,3 +1,23 @@
1
- FROM gzzhongqi/geminicli2api:latest
2
 
3
- EXPOSE 7860
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ FROM python:3.10-slim
2
 
3
+ WORKDIR /app
4
+
5
+ RUN pip install fastapi uvicorn google-generativeai
6
+
7
+ # 创建代理服务 main.py
8
+ RUN echo 'from fastapi import FastAPI, Request' > main.py && \
9
+ echo ' from fastapi.responses import JSONResponse' >> main.py && \
10
+ echo ' import google.generativeai as genai' >> main.py && \
11
+ echo ' import os, uvicorn' >> main.py && \
12
+ echo ' app = FastAPI()' >> main.py && \
13
+ echo ' GOOGLE_API_KEY = os.environ["GOOGLE_API_KEY"]' >> main.py && \
14
+ echo ' genai.configure(api_key=GOOGLE_API_KEY)' >> main.py && \
15
+ echo ' @app.get("/v1/models")' >> main.py && \
16
+ echo ' @app.post("/v1/embeddings")' >> main.py && \
17
+ echo ' @app.post("/v1/chat/completions")' >> main.py && \
18
+ echo ' async def catch_all(request: Request):' >> main.py && \
19
+ echo ' return JSONResponse({"message": "Gemini Proxy is running"})' >> main.py && \
20
+ echo ' if __name__ == "__main__":' >> main.py && \
21
+ echo ' uvicorn.run(app, host="0.0.0.0", port=7860)' >> main.py
22
+
23
+ CMD ["python", "main.py"]