File size: 909 Bytes
32a36b2 d7d0d8e 86258ef f5b43ca 32a36b2 d7d0d8e 32a36b2 d7d0d8e f5b43ca 55174eb 86258ef 55174eb d7d0d8e 32a36b2 55174eb f5b43ca 86258ef |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 |
from fastapi import APIRouter
from App.Transcription.Model import Transcriptions
from .utils.Initialize import generateChunks, encode, search
from .Schemas import SearchRequest
embeddigs_router = APIRouter(tags=["embeddings"])
# create
@embeddigs_router.get("/create_embeddings")
async def create_embeddings(task_id):
item = await Transcriptions.objects.filter(task_id=task_id).first()
temp = item.content
chunks = generateChunks(temp, task_id)
encode(chunks)
return
@embeddigs_router.get("/create_summary")
async def create_summary(task_id):
item = await Transcriptions.objects.filter(task_id=task_id).first()
temp = item.content
chunks = generateChunks(temp, task_id)
encode(chunks)
return
# search
# update?
@embeddigs_router.post("/search_embeddings")
async def search_embeddings(req: SearchRequest):
return search(query=req.query, task_id=req.taskId)
|