Updated state and tools

brainstroming_agent/utils/state.py

@@ -23,3 +23,8 @@ class BrainstromTopicFormatter(BaseModel):
     topic2:str=Field(description="Second brainstorming topic of the story")
     topic3:str=Field(description="Third brainstorming topic of the story")
     topic4:str=Field(description="Fourth brainstorming topic of the story")
+
+class QueryFormatter(BaseModel):
+    messages: str = Field(description="The message that user passes to the agent")
+    business_details: dict = Field (description="The business details that user passes to the agent")
+

brainstroming_agent/utils/tools.py

@@ -14,7 +14,7 @@ import re
 import faiss
 import ast
 import pandas as pd
-from .validators import QueryFormatter
+from .state import QueryFormatter
 
 os.environ['GROQ_API_KEY']=os.getenv('GROQ_API_KEY')
 @tool("influencer's data-retrieval-tool", args_schema=QueryFormatter, return_direct=False,description="Retrieve influencer-related data for a given query.")