dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-30 01:04:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

wired up fnames

Browse source
This commit is contained in:
Paul Gauthier 2024-04-27 06:37:57 -07:00
parent 89a721b386
commit 4c8d32601c
1 changed files with 14 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

23
aider/gui.py
View file

@ -133,15 +133,21 @@ class GUI:
def do_add_to_chat(self): def do_add_to_chat(self):
with st.expander("Add to the chat", expanded=True): with st.expander("Add to the chat", expanded=True):
st.multiselect( fnames = st.multiselect(
"Files for the LLM to edit", "Files for the LLM to edit",
self.coder.get_all_relative_files(), sorted(self.coder.get_all_relative_files()),
default=self.coder.get_inchat_relative_files(), default=sorted(self.coder.get_inchat_relative_files()),
placeholder="Files to edit",
help=( help=(
"Only add the files that need to be *edited* for the task you are working" "Only add the files that need to be *edited* for the task you are working"
" on. Aider will pull in other code to provide relevant context to the LLM." " on. Aider will pull in other code to provide relevant context to the LLM."
), ),
) )
for fname in fnames:
if fname not in self.coder.get_inchat_relative_files():
self.coder.add_rel_fname(fname)
with st.popover("Add web page"): with st.popover("Add web page"):
st.markdown("www") st.markdown("www")
st.text_input("URL?") st.text_input("URL?")
@ -219,8 +225,7 @@ class GUI:
for msg in st.session_state.messages: for msg in st.session_state.messages:
with self.messages.chat_message(msg["role"]): with self.messages.chat_message(msg["role"]):
st.write(msg["content"]) st.write(msg["content"])
cost = random.random() * 0.003 + 0.001 # self.cost()
st.caption(f"${cost:0.4f}")
self.chat_controls = st.empty() self.chat_controls = st.empty()
@ -274,14 +279,14 @@ class GUI:
while prompt: while prompt:
with self.messages.chat_message("assistant"): with self.messages.chat_message("assistant"):
res = st.write(self.coder.run_stream(prompt)) res = st.write_stream(self.coder.run_stream(prompt))
self.cost() st.session_state.messages.append({"role": "assistant", "content": res})
# self.cost()
if self.coder.reflected_message: if self.coder.reflected_message:
self.messages.info(self.coder.reflected_message) self.messages.info(self.coder.reflected_message)
prompt = self.coder.reflected_message prompt = self.coder.reflected_message
st.session_state.messages.append({"role": "assistant", "content": res}) dump(st.session_state.messages)
with self.messages: with self.messages:
self.mock_tool_output() self.mock_tool_output()