Update

Author: yjg30737

pyqt_openai/__init__.py

@@ -1139,6 +1139,7 @@ def move_bin(filename, dst_dir):
             "model",
             "width",
             "height",
+            "provider",
             "prompt",
             "negative_prompt",
             "n",

pyqt_openai/aboutDialog.py

@@ -54,6 +54,7 @@ def __initUi(self):
         <p>Powered by PySide6</p>
         <p>Powered by GPT4Free</p>
         <p>Powered by LiteLLM</p>
+        <p>Powered by LlamaIndex</p>
         """
         )
 

pyqt_openai/chat_widget/right_sidebar/usingAPIPage.py

@@ -40,7 +40,7 @@
     init_llama,
 )
 from pyqt_openai.widgets.linkLabel import LinkLabel
-from pyqt_openai.widgets.modernButton import ModernButton
+from pyqt_openai.widgets.APIInputButton import APIInputButton
 
 
 class UsingAPIPage(QWidget):
@@ -123,15 +123,12 @@ def __initUi(self):
         modelSearchBar.setText(self.__model)
         modelSearchBar.textChanged.connect(self.__modelChanged)
 
-        prefixListBtn = ModernButton()
-        prefixListBtn.setText("Prefix List")
-
         lay = QHBoxLayout()
         lay.addWidget(QLabel(LangClass.TRANSLATIONS["Model"]))
         lay.addWidget(modelSearchBar)
         lay.setContentsMargins(0, 0, 0, 0)
 
-        setApiBtn = ModernButton()
+        setApiBtn = APIInputButton()
         # TODO LANGUAGE
         setApiBtn.setText("Set API Key")
 

pyqt_openai/dalle_widget/dalleRightSideBar.py

@@ -13,7 +13,7 @@
 from pyqt_openai.dalle_widget.dalleThread import DallEThread
 from pyqt_openai.lang.translations import LangClass
 from pyqt_openai.widgets.imageControlWidget import ImageControlWidget
-from pyqt_openai.widgets.modernButton import ModernButton
+from pyqt_openai.widgets.APIInputButton import APIInputButton
 
 
 class DallERightSideBarWidget(ImageControlWidget):
@@ -51,7 +51,7 @@ def _initUi(self):
         super()._initUi()
 
         # TODO LANGUAGE
-        self.__setApiBtn = ModernButton()
+        self.__setApiBtn = APIInputButton()
         self.__setApiBtn.setText("Set API Key")
 
         self.__promptTypeToShowRadioGrpBox = QGroupBox(

pyqt_openai/replicate_widget/replicateRightSideBar.py

@@ -13,7 +13,7 @@
 from pyqt_openai.lang.translations import LangClass
 from pyqt_openai.replicate_widget.replicateThread import ReplicateThread
 from pyqt_openai.widgets.imageControlWidget import ImageControlWidget
-from pyqt_openai.widgets.modernButton import ModernButton
+from pyqt_openai.widgets.APIInputButton import APIInputButton
 
 
 class ReplicateRightSideBarWidget(ImageControlWidget):
@@ -49,7 +49,7 @@ def _initUi(self):
         super()._initUi()
 
         # TODO LANGUAGE
-        self.__setApiBtn = ModernButton()
+        self.__setApiBtn = APIInputButton()
         self.__setApiBtn.setText("Set API Key")
 
         self.__modelTextEdit = QPlainTextEdit()

pyqt_openai/settings_dialog/apiWidget.py

@@ -39,7 +39,7 @@ def __initVal(self):
     def __initUi(self):
         self.setWindowTitle("API Key")
 
-        columns = ["Provider", "API Key", "Get API Key"]
+        columns = ["Provider", "API Key", "Manual URL"]
         self.__tableWidget = QTableWidget()
         self.__tableWidget.setColumnCount(len(columns))
         self.__tableWidget.setHorizontalHeaderLabels(columns)
@@ -51,6 +51,8 @@ def __initUi(self):
         for i, obj in enumerate(self.__api_keys):
             self.__tableWidget.insertRow(i)
             modelItem = QTableWidgetItem(obj["display_name"])
+            # Make item not editable
+            modelItem.setFlags(modelItem.flags() & ~Qt.ItemFlag.ItemIsEditable)
             self.__tableWidget.setItem(i, 0, modelItem)
 
             apiKeyLineEdit = QLineEdit(obj["api_key"])

pyqt_openai/util/common.py

@@ -66,7 +66,7 @@
     OPENAI_CHAT_ENDPOINT,
     STT_MODEL,
     DEFAULT_DATETIME_FORMAT,
-    DEFAULT_TOKEN_CHUNK_SIZE,
+    DEFAULT_TOKEN_CHUNK_SIZE, DEFAULT_API_CONFIGS,
 )
 from pyqt_openai.config_loader import CONFIG_MANAGER
 from pyqt_openai.globals import (
@@ -1335,3 +1335,7 @@ def stream_to_speakers(voice_provider, input_args):
     stream_thread = TTSThread(voice_provider, input_args)
     pyqt_openai.util.common.current_tts_thread = stream_thread
     return stream_thread
+
+
+def get_litellm_prefixes():
+    return [{'Provider': obj.get('display_name', ''), 'Prefix': obj.get('prefix', '')} for obj in DEFAULT_API_CONFIGS]

pyqt_openai/widgets/modernButton.py renamed to pyqt_openai/widgets/APIInputButton.py

@@ -5,7 +5,7 @@
 from pyqt_openai.settings_dialog.settingsDialog import SettingsDialog
 
 
-class ModernButton(QPushButton):
+class APIInputButton(QPushButton):
     def __init__(self, base_color="#007BFF"):
         super().__init__()
         self.setObjectName("modernButton")

pyqt_openai/widgets/modelInputManualDialog.py

@@ -0,0 +1,44 @@
+from PySide6.QtCore import Qt
+from PySide6.QtGui import QFont
+from PySide6.QtWidgets import QDialog, QLabel, QHBoxLayout, QTableWidget
+
+from pyqt_openai import SMALL_LABEL_PARAM
+from pyqt_openai.util.common import get_litellm_prefixes
+
+
+class ModelInputManualDialog(QDialog):
+    def __init__(self, parent=None):
+        super().__init__(parent)
+        self.__initVal()
+        self.__initUi()
+
+    def __initVal(self):
+        self.__warningMessage = (
+            "💡 <b>Tip</b> For models other than OpenAI and Anthropic, enter the model name as `[ProviderName]/[ModelName]`.\n\n"
+            "🔗 For details on `ProviderName` and `ModelName`, check out the "
+            "<a href='https://docs.litellm.ai/docs/providers'>LiteLLM documentation</a>! 😊\n\n"
+            "⚠️ Note: Some models may not support <b>JSON Mode</b> or <b>LlamaIndex</b> features."
+        )
+
+    def __initUi(self):
+        self.__warningLbl = QLabel()
+        self.__warningLbl.setStyleSheet("color: orange;")
+        self.__warningLbl.setOpenExternalLinks(True)
+        self.__warningLbl.setFont(QFont(SMALL_LABEL_PARAM))
+        self.__warningLbl.setText(self.__warningMessage)
+        self.__warningLbl.setTextInteractionFlags(
+            Qt.TextInteractionFlag.TextSelectableByMouse
+        )
+
+        # prefixTable = QTableWidget()
+        # prefixes = get_litellm_prefixes()
+        # prefixTable.setColumnCount(len(list(prefixes[0].keys())))
+        # prefixTable.setHorizontalHeaderLabels(list(prefixes[0].keys()))
+        # for prefix in prefixes:
+        #     prefixTable.insertRow(prefixTable.rowCount())
+        #     prefixTable.setItem(prefixTable.rowCount() - 1, 0, QLabel(prefix))
+
+        lay = QHBoxLayout()
+        lay.addWidget(self.__warningLbl)
+        # lay.addWidget(prefixTable)
+        self.setLayout(lay)