mirror of
https://github.com/yihong0618/bilingual_book_maker.git
synced 2025-06-06 11:35:49 +00:00
21 lines
564 B
Python
21 lines
564 B
Python
from openai import OpenAI
|
|
from .chatgptapi_translator import ChatGPTAPI
|
|
from os import linesep
|
|
from itertools import cycle
|
|
|
|
|
|
XAI_MODEL_LIST = [
|
|
"grok-beta",
|
|
]
|
|
|
|
|
|
class XAIClient(ChatGPTAPI):
|
|
def __init__(self, key, language, api_base=None, **kwargs) -> None:
|
|
super().__init__(key, language)
|
|
self.model_list = XAI_MODEL_LIST
|
|
self.api_url = str(api_base) if api_base else "https://api.x.ai/v1"
|
|
self.openai_client = OpenAI(api_key=key, base_url=self.api_url)
|
|
|
|
def rotate_model(self):
|
|
self.model = self.model_list[0]
|