Merge pull request #123 from ttt246/feature/instant_answer_74

Brain/src/model/requests/request_model.py

@@ -205,3 +205,13 @@ class Train(BasicReq):
     id: str
     data: TrainData
     status: str
+
+
+"""endpoint /auto_task/delete"""
+
+
+class AutoTaskDelete(BasicReq):
+    class Body(BaseModel):
+        reference_link: str
+
+    data: Body

Brain/src/router/api.py

@@ -18,6 +18,7 @@ from Brain.src.model.requests.request_model import (
     BasicReq,
     ClientInfo,
     get_client_info,
+    AutoTaskDelete,
 )
 from Brain.src.rising_plugin.risingplugin import (
     getCompletion,
@@ -36,6 +37,7 @@ from Brain.src.service.contact_service import ContactsService
 from Brain.src.service.feedback_service import FeedbackService
 from Brain.src.service.llm.chat_service import ChatService
 from Brain.src.service.twilio_service import TwilioService
+from Brain.src.service.auto_task_service import delete_db_data
 
 from fastapi import APIRouter, Depends
 
@@ -452,4 +454,42 @@ def construct_blueprint_api() -> APIRouter:
             200, "Deleted all contacts from pinecone successfully", ""
         )
 
+    """@generator.request_body(
+        {
+            "token": "String",
+            "uuid": "String",
+            "data": {
+                "reference_link": "test link",
+            },
+        }
+    )
+
+    @generator.response(
+        status_code=200, schema={"message": "message", "result": "test_result"}
+    )
+
+    """
+
+    @router.post("/auto_task/delete")
+    def delete_data(data: AutoTaskDelete):
+        # firebase admin init
+        try:
+            setting, firebase_app = firebase_admin_with_setting(data)
+        except BrainException as ex:
+            return assembler.to_response(ex.code, ex.message, "")
+        try:
+            token = setting.token
+            uuid = setting.uuid
+
+            # parsing contacts
+            # train contact
+            delete_db_data(data.data.reference_link, firebase_app)
+        except Exception as e:
+            if isinstance(e, BrainException):
+                return e.get_response_exp()
+            return assembler.to_response(400, "Failed to delete data", "")
+        return assembler.to_response(
+            200, "Deleted data from real-time database of firebase", ""
+        )
+
     return router

Brain/src/service/auto_task_service.py

@@ -1,5 +1,6 @@
 """auto task management to get the expected output"""
 import firebase_admin
+from firebase_admin import db
 
 from Brain.src.model.req_model import ReqModel
 from Brain.src.rising_plugin.llm.autogpt_llm import AutoGPTLLM
@@ -7,6 +8,14 @@ import time
 import asyncio
 import threading
 
+"""delete data from real time database of firebase using reference link    
+"""
+
+
+def delete_db_data(reference_link: str, firebase_app: firebase_admin.App):
+    ref = db.reference(reference_link, app=firebase_app)
+    ref.delete()
+
 
 class AutoTaskService:
     """self task archivement with autogpt based on langchain

Brain/tests/functional/test_api.py

@@ -209,3 +209,29 @@ def test_chat_rising(body):
 def test_train_contacts(body):
     response = client.post("/train/contacts", json=body)
     assert response.status_code == 200
+
+
+@pytest.mark.parametrize(
+    "body",
+    [
+        (
+            {
+                "data": {
+                    "reference_link": "test link",
+                },
+                "confs": {
+                    "token": "test_token",
+                    "uuid": "test-uuid",
+                    "openai_key": "",
+                    "pinecone_key": "",
+                    "pinecone_env": "",
+                    "firebase_key": "",
+                    "settings": {"temperature": 0.6},
+                },
+            }
+        )
+    ],
+)
+def test_delete_data(body):
+    response = client.post("/auto_task/delete", json=body)
+    assert response.status_code == 200