update
Browse files
app.py
CHANGED
@@ -28,11 +28,11 @@ except Exception as e:
|
|
28 |
logger.error(f"Failed to load model: {e}")
|
29 |
raise HTTPException(status_code=500, detail="Model loading failed")
|
30 |
|
31 |
-
class EmbeddingRequest(BaseModel):
|
32 |
-
|
33 |
|
34 |
@app.post("/v1/embeddings")
|
35 |
-
async def embeddings(request
|
36 |
# async def embeddings(request: EmbeddingRequest):
|
37 |
# logger.info("Received request for embeddings")
|
38 |
# return '2222222222'
|
@@ -53,7 +53,7 @@ async def embeddings(request: EmbeddingRequest, authorization: str = Depends(che
|
|
53 |
|
54 |
# Calculate embeddings
|
55 |
# embeddings = model.encode(input_text)
|
56 |
-
|
57 |
|
58 |
# Format the embeddings in OpenAI compatible format
|
59 |
data = {
|
|
|
28 |
logger.error(f"Failed to load model: {e}")
|
29 |
raise HTTPException(status_code=500, detail="Model loading failed")
|
30 |
|
31 |
+
# class EmbeddingRequest(BaseModel):
|
32 |
+
# input: str = Field(..., min_length=1, max_length=1000)
|
33 |
|
34 |
@app.post("/v1/embeddings")
|
35 |
+
async def embeddings(request, authorization: str = Depends(check_authorization)):
|
36 |
# async def embeddings(request: EmbeddingRequest):
|
37 |
# logger.info("Received request for embeddings")
|
38 |
# return '2222222222'
|
|
|
53 |
|
54 |
# Calculate embeddings
|
55 |
# embeddings = model.encode(input_text)
|
56 |
+
embeddings_1 = model.encode(input_texts, normalize_embeddings=True)
|
57 |
|
58 |
# Format the embeddings in OpenAI compatible format
|
59 |
data = {
|