TODOs for making image code robust

This commit is contained in:
Joshua Vial 2023-11-30 11:09:56 +13:00
parent dc6aeb638d
commit 91bbb0a02c
3 changed files with 3 additions and 0 deletions

View file

@ -418,6 +418,7 @@ class Coder:
self.done_messages += self.cur_messages
self.summarize_start()
#TODO check for impact on image messages
if message:
self.done_messages += [
dict(role="user", content=message),
@ -464,6 +465,7 @@ class Coder:
dict(role="system", content=self.fmt_system_prompt(self.gpt_prompts.system_reminder)),
]
#TODO review impact of token count on image messages
messages_tokens = self.main_model.token_count(messages)
reminder_tokens = self.main_model.token_count(reminder_message)
cur_tokens = self.main_model.token_count(self.cur_messages)

View file

@ -327,6 +327,7 @@ class Commands:
if abs_file_path in self.coder.abs_fnames:
self.io.tool_error(f"{matched_file} is already in the chat")
else:
#TODO put in guard to stop images being added to non openai / gpt-4
content = self.io.read_text(abs_file_path)
if content is None:
self.io.tool_error(f"Unable to read {matched_file}")

View file