Update code.py

agential/cog/strategies/critic/code.py

@@ -12,41 +12,17 @@
 class CriticPydanticOutput(BaseModel):
     """Critic output
     Attributes:
-        For "qa" mode:
-        answer (str): The answer generated by the agent. 
-        critique (str): The critique of the answer generated by the agent. 
-        query (str): The query requested by the agent. 
-        search_result (str): The search result requested by the agent. 
-        revised_answer (str): The revised answer generated by the agent. 
-        For "math" mode: 
-        
-        code (int/str?): The code generated by the agent. 
+                
+        code (str): The code generated by the agent.
         critique (str): The critique of the answer generated by the agent.
-        If use_interpreter_tool == True:
         execution_status (str): The execution status of the agent.
-        code_answer (int/str?): The code answer generated by the agent.
-        improved_code (int/str?): The improved code generated by the agent.
-        For "code" mode: 
-        
-        code (int/str?): The code generated by the agent.
-        critique (str): The critique of the answer generated by the agent.
-        if use_interpreter_tool == True:
-        execution_status (str): The execution status of the agent.
-        improved_code (int/str?): The improved code generated by the agent.
-        """
-
-        answer: str = Field(..., description = "The answer generated by the agent.")
-        critique : str = Field(..., description = "The critique of the answer generated by the agent.")
-        query: str = Field(..., description = "The query requested by the agent. ")
-        search_result: str = Field(..., description = "The search result requested by the agent.")
-        revised_answer: str = Field(..., description = "The revised answer generated by the agent. ")
+        improved_code (str): The improved code generated by the agent.
+    """
 
-        code: int/str? = Field(..., description = "The code generated by the agent.")
+        code: str = Field(..., description = "The code generated by the agent.")
         critique: str = Field(..., description = " The critique of the answer generated by the agent.")
-
         execution_status: str = Field(..., description = "The execution status of the agent.")
-        code_answer: int/str? = Field(..., description = "The code answer generated by the agent.")
-        improved_code: int/str? = Field(..., description = "The improved code generated by the agent.")
+        improved_code: str = Field(..., description = "The improved code generated by the agent.")
 
 
 
@@ -94,19 +70,17 @@ def generate_pydantic_output(
     ) -> CriticPydanticOutput:
         """Creates a pydantic model of the output.
         Args:
-            answer (str): The answer generated by the agent.
+            code (str): The code generated by the agent.
             critique (str): The critique of the answer generated by the agent.
-            query (str): The query requested by the agent.
-            search_result (str): The search result requested by the agent.
-            revised_answer (str): The revised answer generated by the agent.
+            execution_status (str): The execution status of the agent.
+            improved_code (str): The improved code generated by the agent.
         """
 
         return CriticPydanticOutput(
-            answer=answer,
+            code=code,
             critique=critique,
-            query=query,
-            search_result=search_result,
-            revised_answer=revised_answer,
+            execution_status=execution_status,
+            improved_code=improved_code,
         )
 
     def generate_critique(