made changes in function_agent.py file (#124)

Co-authored-by: THEAVINASHREDDY <[email protected]> Co-authored-by: Dominic <[email protected]>
shroominic · Oct 10, 2023 · 147f09a · 147f09a
1 parent c7e242f
commit 147f09a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/codeinterpreterapi/agents/functions_agent.py b/codeinterpreterapi/agents/functions_agent.py
@@ -128,13 +128,13 @@ def get_allowed_tools(self) -> List[str]:
         """Get allowed tools."""
         return list([t.name for t in self.tools])
 
-    @root_validator
+    @root_validator(allow_reuse=True)
     def validate_llm(cls, values: dict) -> dict:
         if not isinstance(values["llm"], ChatOpenAI):
             raise ValueError("Only supported with ChatOpenAI models.")
         return values
 
-    @root_validator
+    @root_validator(allow_reuse=True)
     def validate_prompt(cls, values: dict) -> dict:
         prompt: BasePromptTemplate = values["prompt"]
         if "agent_scratchpad" not in prompt.input_variables: