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

Support title in aidg import

Browse Source
This commit is contained in:
somebody
2022-10-20 18:04:11 -05:00
parent 5f488c0042
commit de3c408cc1
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
aiserver.py
View File

@@ -297,6 +297,7 @@ class ImportBuffer:
authors_note: Optional[str] = None authors_note: Optional[str] = None
notes: Optional[str] = None notes: Optional[str] = None
world_infos: Optional[dict] = None world_infos: Optional[dict] = None
title: Optional[str] = None
@dataclass @dataclass
class PromptPlaceholder: class PromptPlaceholder:
@@ -416,6 +417,7 @@ class ImportBuffer:
self.memory = j["memory"] self.memory = j["memory"]
self.authors_note = j["authorsNote"] self.authors_note = j["authorsNote"]
self.notes = j["description"] self.notes = j["description"]
self.title = j["title"] or "Imported Story"
self.world_infos = [] self.world_infos = []
@@ -449,6 +451,7 @@ class ImportBuffer:
koboldai_vars.memory = self.memory or "" koboldai_vars.memory = self.memory or ""
koboldai_vars.authornote = self.authors_note or "" koboldai_vars.authornote = self.authors_note or ""
koboldai_vars.notes = self.notes koboldai_vars.notes = self.notes
koboldai_vars.story_name = self.title
for wi in self.world_infos: for wi in self.world_infos:
koboldai_vars.worldinfo_v2.add_item( koboldai_vars.worldinfo_v2.add_item(