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)