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

Merge pull request #134 from one-some/ui2-club-import-overhaul

Browse Source 
Fix type annotation on import buffer
This commit is contained in:
ebolam
2022-09-16 17:31:04 -04:00
committed by GitHub
parent eabd418439 c43b24042d
commit aadfd488e5
1 changed files with 55 additions and 59 deletions
Download Patch File Download Diff File Expand all files Collapse all files

114
aiserver.py
View File

@@ -268,74 +268,74 @@ class ImportBuffer:
"description" "description"
]} ]}
#def request_client_configuration(self, placeholders: list[PromptPlaceholder]) -> None: def request_client_configuration(self, placeholders: List[PromptPlaceholder]) -> None:
# emit("request_prompt_config", [x.to_json() for x in placeholders], broadcast=False, room="UI_2") emit("request_prompt_config", [x.to_json() for x in placeholders], broadcast=False, room="UI_2")
#def extract_placeholders(self, text: str) -> list[PromptPlaceholder]: def extract_placeholders(self, text: str) -> List[PromptPlaceholder]:
# placeholders = [] placeholders = []
# for match in re.finditer(r"\${(.*?)}", text): for match in re.finditer(r"\${(.*?)}", text):
# ph_text = match.group(1) ph_text = match.group(1)
# try: try:
# ph_order, ph_text = ph_text.split("#") ph_order, ph_text = ph_text.split("#")
# except ValueError: except ValueError:
# ph_order = None ph_order = None
# if "[" not in ph_text: if "[" not in ph_text:
# ph_id = ph_text ph_id = ph_text
# Already have it! # Already have it!
# if any([x.id == ph_id for x in placeholders]): if any([x.id == ph_id for x in placeholders]):
# continue continue
# Apparently, none of these characters are supported: # Apparently, none of these characters are supported:
# "${}[]#:@^|", however I have found some prompts using these, # "${}[]#:@^|", however I have found some prompts using these,
# so they will be allowed. # so they will be allowed.
# for char in "${}[]": for char in "${}[]":
# if char in ph_text: if char in ph_text:
# print("[eph] Weird char") print("[eph] Weird char")
# print(f"{char=}") print(f"{char=}")
# print(f"{ph_id=}") print(f"{ph_id=}")
# return return
# placeholders.append(self.PromptPlaceholder( placeholders.append(self.PromptPlaceholder(
# id=ph_id, id=ph_id,
# order=int(ph_order) if ph_order else None, order=int(ph_order) if ph_order else None,
# )) ))
# continue continue
# ph_id, _ = ph_text.split("[") ph_id, _ = ph_text.split("[")
# ph_text = ph_text.replace(ph_id, "", 1) ph_text = ph_text.replace(ph_id, "", 1)
# Already have it! # Already have it!
# if any([x.id == ph_id for x in placeholders]): if any([x.id == ph_id for x in placeholders]):
# continue continue
# Match won't match it for some reason (???), so we use finditer and next() # Match won't match it for some reason (???), so we use finditer and next()
# try: try:
# default_match = next(re.finditer(r"\[(.*?)\]", ph_text)) default_match = next(re.finditer(r"\[(.*?)\]", ph_text))
# except StopIteration: except StopIteration:
# print("[eph] Weird brackets") print("[eph] Weird brackets")
# return placeholders return placeholders
# ph_default = default_match.group(1) ph_default = default_match.group(1)
# ph_text = ph_text.replace(default_match.group(0), "") ph_text = ph_text.replace(default_match.group(0), "")
# try: try:
# ph_title, ph_desc = ph_text.split(":") ph_title, ph_desc = ph_text.split(":")
# except ValueError: except ValueError:
# ph_title = ph_text or None ph_title = ph_text or None
# ph_desc=None ph_desc=None
# placeholders.append(self.PromptPlaceholder( placeholders.append(self.PromptPlaceholder(
# id=ph_id, id=ph_id,
# order=int(ph_order) if ph_order else None, order=int(ph_order) if ph_order else None,
# default=ph_default, default=ph_default,
# title=ph_title, title=ph_title,
# description=ph_desc description=ph_desc
# )) ))
# return placeholders return placeholders
def _replace_placeholders(self, text: str, ph_ids: dict): def _replace_placeholders(self, text: str, ph_ids: dict):
for ph_id, value in ph_ids.items(): for ph_id, value in ph_ids.items():
@@ -402,11 +402,7 @@ class ImportBuffer:
koboldai_vars.authornote = self.authors_note or "" koboldai_vars.authornote = self.authors_note or ""
koboldai_vars.notes = self.notes koboldai_vars.notes = self.notes
# ???: Was this supposed to increment?
num = 0
for wi in self.world_infos: for wi in self.world_infos:
# koboldai_vars.worldinfo += self.world_infos
koboldai_vars.worldinfo_v2.add_item( koboldai_vars.worldinfo_v2.add_item(
wi["key_list"][0], wi["key_list"][0],
wi["key_list"], wi["key_list"],