File size: 614 Bytes
32a36b2 d7d0d8e 32a36b2 d7d0d8e 32a36b2 d7d0d8e 32a36b2 d7d0d8e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
from fastapi import APIRouter
from App.Transcription.Model import Transcriptions
from .utils.Initialize import generateChunks, encode, search
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
# search
# update?
@embeddigs_router.get("/search_embeddings")
async def search_embeddings(query, task_id):
return search(query=query, task_id=task_id)
|