Use black.Mode instead of .FileMode

Author: gvanrossum-ms

typeagent/aitools/utils.py

@@ -51,7 +51,7 @@ def format_code(text: str, line_width=None) -> str:
         # Use the terminal width, but cap it to 200 characters.
         line_width = min(200, shutil.get_terminal_size().columns)
     formatted_text = black.format_str(
-        text, mode=black.FileMode(line_length=line_width)
+        text, mode=black.Mode(line_length=line_width)
     ).rstrip()
     return reindent(formatted_text)
 

typeagent/knowpro/answers.py

@@ -128,7 +128,7 @@ def create_context_prompt(context: AnswerContext) -> str:
     prompt = [
         "[ANSWER CONTEXT]",
         "===",
-        black.format_str(str(dictify(context)), mode=black.FileMode(line_length=200)),
+        black.format_str(str(dictify(context)), mode=black.Mode(line_length=200)),
         "===",
     ]
     return "\n".join(prompt)