ExternalVendorCode/lollms
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms.git synced 2024-12-20 05:08:00 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update personality.py

Browse Source
This commit is contained in:
Saifeddine ALOUI 2023-11-27 10:24:31 +01:00 committed by GitHub
parent 6a4c48c724
commit 9963df3e2b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
lollms/personality.py
View File

@ -1631,7 +1631,10 @@ class APScript(StateMachine):
import autopep8 import autopep8
global_prompt = f"{prompt}\n!@>Code Builder:```python\n" global_prompt = f"{prompt}\n!@>Code Builder:```python\n"
code = self.fast_gen(global_prompt, max_title_length) code = self.fast_gen(global_prompt, max_title_length)
try:
back_quote_index = code.index("```") # Remove trailing backticks back_quote_index = code.index("```") # Remove trailing backticks
except:
pass
if back_quote_index>=0: if back_quote_index>=0:
# Removing any extra text # Removing any extra text
code = code[:back_quote_index] code = code[:back_quote_index]