kaizenkodo/ChatMastermind
2
0
Fork 0
Code Issues 2 Pull Requests Packages Projects Releases Wiki Activity

Fix extracting source file with type specification.

Browse Source
This commit is contained in:
Oleksandr Kozachuk 2023-09-09 11:53:32 +02:00
parent ed567afbea
commit 3ef1339cc0
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
chatmastermind/utils.py
View File

@ -55,9 +55,10 @@ def message_to_chat(message: dict[str, str],
def display_source_code(content: str) -> None:
try:
content_start = content.index('```')
content_start = content.index('\n', content_start) + 1
content_end = content.rindex('```')
if content_start + 3 < content_end:
print(content[content_start + 3:content_end].strip())
if content_start < content_end:
print(content[content_start:content_end].strip())
except ValueError:
pass