Update app.py

app.py

@@ -94,18 +94,35 @@ async def get_embeddings(request: EmbeddingRequest, credentials: HTTPAuthorizati
     # embeddings = [model.encode(text) for text in request.input]
     print(request.json())
     embeddings = query(request.input)
+    prompt_tokens = sum(len(text.split()) for text in request.input)
+    total_tokens = sum(num_tokens_from_string(text) for text in request.input)
+
+    response = {
+        "data": [
+            {
+                "embedding": embedding,
+                "index": index,
+                "object": "embedding"
+            } for index, embedding in enumerate(embeddings)
+        ],
+        "model": request.model,
+        "object": "list",
+        "usage": {
+            "prompt_tokens": prompt_tokens,
+            "total_tokens": total_tokens,
+        }
+    }
 
+    return response
 
-    # 如果嵌入向量的维度不为1536，则使用插值法扩展至1536维度
-    # embeddings = [interpolate_vector(embedding, 1536) if len(embedding) < 1536 else embedding for embedding in embeddings]
-    # 如果嵌入向量的维度不为1536，则使用特征扩展法扩展至1536维度
-    #embeddings = [expand_features(embedding, 1536) if len(embedding) < 1536 else embedding for embedding in embeddings]
+@app.post("/v1/embeddings", response_model=EmbeddingResponse)
+async def get_embeddings(request: EmbeddingRequest, credentials: HTTPAuthorizationCredentials = Depends(security)):
+    if credentials.credentials != sk_key:
+        raise HTTPException(
+            status_code=status.HTTP_401_UNAUTHORIZED,
+            detail="Invalid authorization code",
+        )
 
-    # Min-Max normalization
-    # embeddings = [(embedding - np.min(embedding)) / (np.max(embedding) - np.min(embedding)) if np.max(embedding) != np.min(embedding) else embedding for embedding in embeddings]
-    #embeddings = [embedding / np.linalg.norm(embedding) for embedding in embeddings]
-    # 将numpy数组转换为列表
-    #embeddings = [embedding.tolist() for embedding in embeddings]
     prompt_tokens = sum(len(text.split()) for text in request.input)
     total_tokens = sum(num_tokens_from_string(text) for text in request.input)
 
@@ -127,6 +144,5 @@ async def get_embeddings(request: EmbeddingRequest, credentials: HTTPAuthorizati
 
     return response
 
-
 if __name__ == "__main__":
     uvicorn.run("app:app", host='0.0.0.0', port=7860, workers=2)