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

AI Context Fixes

Browse Source
This commit is contained in:
ebolam
2022-10-07 08:33:53 -04:00
parent b3e0147b5d
commit a0c347a4b4
2 changed files with 49 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
koboldai_settings.py
View File

@@ -126,10 +126,14 @@ class koboldai_vars(object):
if story_name in self._story_settings:
self._story_settings[story_name].no_save = True
self._story_settings[story_name].from_json(json_data)
logger.debug("Calcing AI text after load story")
ignore = self.calc_ai_text()
self._story_settings[story_name].no_save = False
else:
#self._story_settings[story_name].no_save = True
self.create_story(story_name, json_data=json_data)
logger.debug("Calcing AI text after create story")
ignore = self.calc_ai_text()
#self._story_settings[story_name].no_save = False
self._system_settings.story_loads[original_story_name] = datetime.datetime.now().strftime("%m/%d/%Y, %H:%M:%S")
with open("settings/system_settings.v2_settings", "w") as settings_file:
@@ -193,7 +197,7 @@ class koboldai_vars(object):
# TODO: We may want to replace the "text" variable with a list-type
# class of context blocks, the class having a __str__ function.
if self.sp_length > 0:
context.append({"type": "soft_prompt", "text": f"<{self.sp_length} tokens of Soft Prompt.>"})
context.append({"type": "soft_prompt", "text": f"<{self.sp_length} tokens of Soft Prompt.>", "tokens": self.sp_length})
# Header is never used?
# if koboldai_vars.model not in ("Colab", "API", "OAI") and self.tokenizer._koboldai_header:
# context.append({"type": "header", "text": f"{len(self.tokenizer._koboldai_header})
@@ -207,7 +211,7 @@ class koboldai_vars(object):
if self.tokenizer is not None and self.memory_length > self.max_memory_length:
memory_text = self.tokenizer.decode(self.tokenizer.encode(self.memory)[-self.max_memory_length-1:])
context.append({"type": "memory", "text": memory_text})
context.append({"type": "memory", "text": memory_text, "tokens": memory_length})
text += memory_text
#Add constant world info entries to memory
@@ -218,7 +222,7 @@ class koboldai_vars(object):
used_world_info.append(wi['uid'])
self.worldinfo_v2.set_world_info_used(wi['uid'])
wi_text = wi['content']
context.append({"type": "world_info", "text": wi_text})
context.append({"type": "world_info", "text": wi_text, "tokens": wi['token_length']})
text += wi_text
@@ -263,7 +267,7 @@ class koboldai_vars(object):
used_tokens+=0 if wi['token_length'] is None else wi['token_length']
used_world_info.append(wi['uid'])
wi_text = wi['content']
context.append({"type": "world_info", "text": wi_text})
context.append({"type": "world_info", "text": wi_text, "tokens": wi['token_length']})
text += wi_text
self.worldinfo_v2.set_world_info_used(wi['uid'])
@@ -288,17 +292,20 @@ class koboldai_vars(object):
for i in range(len(action_text_split)-1, -1, -1):
if action_text_split[i][3] or action_text_split[i][1] == [-1]:
#We've hit an item we've already included or items that are only prompt. Stop
for action in action_text_split[i][1]:
if action >= 0:
self.actions.set_action_in_ai(action)
break;
if len(action_text_split) - i - 1 == self.andepth and self.authornote != "":
game_text = "{}{}".format(authors_note_final, game_text)
game_context.insert(0, {"type": "authors_note", "text": authors_note_final})
game_context.insert(0, {"type": "authors_note", "text": authors_note_final, "tokens": authornote_length})
length = 0 if self.tokenizer is None else len(self.tokenizer.encode(action_text_split[i][0]))
if length+used_tokens <= token_budget and not used_all_tokens:
used_tokens += length
selected_text = action_text_split[i][0]
action_text_split[i][3] = True
game_text = "{}{}".format(selected_text, game_text)
game_context.insert(0, {"type": "action", "text": selected_text})
game_context.insert(0, {"type": "action", "text": selected_text, "tokens": length})
for action in action_text_split[i][1]:
if action >= 0:
self.actions.set_action_in_ai(action)
@@ -327,10 +334,10 @@ class koboldai_vars(object):
wi_text = wi["content"]
if method == 1:
text = "{}{}".format(wi_text, game_text)
context.insert(0, {"type": "world_info", "text": wi_text})
context.insert(0, {"type": "world_info", "text": wi_text, "tokens": wi['token_length']})
else:
game_text = "{}{}".format(wi_text, game_text)
game_context.insert(0, {"type": "world_info", "text": wi_text})
game_context.insert(0, {"type": "world_info", "text": wi_text, "tokens": wi['token_length']})
self.worldinfo_v2.set_world_info_used(wi['uid'])
else:
used_all_tokens = True
@@ -339,11 +346,11 @@ class koboldai_vars(object):
#if we don't have enough actions to get to author's note depth then we just add it right before the game text
if len(action_text_split) < self.andepth and self.authornote != "":
game_text = "{}{}".format(authors_note_final, game_text)
game_context.insert(0, {"type": "authors_note", "text": authors_note_final})
game_context.insert(0, {"type": "authors_note", "text": authors_note_final, "tokens": authornote_length})
if self.useprompt:
text += prompt_text
context.append({"type": "prompt", "text": prompt_text})
context.append({"type": "prompt", "text": prompt_text, "tokens": prompt_length})
elif not used_all_tokens:
prompt_length = 0
prompt_text = ""
@@ -381,12 +388,12 @@ class koboldai_vars(object):
used_tokens+=0 if wi['token_length'] is None else wi['token_length']
used_world_info.append(wi['uid'])
wi_text = wi['content']
context.append({"type": "world_info", "text": wi_text})
context.append({"type": "world_info", "text": wi_text, "tokens": wi['token_length']})
text += wi_text
self.worldinfo_v2.set_world_info_used(wi['uid'])
text += prompt_text
context.append({"type": "prompt", "text": prompt_text})
context.append({"type": "prompt", "text": prompt_text, "tokens": prompt_length})
self.prompt_in_ai = True
else:
self.prompt_in_ai = False
@@ -805,6 +812,8 @@ class story_settings(settings):
ignore = self.koboldai_vars.calc_ai_text()
elif name == 'actions':
self.actions.story_settings = self
logger.debug("Calcing AI text after setting actions")
ignore = self.koboldai_vars.calc_ai_text()
elif name == 'story_name':
#reset the story id if we change the name
self.story_id = int.from_bytes(os.urandom(16), 'little', signed=True)
@@ -1159,6 +1168,8 @@ class KoboldStoryRegister(object):
process_variable_changes(self.socketio, "story", 'actions', data_to_send, None)
self.actions = temp
logger.debug("Calcing AI Text from Action load from json")
ignore = self.koboldai_vars.calc_ai_text()
self.set_game_saved()
if self.story_settings is not None:
self.story_settings.save_story()

