validated initial interaction

my_agent/utils/initial_interaction.py

@@ -7,15 +7,8 @@ from pydantic import BaseModel, ConfigDict, Field
 from typing import Optional, List
 from .models_loader import llm
 from .prompts import introduction_prompt , details_extract_prompt
+from .validators import DetailsFormatter
 
-# Pydantic model for extracted business info
-class DetailsFormatter(BaseModel):
-    business_type: str = Field(description="The type of the business")
-    platform: str = Field(description="The platform used for the business")
-    target_audience: str = Field(description="The target audience of the business")
-    business_goals: str = Field(description="The business goals of the business")
-    offerings: str = Field(description="The offerings of the business")
-    Challenges_faced: str = Field(description="The challenges faced by the business")
 
 # State model
 class State(BaseModel):
@@ -62,11 +55,7 @@ class IntroductionChatbot:
         template = details_extract_prompt(business_state.interactions)
 
         messages = [SystemMessage(content=template)]
-        response = self.llm.bind_tools([DetailsFormatter]).invoke(messages)
+        response = self.llm.with_structured_output(DetailsFormatter).invoke(messages)
+        print('Extracetd details:',response)
+        return response
 
-        if hasattr(response, 'tool_calls') and response.tool_calls:
-            return response.tool_calls[0]['args']
-        elif hasattr(response, 'content'):
-            return response.content
-        else:
-            return "No response"