diff options
| author | Martin Bielik <martin.bielik@instea.sk> | 2024-12-20 17:34:20 +0100 |
|---|---|---|
| committer | Martin Bielik <martin.bielik@instea.sk> | 2024-12-21 00:45:31 +0100 |
| commit | 933a90d43ce9e360bb139dda2040b4360b9b12ce (patch) | |
| tree | 2313e9e412801fcc73371ba525842ced4d2daeb6 /py | |
| parent | 0f448c88e3eec091cedde2de535f69294cec5b63 (diff) | |
| download | vim-ai-933a90d43ce9e360bb139dda2040b4360b9b12ce.tar.gz | |
parse chat messages tests
Diffstat (limited to '')
| -rw-r--r-- | py/utils.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/py/utils.py b/py/utils.py index 931eeab..f5553ee 100644 --- a/py/utils.py +++ b/py/utils.py @@ -160,10 +160,12 @@ def parse_chat_messages(chat_content): try: with open(path, "r") as file: - message["content"] += f"\n\n==> {path} <==\n" + file.read() + file_content = file.read().strip() + message["content"] += f"\n\n==> {path} <==\n" + file_content except UnicodeDecodeError: message["content"] += "\n\n" + f"==> {path} <==" message["content"] += "\n" + "Binary file, cannot display" + message['content'] = message['content'].strip() return messages |