Bavarder/Bavarder
Bavarder/Bavarder
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Refactor OpenAI provider initialization and error handling

Browse Source
This commit is contained in:
0xmrtt 2024-02-25 00:57:53 +01:00
parent 62367ce9f4
commit c543ca1538
1 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
src/providers/openai.py
View File

@ -3,7 +3,6 @@ import openai
from openai import OpenAI from openai import OpenAI
import socket import socket
import os import os
import httpx
from gi.repository import Gtk, Adw, GLib from gi.repository import Gtk, Adw, GLib
@ -11,18 +10,23 @@ from gi.repository import Gtk, Adw, GLib
class BaseOpenAIProvider(BaseProvider): class BaseOpenAIProvider(BaseProvider):
model = None model = None
api_key_title = "API Key" api_key_title = "API Key"
client = OpenAI(
# This is the default and can be omitted
api_key=os.environ.get("OPENAI_API_KEY"),
)
def __init__(self, app, window): def __init__(self, app, window):
super().__init__(app, window) super().__init__(app, window)
try:
self.client = OpenAI(
api_key=os.environ.get("OPENAI_API_KEY"),
)
except openai.OpenAIError:
self.client = OpenAI(
api_key="",
)
if self.data.get("api_key"): if self.data.get("api_key"):
self.client.api_key = self.data["api_key"] self.client.api_key = self.data["api_key"]
if self.data.get("api_base"): if self.data.get("api_base"):
self.client.base_url = httpx.URL(self.data["api_base"]) self.client.base_url = self.data["api_base"]
def ask(self, prompt, chat): def ask(self, prompt, chat):
_chat = [] _chat = []