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

feat: Add files to the chat asynchronously.

Browse source
This commit is contained in:
Adam Albastov 2024-07-02 20:49:05 +02:00
parent 3953db340a
commit 912bfd1538
No known key found for this signature in database
GPG key ID: EDC12C536A39E0A1
1 changed files with 22 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

28
aider/gui.py
View file

@ -3,10 +3,12 @@
import os
import random
import sys
import asyncio
import streamlit as st
from aider import urls
from streamlit.runtime.scriptrunner import add_script_run_ctx
from aider.coders import Coder
from aider.dump import dump # noqa: F401
from aider.io import InputOutput
@ -181,10 +183,16 @@ class GUI:
self.do_add_web_page()
def do_add_files(self):
if 'files_being_added' not in st.session_state:
st.session_state.files_being_added = set()
all_files = self.coder.get_all_relative_files()
current_files = self.coder.get_inchat_relative_files()
fnames = st.multiselect(
"Add files to the chat",
self.coder.get_all_relative_files(),
default=self.state.initial_inchat_files,
all_files,
default=list(current_files),
placeholder="Files to edit",
disabled=self.prompt_pending(),
help=(
@ -194,15 +202,23 @@ class GUI:
)
for fname in fnames:
if fname not in self.coder.get_inchat_relative_files():
self.coder.add_rel_fname(fname)
self.info(f"Added {fname} to the chat")
if fname not in current_files and fname not in st.session_state.files_being_added:
st.session_state.files_being_added.add(fname)
asyncio.run(self.add_file_async(fname))
for fname in self.coder.get_inchat_relative_files():
for fname in current_files:
if fname not in fnames:
self.coder.drop_rel_fname(fname)
self.info(f"Removed {fname} from the chat")
async def add_file_async(self, fname):
try:
self.coder.add_rel_fname(fname)
self.info(f"Added {fname} to the chat")
finally:
st.session_state.files_being_added.remove(fname)
st.rerun()
def do_add_web_page(self):
with st.popover("Add a web page to the chat"):
self.do_web()