Updated business interaction agent

business_interaction_agent/agent.py

@@ -8,6 +8,7 @@ from utils.models_loader import llm
 from langchain_core.messages import SystemMessage
 from .utils.prompts import business_retrieval_prompt, check_state_update_prompt
 from .utils.utils import manual_retrieval
+from context_analysis_agent.utils.utils import save_to_db
 
 business_state = State()
 
@@ -43,6 +44,7 @@ class BusinessInteractionChatbot:
         retrievals = manual_retrieval(str([msg['content'] for msg in self.messages if msg['role'] == 'user']),business_state.business_details)
         template = business_retrieval_prompt(str([msg['content'] for msg in self.messages if msg['role'] == 'user']),retrievals,str(business_state.business_details))
         messages = [SystemMessage(content=template)] + state["messages"]
+        print('The message is:',messages)
         backup_response = self.react_agent.invoke({'messages':messages})['messages'][-1]
         print('Backup response:',backup_response.content)
         return {"messages": [backup_response.content]}
@@ -59,13 +61,15 @@ class BusinessInteractionChatbot:
         return response.model_dump()
 
 
-
-
     def chat(self, user_input: str, business_details:dict):
         print('Entered into chat')
         business_state.business_details=business_details
         self.messages.append({"role": "user", "content": f'{user_input}'})
-        business_state.business_details=self.check_state_update()
+        checked_details = self.check_state_update()
+        if checked_details!= business_state.business_details:
+            save_to_db(checked_details)
+            print('Database Updated as the state changed....')
+            business_state.business_details = checked_details
 
         config = {"configurable": {"thread_id": "2"}}
         response = self.interact_agent.invoke({"messages":self.messages}, config)['messages'][-1].content