from fastapi import FastAPI,Request
from langchain.embeddings import HuggingFaceEmbeddings
from sentence_transformers import SentenceTransformer

app = FastAPI()


@app.get("/")
def read_root(query):
    return {"Hello": "World!" + query}


@app.post("/your-endpoint")
async def process_post_request(request: Request):
    data = await request.json()
    # 在这里可以对接收到的数据进行处理
    # 例如，可以通过data["key"]来访问请求体中的参数
    print(data)
    embeddings = HuggingFaceEmbeddings(model_name="GanymedeNil/text2vec-large-chinese",model_kwargs={'device': "cpu"})
    result= embeddings.embed_query(data['data'])
    return result


if __name__ == '__main__':
    embeddings = HuggingFaceEmbeddings(model_name="GanymedeNil/text2vec-large-chinese", model_kwargs={'device': "cpu"})
    result= embeddings.embed_query("你好")
    print(result)