fix: stop thread
Build / Flatpak (x86_64) (push) Failing after 49m32s
Details
Build / Flatpak (x86_64) (push) Failing after 49m32s
Details
This commit is contained in:
parent
b769d3e910
commit
2410fd3fd4
|
@ -234,7 +234,11 @@ class BavarderApplication(Adw.Application):
|
||||||
if not self.setup_chat(): # NO MODELS:
|
if not self.setup_chat(): # NO MODELS:
|
||||||
return _("Please download a model from Preferences by clicking on the Dot Menu at the top!")
|
return _("Please download a model from Preferences by clicking on the Dot Menu at the top!")
|
||||||
else:
|
else:
|
||||||
with self.model.chat_session():
|
for p in ["Hi", "Hello"]:
|
||||||
|
if p.lower() in prompt.lower():
|
||||||
|
return _("Hello, I am Bavarder, a Chit-Chat AI")
|
||||||
|
system_template = """A chat between a curious user and an artificial intelligence assistant."""
|
||||||
|
with self.model.chat_session(system_template):
|
||||||
self.model.current_chat_session = chat["content"].copy()
|
self.model.current_chat_session = chat["content"].copy()
|
||||||
response = self.model.generate(prompt=prompt, top_k=1)
|
response = self.model.generate(prompt=prompt, top_k=1)
|
||||||
else:
|
else:
|
||||||
|
|
|
@ -25,7 +25,7 @@ class KillableThread(threading.Thread):
|
||||||
def localtrace(self, frame, event, arg):
|
def localtrace(self, frame, event, arg):
|
||||||
if self.killed:
|
if self.killed:
|
||||||
if event == 'line':
|
if event == 'line':
|
||||||
raise SystemExit()
|
raise Exception("Killed")
|
||||||
return self.localtrace
|
return self.localtrace
|
||||||
|
|
||||||
def kill(self):
|
def kill(self):
|
||||||
|
|
|
@ -374,8 +374,8 @@ class BavarderWindow(Adw.ApplicationWindow):
|
||||||
del self.t
|
del self.t
|
||||||
self.toast.dismiss()
|
self.toast.dismiss()
|
||||||
except AttributeError: # nothing to stop
|
except AttributeError: # nothing to stop
|
||||||
print("Nothing to stop")
|
pass
|
||||||
except SystemExit:
|
except Exception:
|
||||||
self.t.join()
|
self.t.join()
|
||||||
del self.t
|
del self.t
|
||||||
self.toast.dismiss()
|
self.toast.dismiss()
|
||||||
|
|
Loading…
Reference in New Issue