style: Fix formatting and style issues in base_coder.py

This commit is contained in:
Paul Gauthier (aider) 2024-08-06 09:46:42 -03:00
parent 7d56889880
commit 541ab83f15

View file

@ -2,6 +2,7 @@
import hashlib import hashlib
import json import json
import locale
import math import math
import mimetypes import mimetypes
import os import os
@ -11,7 +12,6 @@ import sys
import threading import threading
import time import time
import traceback import traceback
import locale
from collections import defaultdict from collections import defaultdict
from datetime import datetime from datetime import datetime
from json.decoder import JSONDecodeError from json.decoder import JSONDecodeError
@ -744,15 +744,15 @@ class Coder:
try: try:
lang = locale.getlocale()[0] lang = locale.getlocale()[0]
if lang: if lang:
return lang.split('_')[0] # Extract just the language code return lang.split("_")[0] # Extract just the language code
except: except:
pass pass
for env_var in ['LANG', 'LANGUAGE', 'LC_ALL', 'LC_MESSAGES']: for env_var in ["LANG", "LANGUAGE", "LC_ALL", "LC_MESSAGES"]:
lang = os.environ.get(env_var) lang = os.environ.get(env_var)
if lang: if lang:
return lang.split('_')[0] # Extract just the language code return lang.split("_")[0] # Extract just the language code
return None return None
def fmt_system_prompt(self, prompt): def fmt_system_prompt(self, prompt):
@ -766,11 +766,14 @@ class Coder:
val = os.getenv(var) val = os.getenv(var)
platform_text += f"- The user's shell: {var}={val}\n" platform_text += f"- The user's shell: {var}={val}\n"
user_lang = self.get_user_language() user_lang = self.get_user_language()
if user_lang: if user_lang:
platform_text += f"- The user's language: {user_lang} (detected from locale or environment variables)\n" platform_text += (
f"- The user's language: {user_lang} (detected from locale or environment"
" variables)\n"
)
dt = datetime.now().isoformat() dt = datetime.now().isoformat()
platform_text += f"- The current date/time: {dt}" platform_text += f"- The current date/time: {dt}"