rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-06-05 21:59:24 +02:00
Code Issues Projects Releases Wiki Activity

Add prompt "support"

Browse Source
This commit is contained in:
somebody
2022-12-27 18:52:22 -06:00
parent 25c089e171
commit 0a6dd8f7d8
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
koboldai_settings.py
View File

@@ -2111,9 +2111,15 @@ class KoboldStoryRegister(object):
""" """
Returns a list of chunks that comprise an action in dictionaries Returns a list of chunks that comprise an action in dictionaries
formatted as follows: formatted as follows:
type: string identifying chunk type ("ai", "user", or "edit") type: string identifying chunk type ("ai", "user", "edit", or "prompt")
content: the actual content of the chunk content: the actual content of the chunk
""" """
# Prompt doesn't need standard edit data
if action_id == -1:
if self.koboldai_vars.prompt:
return [{"type": "prompt", "content": self.koboldai_vars.prompt}]
return []
current_text = self.actions[action_id]["Selected Text"] current_text = self.actions[action_id]["Selected Text"]
action_original_type = "ai" action_original_type = "ai"