Update API endpoint embeddings to embed

m3_server.py

@@ -2,7 +2,6 @@ import asyncio
 import os
 import time
 from concurrent.futures import ThreadPoolExecutor
-from pathlib import Path
 from textwrap import dedent
 from typing import List, Tuple, Union
 from uuid import uuid4
@@ -179,10 +178,11 @@ class RequestProcessor:
 
 
 description = dedent(
-    """\
+    r"""
+    embed example:
     ```bash
     curl -X 'POST' \
-      'https://mikeee-baai-m3.hf.space/embeddings/' \
+      'https://mikeee-baai-m3.hf.space/embed/' \
       -H 'accept: application/json' \
       -H 'Content-Type: application/json' \
       -d '{
@@ -190,11 +190,16 @@ description = dedent(
         "string", "string1"
       ]
     }'
-    ```"""
+    ```
+    rerank example:
+    ```bash
+    ...
+    ```
+    """
 )
 
 app = FastAPI(
-    title="baai m3, serving embed and rerank",
+    title="Serving BAAI/bge-m3 embed and rerank",
     # description="Swagger UI at https://mikeee-baai-m3.hf.space/docs",
     description=description,
     version="0.1.0a0",
@@ -230,7 +235,7 @@ async def landing():
     return "Swagger UI at https://mikeee-baai-m3.hf.space/docs"
 
 
-@app.post("/embeddings/", response_model=EmbedResponse)
+@app.post("/embed/", response_model=EmbedResponse)
 async def get_embeddings(request: EmbedRequest):
     embeddings = await processor.process_request(request, "embed")
     return EmbedResponse(embeddings=embeddings)
@@ -244,5 +249,6 @@ async def rerank(request: RerankRequest):
 
 if __name__ == "__main__":
     import uvicorn
+
     print("started")
     uvicorn.run(app, host="0.0.0.0", port=port)