mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-31 01:35:00 +00:00
37 lines
926 B
Python
37 lines
926 B
Python
from pathlib import Path
|
|
|
|
from .dump import dump # noqa: F401
|
|
|
|
|
|
def quoted_file(fname, display_fname, fence=("```", "```"), number=False):
|
|
prompt = "\n"
|
|
prompt += display_fname
|
|
prompt += f"\n{fence[0]}\n"
|
|
|
|
file_content = Path(fname).read_text()
|
|
lines = file_content.splitlines()
|
|
for i, line in enumerate(lines, start=1):
|
|
if number:
|
|
prompt += f"{i:4d} "
|
|
prompt += line + "\n"
|
|
|
|
prompt += f"{fence[1]}\n"
|
|
return prompt
|
|
|
|
|
|
def show_messages(messages, title=None, functions=None):
|
|
if title:
|
|
print(title.upper(), "*" * 50)
|
|
|
|
for msg in messages:
|
|
role = msg["role"].upper()
|
|
content = msg.get("content")
|
|
if content:
|
|
for line in content.splitlines():
|
|
print(role, line)
|
|
content = msg.get("function_call")
|
|
if content:
|
|
print(role, content)
|
|
|
|
if functions:
|
|
dump(functions)
|