chaton2api1

Paused

App Files Files Community

smgc commited on Nov 15, 2024

Commit

641eced

verified ·

1 Parent(s): 4e08475

Update main.py

Browse files

Files changed (1) hide show

main.py +50 -15

main.py CHANGED Viewed

@@ -137,21 +137,56 @@ def verify_api_key(credentials: HTTPAuthorizationCredentials = Security(security
     return credentials.credentials
 # 根路径GET请求处理
-@app.get("/", response_class=HTMLResponse)
-async def read_root():
-    """返回欢迎页面"""
-    html_content = """
-    <html>
-        <head>
-            <title>Welcome to API</title>
-        </head>
-        <body>
-            <h1>Welcome to API</h1>
-            <p>This API is used to interact with the ChatGPT model. You can send messages to the model and receive responses.</p>
-        </body>
-    </html>
-    """
-    return HTMLResponse(content=html_content, status_code=200)
 # 聊天完成处理
 @app.post("/ai/v1/chat/completions")

     return credentials.credentials
 # 根路径GET请求处理
+@app.get("/")
+async def root():
+    return JSONResponse(content={
+        "service": "AI Chat Completion Proxy",
+        "usage": {
+            "endpoint": "/ai/v1/chat/completions",
+            "method": "POST",
+            "headers": {
+                "Content-Type": "application/json",
+                "Authorization": "Bearer YOUR_API_KEY"
+            },
+            "body": {
+                "model": "One of: " + ", ".join(MODELS),
+                "messages": [
+                    {"role": "system", "content": "You are a helpful assistant."},
+                    {"role": "user", "content": "Hello, who are you?"}
+                ],
+                "stream": False,
+                "temperature": 0.7,
+                "max_tokens": 8000
+            }
+        },
+        "availableModels": MODELS,
+        "endpoints": {
+            "/ai/v1/chat/completions": "Chat completion endpoint",
+            "/ai/v1/images/generations": "Image generation endpoint",
+            "/ai/v1/models": "List available models"
+        },
+        "note": "Replace YOUR_API_KEY with your actual API key."
+    })
+# 返回模型列表
+@app.get("/ai/v1/models")
+async def list_models(api_key: str = Depends(verify_api_key)):
+    """返回可用模型列表。"""
+    models = [
+        {
+            "id": model,
+            "object": "model",
+            "created": int(time.time()),
+            "owned_by": "chaton",
+            "permission": [],
+            "root": model,
+            "parent": None,
+        } for model in MODELS
+    ]
+    return JSONResponse(content={
+        "object": "list",
+        "data": models
+    })
 # 聊天完成处理
 @app.post("/ai/v1/chat/completions")