feature(#74): implement api to delete data of autotask from firebase database

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,
@@ -35,6 +36,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
 
@@ -444,4 +446,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