fix

controller.py

@@ -1,7 +1,7 @@
 import os
 from app_agent_config import AgentConfig
 from utils.logger import log_response
-from model.custom_agent import CustomHfAgent 
+from model.custom_agent import CustomHfAgent
 from model.conversation_chain_singleton import ConversationChainSingleton
 
 def cut_text_after_keyword(text, keyword):
@@ -17,19 +17,17 @@ class Controller:
         #self.app_config = AppConfig()
 
     image = []
-    def handle_submission(self, user_message ):
-        
+    def handle_submission(self, user_message ):        
         log_response("User input \n {}".format(user_message))
         log_response("selected_tools \n {}".format(self.agent_config.tool_checkboxes))
         log_response("url_endpoint \n {}".format(self.agent_config.url_endpoint))
         log_response("document \n {}".format(self.agent_config.document))
         log_response("image \n {}".format(self.agent_config.image))
-        log_response("context \n {}".format(self.agent_config.context))
-        
+        log_response("context \n {}".format(self.agent_config.context))   
         agent = CustomHfAgent(
             url_endpoint=self.agent_config.url_endpoint,
             token=os.environ['HF_token'],
-            additional_tools=self.agent_config.elected_tools,
+            additional_tools=self.agent_config.tool_checkboxes,
             input_params={"max_new_tokens": 192},
         )
 
@@ -41,15 +39,12 @@ class Controller:
 
     def handle_submission_chat(self, user_message, angent_respone):
         # os.environ['HUGGINGFACEHUB_API_TOKEN'] = os.environ['HF_token']
-        agent_chat_bot = ConversationChainSingleton().get_conversation_chain()    
-
+        agent_chat_bot = ConversationChainSingleton().get_conversation_chain()
         if angent_respone is not None:
             text = agent_chat_bot.predict(input=user_message + angent_respone)
         else:
             text = agent_chat_bot.predict(input=user_message)
 
         result = cut_text_after_keyword(text, "Human:")
-       
         print(result)
-
         return result

model/conversation_chain_singleton.py

@@ -1,6 +1,5 @@
-from langchain.vectorstores import FAISS
 from langchain.memory import ConversationBufferMemory
-from langchain.chains import ConversationalRetrievalChain, ConversationChain
+from langchain.chains import ConversationChain
 from langchain.llms import HuggingFaceHub
 
 class ConversationChainSingleton:

model/custom_agent_config.py

@@ -1,4 +1,4 @@
 
 class CustomHfAgentConfig():
-        def __init__():
- 
+        def __init__(self):
+                print("")