edited bigmessage

core/types.py

@@ -8,10 +8,10 @@ from textwrap import dedent
 class ThoughtSteps(BaseModel):
     step_title: str = Field(..., description="steps to use for the problem/question")
     thought: str = Field(..., description="internal monologue, this contails your questions and its answers")
-    next_step: bool =  Field(..., description="Does the problem require more thinking? if yes then set to true, else set to false,")
+    next_step: bool =  Field(default=True, description="Does the problem require more thinking? if yes then set to true, else set to false,")
     answer: str | None = Field(..., description="generate a answer based on inner thoughts")
     critic: str | None = Field(..., description="criticize the answer, try to prove it wrong , have a different perspective, fight it")
-    is_final_answer: bool = Field(..., description="this is final answer no next step required,")
+    is_final_answer: bool = Field(default=False, description="this is final answer no next step required,")
     
     def to_thought_steps_display(self):
         return ThoughtStepsDisplay(
@@ -44,13 +44,20 @@ class ThoughtStepsDisplay(BaseModel):
 
 class BigMessage(BaseModel):
     role:str
-    content:ThoughtSteps | str
-    thoughts:list[ThoughtSteps|None]|None = Field(default_factory=list)
+    content:ThoughtStepsDisplay | str
+    thoughts:list[ThoughtStepsDisplay|None]|None = Field(default_factory=list)
     
     def to_message(self):
+        if isinstance(self.content, ThoughtStepsDisplay):
+            content = self.content.model_dump()
+            content = ThoughtSteps(**content).model_dump_json()
+        
+        else:
+            content = self.content
+        
         return {
             "role": self.role,
-            "content": self.content.model_dump_json() if isinstance(self.content, ThoughtSteps) else self.content,
+            "content": content,
         }
     
 class Message(BaseModel):