Updated the prompts

my_agent/agent.py

@@ -37,7 +37,7 @@ def build_graph_old():
 
 def build_graph():
     builder = StateGraph(State)
-    # builder.add_node(caption_image)
+    builder.add_node(caption_image)
     builder.add_node(retrieve)
     builder.add_node(generate_story)
     builder.add_node(generate_brainstroming)
@@ -46,12 +46,12 @@ def build_graph():
 
 
     # Normal edges
-    # builder.add_edge(START, "caption_image")
+    builder.add_edge(START, "caption_image")
+    builder.add_edge("caption_image", "retrieve")
 
-    builder.add_edge(START, "retrieve")
+    # builder.add_edge(START, "retrieve")
     builder.add_edge("retrieve", "generate_story")
 
-    # builder.add_edge("caption_image", "retrieve")
     # builder.add_edge("retrieve", "generate_story")
     builder.add_edge("generate_story", "generate_brainstroming")
 

my_agent/utils/initial_interaction.py

@@ -6,6 +6,7 @@ from langchain_core.messages import SystemMessage
 from pydantic import BaseModel, ConfigDict, Field
 from typing import Optional, List
 from .models_loader import llm
+from .prompts import business_interaction_prompt
 
 # Pydantic model for extracted business info
 class DetailsFormatter(BaseModel):
@@ -41,7 +42,7 @@ class BusinessInteractionChatbot:
         return workflow
 
     def _call_model(self, state):
-        template = self._get_prompt_template()
+        template = business_interaction_prompt
         messages = [SystemMessage(content=template)] + state["messages"]
         response = self.llm.invoke(messages)
         return {"messages": [response]}

my_agent/utils/prompts.py

@@ -59,3 +59,21 @@ I want to create a detailed storyline for a video in the given topic. You have t
 
       **Final Reminder** You have to strongly focus on these topics while creating the storyline: {[item for sublist in final_state['preferred_topics'] for item in sublist]}'''
     )
+
+
+business_interaction_prompt = '''
+You are a business assistant who collects only valid and relevant data.
+Your job is to gather details from business owners in a friendly and conversational manner to understand their business better. Ask in very easy and short way.
+
+We need these details:
+1. Business Type (e.g., e-commerce, SaaS, consulting),
+2. Platform(s) used (e.g., website, app, Instagram),
+3. Target Audience (who are their customers or clients),
+4. Business Goals (short-term or long-term objectives),
+5. Offerings (products or services they provide),
+6. Challenges faced (any current business problems or limitations).
+
+Keep interacting until all valid details are collected.
+
+VERY IMPORTANT: Once all valid details are received, say: '**Thanks for providing all your required business details.**'
+'''

my_agent/utils/tools.py

@@ -1,4 +1,3 @@
-from langchain_core.messages import SystemMessage
 from langchain_groq import ChatGroq
 from pydantic import BaseModel, Field
 from dotenv import load_dotenv