Added pydantic validator to update the state

business_interaction_agent/utils/state.py

@@ -1,6 +1,18 @@
-from pydantic import BaseModel, ConfigDict
+from pydantic import BaseModel, ConfigDict , Field
 from typing import Optional
 
 class State(BaseModel):
     interactions: Optional[list] = []
-    model_config = ConfigDict(arbitrary_types_allowed=True)
+    business_details : Optional[dict] = {}
+    model_config = ConfigDict(arbitrary_types_allowed=True)
+
+class StateUpdateFormatter(BaseModel):
+    business_type: dict = Field(description="The type of the business. If updated, the new one.")
+    platform: str = Field(description="The platform used for the business. If updated, the new one.")
+    target_audience: str = Field(description="The target audience of the business. If updated, the new one.")
+    business_goals: str = Field(description="The business goals of the business. If updated, the new one.")
+    offerings: str = Field(description="The offerings of the business. If updated, the new one.")
+    Challenges_faced: str = Field(description="The challenges faced by the business. If updated, the new one.")
+
+
+