From 426041f1cf1f9f7e209f2a66927ce0c7972e7e2d Mon Sep 17 00:00:00 2001 From: ebolam Date: Fri, 10 Feb 2023 16:22:37 -0500 Subject: [PATCH] Class Fix --- koboldai_settings.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/koboldai_settings.py b/koboldai_settings.py index 19fad8f3..98e3d792 100644 --- a/koboldai_settings.py +++ b/koboldai_settings.py @@ -96,6 +96,7 @@ class koboldai_vars(object): self._user_settings = user_settings(socketio) self._system_settings = system_settings(socketio, self) self._story_settings = {'default': story_settings(socketio, self)} + self._undefined_settings = undefined_settings() self._socketio = socketio self.tokenizer = None @@ -549,8 +550,10 @@ class koboldai_vars(object): setattr(self._user_settings, name, value) elif name in self._system_settings.__dict__: setattr(self._system_settings, name, value) - else: + elif name in self._story_settings[self.get_story_name()]: setattr(self._story_settings[self.get_story_name()], name, value) + else: + setattr(self._undefined_settings, name, value) def __getattr__(self, name): if name in self.__dict__: @@ -561,8 +564,10 @@ class koboldai_vars(object): return getattr(self._user_settings, name) elif name in self._system_settings.__dict__: return getattr(self._system_settings, name) - else: + elif name in self._story_settings[self.get_story_name()].__dict__: return getattr(self._story_settings[self.get_story_name()], name) + else: + return getattr(self._undefined_settings, name) class settings(object):