Use file type info and modification time when sending file context

In the 'prompts.py' file, the method of generating file context has been refactored to include more detailed information about each file, such as whether it's binary and when it was last modified. The 'arrow' library has been added to facilitate this change. This library has also been added to the 'requirements.in' and 'requirements.txt' files.

Author: TechNickAI

aicodebot/prompts.py

@@ -6,7 +6,7 @@
 from pathlib import Path
 from pydantic import BaseModel, Field
 from types import SimpleNamespace
-import functools, os
+import arrow, functools, os
 
 # ---------------------------------------------------------------------------- #
 #                              Personality helpers                             #
@@ -228,10 +228,15 @@ def generate_files_context(files):
 
     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()
-        files_context += f"--- START OF FILE: {file_name} ---\n"
-        files_context += contents
-        files_context += f"\n--- END OF FILE: {file_name} ---\n\n"
+        is_binary, file_info = Coder.get_file_info(file_name)
+        modification_ago = arrow.get(Path(file_name).stat().st_mtime).humanize()
+        if is_binary:
+            files_context += f"Binary file: {file_name}, modified {modification_ago}\n"
+        else:
+            files_context += f"--- START OF FILE: {file_name} {file_info} file, modified {modification_ago} ---\n"
+            contents = Path(file_name).read_text()
+            files_context += contents
+            files_context += f"\n--- END OF FILE: {file_name} ---\n\n"
 
     return files_context
 

requirements/requirements.in

@@ -5,9 +5,10 @@
 
 # Keep this list sorted alphabetically.
 
+arrow
 beautifulsoup4 # needed by langchain
-click # command line interface helpers
-faiss-cpu
+click
+faiss-cpu # for learn
 GitPython
 humanize
 langchain

requirements/requirements.txt

@@ -10,6 +10,8 @@ aiohttp==3.8.4
     #   openai
 aiosignal==1.3.1
     # via aiohttp
+arrow==1.2.3
+    # via -r requirements.in
 async-timeout==4.0.2
     # via aiohttp
 attrs==23.1.0
@@ -85,6 +87,8 @@ pydantic==1.10.9
     #   openapi-schema-pydantic
 pygments==2.15.1
     # via rich
+python-dateutil==2.8.2
+    # via arrow
 pyyaml==6.0.1
     # via
     #   -r requirements.in
@@ -99,6 +103,8 @@ requests==2.31.0
     #   tiktoken
 rich==13.4.2
     # via -r requirements.in
+six==1.16.0
+    # via python-dateutil
 smmap==5.0.0
     # via gitdb
 soupsieve==2.4.1