Update Dockerfile
Browse files- Dockerfile +22 -2
Dockerfile
CHANGED
|
@@ -1,3 +1,23 @@
|
|
| 1 |
-
FROM
|
| 2 |
|
| 3 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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"]
|