preparing search

App/Embedding/EmbeddingRoutes.py

@@ -1,7 +1,7 @@
 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"])
 
 
@@ -16,7 +16,7 @@ async def create_embeddings(task_id):
     return
 
 @embeddigs_router.get("/create_summary")
-async def create_embeddings(task_id):
+async def create_summary(task_id):
     item = await Transcriptions.objects.filter(task_id=task_id).first()
     temp = item.content
     chunks = generateChunks(temp, task_id)
@@ -28,5 +28,5 @@ async def create_embeddings(task_id):
 # search
 # update?
 @embeddigs_router.post("/search_embeddings")
-async def search_embeddings(query:str, task_id:str):
-    return search(query=query, task_id=task_id)
+async def search_embeddings(req:SearchRequest):
+    return search(query=req.query, task_id=req.taskId)

App/Embedding/Schemas.py

@@ -9,6 +9,14 @@ class BaseRequest(BaseModel):
     youtubeLink: Optional[str]
     telegramId: Optional[str]
 
+class SearchRequest(BaseModel):
+    # userId: int
+    taskId: str
+    query: str
+    # youtubeLink: Optional[str]
+    # telegramId: Optional[str]
+
+
 
 class GetTranscriptions(BaseModel):
     userId: int