50
static/koboldai.js
View File

@@ -2618,35 +2618,14 @@ function token_length(text) {
}
}
function calc_token_usage() {
memory_tokens = parseInt(document.getElementById("memory").getAttribute("story_memory_length"));
authors_notes_tokens = parseInt(document.getElementById("authors_notes").getAttribute("story_authornote_length"));
prompt_tokens = parseInt(document.getElementById("story_prompt").getAttribute("story_prompt_length"));
game_text_tokens = 0;
function calc_token_usage(soft_prompt_tokens, memory_tokens, authors_notes_tokens, prompt_tokens, game_text_tokens, world_info_tokens) {
submit_tokens = token_length(document.getElementById("input_text").value);
total_tokens = parseInt(document.getElementById('model_max_length_cur').value);
//find world info entries set to go to AI
world_info_tokens = 0;
for (wi of document.querySelectorAll(".world_info_card.used_in_game")) {
if (wi.getAttribute("uid") in world_info_data) {
world_info_tokens += world_info_data[wi.getAttribute("uid")].token_length;
}
}
//find game text tokens
var game_text_tokens = 0;
var game_text = document.getElementById('Selected Text').querySelectorAll(".within_max_length");
var game_text = Array.prototype.slice.call(game_text).reverse();
for (item of game_text) {
if (total_tokens - memory_tokens - authors_notes_tokens - world_info_tokens - prompt_tokens - game_text_tokens - submit_tokens > parseInt(item.getAttribute("token_length"))) {
game_text_tokens += parseInt(item.getAttribute("token_length"));
}
}
unused_tokens = total_tokens - memory_tokens - authors_notes_tokens - world_info_tokens - prompt_tokens - game_text_tokens - submit_tokens;
document.getElementById("soft_prompt_tokens").style.width = (soft_prompt_tokens/total_tokens)*100 + "%";
document.getElementById("soft_prompt_tokens").title = "Soft Prompt: "+soft_prompt_tokens;
document.getElementById("memory_tokens").style.width = (memory_tokens/total_tokens)*100 + "%";
document.getElementById("memory_tokens").title = "Memory: "+memory_tokens;
document.getElementById("authors_notes_tokens").style.width = (authors_notes_tokens/total_tokens)*100 + "%";
@@ -2899,6 +2878,13 @@ function update_bias_slider_value(slider) {
function update_context(data) {
$(".context-block").remove();
memory_tokens = 0;
authors_notes_tokens = 0;
prompt_tokens = 0;
game_text_tokens = 0;
world_info_tokens = 0;
soft_prompt_tokens = 0;
for (const entry of data) {
//console.log(entry);
let contextClass = "context-" + ({
@@ -2918,6 +2904,22 @@ function update_context(data) {
el.innerHTML = el.innerHTML.replaceAll("<br>", '<span class="material-icons-outlined context-symbol">keyboard_return</span>');
document.getElementById("context-container").appendChild(el);
switch (entry.type) {
case 'soft_prompt':
soft_prompt_tokens = entry.tokens;
case 'prompt':
prompt_tokens = entry.tokens;
case 'world_info':
world_info_tokens += entry.tokens;
case 'memory':
memory_tokens = entry.tokens;
case 'authors_note':
authors_notes_tokens = entry.tokens;
case 'action':
game_text_tokens += entry.tokens;
}
calc_token_usage(soft_prompt_tokens, memory_tokens, authors_notes_tokens, prompt_tokens, game_text_tokens, world_info_tokens);
}