Spaces:
Running
Running
File size: 450 Bytes
51a7f02 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
from fastapi import APIRouter, UploadFile, File
import openai
import io
import os
from pypdf import PdfReader
router = APIRouter()
openai.api_key = os.getenv("OPENAI_API_KEY")
@router.post("/v1/embeddings")
async def embed_doc(file: UploadFile = File(...)):
#for now just truncate based on length of words
content = await file.read()
return openai.Embedding.create(input = content.decode("utf-8"), model = "text-embedding-ada-002") |