bug

app.py

@@ -77,7 +77,8 @@ class SearchDocumentsRequest(BaseModel):
     limit: int = 3
 
 class GenerateRAGRequest(BaseModel):
-    query: str
+    search_query: str
+    
 
 
 @app.get("/")
@@ -135,8 +136,8 @@ async def generate_rag_response_api(
         logging.info("Starting document search")
 
         # Encode the query using the custom embedding function
-        query_embedding = embed_text(body.query)
-        print(body.query)
+        query_embedding = embed_text(body.search_query)
+        print(body.search_query)
         collection_name = "embed"  # Use the collection name where the embeddings are stored   
         # Perform search using the precomputed embeddings
         hits, error = searcher.search_documents(collection_name, query_embedding, user_id)
@@ -148,7 +149,7 @@ async def generate_rag_response_api(
         logging.info("Generating RAG response")
 
         # Generate the RAG response using the retrieved documents
-        response, error = generate_rag_response(hits, body.query)
+        response, error = generate_rag_response(hits, body.search_query)
         
         if error:
             logging.error(f"Generate RAG response error: {error}")