feat: add hugging chat mask
This commit is contained in:
parent
5d073ab5c4
commit
bc2fab30a8
11
src/main.py
11
src/main.py
|
@ -1126,14 +1126,19 @@ Close All Without Dialog: {self.close_all_without_dialog}
|
|||
|
||||
self.win.banner.set_revealed(False)
|
||||
|
||||
for an in self.annoucements.values():
|
||||
for key, an in self.annoucements.items():
|
||||
if an["provider"] == self.provider:
|
||||
if an["status"] == "open":
|
||||
match an["action"]:
|
||||
case "error": # show an error banner with a button to open settings
|
||||
self.win.banner.set_title(an["message"])
|
||||
self.win.banner.props.button_label = "Open settings"
|
||||
self.win.banner.connect("button-clicked", self.on_preferences_action)
|
||||
self.win.banner.set_revealed(True)
|
||||
return
|
||||
case _:
|
||||
self.win.banner.set_title(an["message"])
|
||||
self.win.banner.set_revealed(True)
|
||||
del self.annoucements[key]
|
||||
break
|
||||
|
||||
self.prompt = self.win.prompt_text_view.get_buffer().props.text.strip()
|
||||
|
@ -1151,6 +1156,8 @@ Close All Without Dialog: {self.close_all_without_dialog}
|
|||
response = self.providers[self.provider].ask(self.prompt)
|
||||
except GLib.Error as e:
|
||||
response = e.message
|
||||
except KeyError:
|
||||
del self.providers[self.provider]
|
||||
GLib.idle_add(cleanup, response)
|
||||
|
||||
def cleanup(response):
|
||||
|
|
|
@ -72,6 +72,7 @@ class Preferences(Adw.PreferencesWindow):
|
|||
"""Callback for the allow_remote_fetching_switch toggled event."""
|
||||
if widget.get_active():
|
||||
self.settings.set_boolean("allow-remote-fetching", True)
|
||||
self.app.load_annoucements()
|
||||
self.app.allow_remote_fetching = True
|
||||
else:
|
||||
self.settings.set_boolean("allow-remote-fetching", False)
|
||||
|
|
|
@ -1,4 +1,4 @@
|
|||
#from .huggingchat import HuggingChatProvider
|
||||
# from .huggingchat import HuggingChatProvider
|
||||
from .baichat import BAIChatProvider
|
||||
from .openaigpt35turbo import OpenAIGPT35TurboProvider
|
||||
from .openaigpt4 import OpenAIGPT4Provider
|
||||
|
@ -7,13 +7,13 @@ from .openaitextdavinci003 import OpenAITextDavinci003
|
|||
from .alpacalora import AlpacaLoRAProvider
|
||||
from .hfgoogleflant5xxl import HuggingFaceGoogleFlanT5XXLProvider
|
||||
from .hfgoogleflanu12 import HuggingFaceGoogleFlanU12Provider
|
||||
from .hfopenassistantsft1pythia12b import HuggingFaceOpenAssistantSFT1PythiaProvider
|
||||
from .hfopenassistantsft1pythia12b import HuggingFaceOpenAssistantSFT1PythiaProvider, HuggingChatMask
|
||||
from .hfgpt2 import HuggingFaceGPT2Provider
|
||||
from .hfdialogpt import HuggingFaceDialoGPTLargeProvider
|
||||
from .bard import BardProvider
|
||||
# from .bard import BardProvider
|
||||
from .hfgpt2large import HuggingFaceGPT2LargeProvider
|
||||
from .hfgpt2xl import HuggingFaceGPT2XLProvider
|
||||
#from .starcoder import StarCoderProvider
|
||||
# from .starcoder import StarCoderProvider
|
||||
|
||||
PROVIDERS = {
|
||||
"alpacalora": AlpacaLoRAProvider,
|
||||
|
@ -27,7 +27,7 @@ PROVIDERS = {
|
|||
"hfgpt2large": HuggingFaceGPT2LargeProvider,
|
||||
"hfgpt2xl": HuggingFaceGPT2XLProvider,
|
||||
"hfopenassistantsft1pythia12b": HuggingFaceOpenAssistantSFT1PythiaProvider,
|
||||
"huggingchat": HuggingFaceOpenAssistantSFT1PythiaProvider, # hugging chat is replaced by open assistant
|
||||
"huggingchat": HuggingChatMask, # hugging chat is replaced by open assistant
|
||||
"openaigpt35turbo": OpenAIGPT35TurboProvider,
|
||||
"openaigpt4": OpenAIGPT4Provider,
|
||||
"openaitextdavinci003": OpenAITextDavinci003,
|
||||
|
|
|
@ -47,3 +47,8 @@ class HuggingFaceOpenAssistantSFT1PythiaProvider(BaseHFProvider):
|
|||
print(response)
|
||||
GLib.idle_add(self.update_response, response)
|
||||
return response
|
||||
|
||||
class HuggingChatMask(HuggingFaceOpenAssistantSFT1PythiaProvider):
|
||||
url = "https://bavarder.codeberg.page/help/huggingchat"
|
||||
name = "Open-Assistant SFT-1 12B Model (HuggingChat)"
|
||||
slug = "huggingchat"
|
Loading…
Reference in New Issue
Block a user