Question playground (#78)

* Initial playground

* sub ref

* Add save question via most_similar_by_embedding

* Clean up most_similar_by_embedding

* Remove unused fields

* sub ref fix

* Add question check

* sub ref

Author: anmarhindi

app.py

@@ -3,7 +3,10 @@
 from pydantic import BaseModel
 from typing import List, Optional, Dict, Any, Union
 from neural_search import util
-from submodules.model.business_objects import general
+from submodules.model.business_objects import (
+    general,
+    playground_question as playground_question_db_bo,
+)
 from submodules.model import session
 
 app = FastAPI()
@@ -62,11 +65,13 @@ class MostSimilarByEmbeddingRequest(BaseModel):
     limit: int = 5
     att_filter: Optional[List[Dict[str, Any]]] = None
     threshold: Optional[Union[float, int]] = None
+    question: Optional[str] = None
 
 
 @app.post("/most_similar_by_embedding")
 def most_similar_by_embedding(
-    request: MostSimilarByEmbeddingRequest, include_scores: bool = False
+    request: MostSimilarByEmbeddingRequest,
+    include_scores: bool = False,
 ) -> responses.JSONResponse:
     """Find the n most similar records with respect to the specified embedding.
         Args:
@@ -95,6 +100,14 @@ def most_similar_by_embedding(
         request.threshold,
         include_scores,
     )
+
+    if request.question:
+        playground_question_db_bo.create(
+            request.project_id,
+            request.question,
+            with_commit=True,
+        )
+
     return responses.JSONResponse(
         status_code=status.HTTP_200_OK,
         content=similar_records,