diff --git a/src/typeagent/aitools/utils.py b/src/typeagent/aitools/utils.py index 46e81fc1..eb21f9ec 100644 --- a/src/typeagent/aitools/utils.py +++ b/src/typeagent/aitools/utils.py @@ -11,7 +11,6 @@ import sys import time -import black import colorama import typechat @@ -57,6 +56,8 @@ def format_code(text: str, line_width=None) -> str: NOTE: The text must be a valid Python expression or code block. """ + import black + if line_width is None: # Use the terminal width, but cap it to 200 characters. line_width = min(200, shutil.get_terminal_size().columns) diff --git a/src/typeagent/knowpro/answers.py b/src/typeagent/knowpro/answers.py index 2c300506..58a536ed 100644 --- a/src/typeagent/knowpro/answers.py +++ b/src/typeagent/knowpro/answers.py @@ -5,8 +5,6 @@ from dataclasses import dataclass from typing import Any -import black - import typechat from .answer_context_schema import AnswerContext, RelevantKnowledge, RelevantMessage @@ -127,6 +125,8 @@ def create_question_prompt(question: str) -> str: def create_context_prompt(context: AnswerContext) -> str: # TODO: Use a more compact representation of the context than JSON. + import black + prompt = [ "[ANSWER CONTEXT]", "===",