bugfix(#136): autogpt interface issue fixed for infinit loop and applied with finish command as well.

Brain/src/firebase/firebase.py

@@ -3,6 +3,7 @@ import os
 from typing import Any
 
 import firebase_admin
+from firebase_admin import db
 from firebase_admin import credentials
 
 from Brain.src.common.assembler import Assembler
@@ -62,3 +63,15 @@ def get_firebase_cred(setting: ReqModel):
     else:
         cred = json.loads(setting.firebase_key)
         return credentials.Certificate(cred)
+
+
+"""
+delete data from real time database of firebase using reference link    
+"""
+
+
+def delete_data_from_realtime(
+    reference_link: str, firebase_app: firebase_admin.App
+) -> None:
+    ref = db.reference(reference_link, app=firebase_app)
+    ref.delete()

Brain/src/rising_plugin/llm/autogpt_llm.py

@@ -27,6 +27,11 @@ from marshmallow import ValidationError
 
 import faiss
 
+from Brain.src.rising_plugin.llm.llms import (
+    MAX_AUTO_THINKING,
+    get_finish_command_for_auto_task,
+)
+
 
 class AutoGPTLLM:
     """autogpt run method to get the expected result"""
@@ -52,6 +57,12 @@ class AutoGPTLLM:
             # Discontinue if continuous limit is reached
             loop_count += 1
 
+            # validation thinking counter
+            if loop_count == MAX_AUTO_THINKING:
+                # add finish command of the command
+                ref.push().set(get_finish_command_for_auto_task())
+                break
+
             # Send message to AI, get response
             assistant_reply = agent.chain.run(
                 goals=goals,

Brain/src/rising_plugin/llm/llms.py

@@ -20,6 +20,8 @@ GPT_LLM_MODELS = [GPT_3_5_TURBO, GPT_4, GPT_4_32K]
 """exception message"""
 EXCEPTION_MSG = f"The model is not correct. It should be in {LLM_MODELS}"
 
+"""maximum auto achievement counter"""
+MAX_AUTO_THINKING = 10
 
 """prompt"""
 CATEGORY_PROMPT = """
@@ -99,3 +101,17 @@ def get_llm(
     elif model == FALCON_7B:
         llm = FalconLLM(temperature=temperature, max_new_tokens=max_new_tokens)
     return llm
+
+
+"""
+generate finish command and response for auto achievement
+"""
+
+
+def get_finish_command_for_auto_task() -> Any:
+    return {
+        "command": {
+            "args": {"response": "I have finished all my objectives."},
+            "name": "finish",
+        }
+    }

Brain/src/service/auto_task_service.py

@@ -1,27 +1,17 @@
 """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
 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
+    """self task achievement with autogpt based on langchain
     response -> reference_link :str"""
 
-    def ask_task_with_autogpt(
+    def ask_task_with_llm(
         self, query: str, firebase_app: firebase_admin.App, setting: ReqModel
     ) -> str:
         # init autogpt llm
@@ -39,9 +29,11 @@ class AutoTaskService:
 
         return reference_link
 
-    """generate reference link for autoTask
+    """
+    generate reference link for autoTask
     response type:
-    /auto/{llm_name}_{uuid}_{timestamp}"""
+    /auto/{llm_name}_{uuid}_{timestamp}
+    """
 
     def generate_reference_link(self, llm_name: str, uuid: str) -> str:
         milliseconds = int(time.time() * 1000)