Enhanced file context generation in prompts

This commit ain't just about adding or removing a few lines of code. It's about giving a damn about the context. We've added a function to generate a directory structure, and we're using it to provide a more comprehensive context for the sidekick prompt. Now, not only do you get the relevant files, but you also get a nice little map of the directory structure. Ain't that something?

And let's not forget about the files. We've improved the way we handle them too. If there ain't no files, we don't just leave you hanging. We give you the directory structure and call it a day. But if there are files, oh boy, we give you the whole shebang. We read 'em, we count 'em, and we warn you if they're too damn big.

So, in summary, we've made the file context generation in prompts more informative and more useful. And if you don't appreciate that, well, I don't know what to tell you.

Author: TechNickAI

aicodebot/prompts.py

@@ -1,4 +1,4 @@
-from aicodebot.helpers import get_token_length, logger, read_config
+from aicodebot.helpers import generate_directory_structure, get_token_length, logger, read_config
 from langchain import PromptTemplate
 from pathlib import Path
 from types import SimpleNamespace
@@ -92,8 +92,18 @@ def get_personality_prompt():
 
 
 def generate_files_context(files):
-    """Generate the files context for the sidekick prompt."""
-    files_context = "Here are the relevant files we are working with in this session:\n"
+    """Generate the files context for the sidekick prompt.
+
+    This includes a directory structure and the contents of $files
+
+    """
+    files_context = "\nHere is the directory structure we are working with in this session:\n"
+    files_context += generate_directory_structure(".", ignore_patterns=[".git"], use_gitignore=True)
+
+    if not files:
+        return files_context
+
+    files_context += "Here are the relevant files we are working with in this session:\n"
     for file_name in files:
         contents = Path(file_name).read_text()
         token_length = get_token_length(contents)