Raise ValueError if gpt makes an unknown function_call

This commit is contained in:
Paul Gauthier 2023-06-24 08:32:12 -07:00
parent 4eee71b956
commit a1909611d9
2 changed files with 10 additions and 2 deletions

View file

@ -105,8 +105,11 @@ class WholeFileFunctionCoder(Coder):
return "\n".join(show_diff)
def update_files(self):
args = self.parse_partial_args()
name = self.partial_response_function_call.get("name")
if name and name != "write_file":
raise ValueError(f'Unknown function_call name="{name}", use name="write_file"')
args = self.parse_partial_args()
if not args:
return