(self, OPENAI_API_KEY: Optional[str] = None, OPENAI_API_BASE: Optional[str] = None, MODEL_TYPE="gpt-3.5-turbo")
| 12 | |
| 13 | class ChatGPT(nn.Module): |
| 14 | def __init__(self, OPENAI_API_KEY: Optional[str] = None, OPENAI_API_BASE: Optional[str] = None, MODEL_TYPE="gpt-3.5-turbo"): |
| 15 | super().__init__() |
| 16 | self.config_file = "config.json" |
| 17 | self.app_keys = [] |
| 18 | if OPENAI_API_KEY is None or OPENAI_API_KEY == '': |
| 19 | OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY") |
| 20 | if OPENAI_API_KEY is not None and OPENAI_API_KEY != "": |
| 21 | openai.api_key = OPENAI_API_KEY |
| 22 | self.OPENAI_API_KEY = OPENAI_API_KEY |
| 23 | if OPENAI_API_BASE is None or OPENAI_API_BASE == '': |
| 24 | OPENAI_API_BASE = os.environ.get("OPENAI_API_BASE") |
| 25 | if OPENAI_API_BASE is not None and OPENAI_API_BASE != "": |
| 26 | openai.api_base = OPENAI_API_BASE |
| 27 | self.OPENAI_API_BASE = OPENAI_API_BASE |
| 28 | self.OPENAI_API_KEY = OPENAI_API_KEY |
| 29 | self.OPENAI_API_BASE = OPENAI_API_BASE |
| 30 | self.MODEL_TYPE = MODEL_TYPE |
| 31 | self.MAX_TOKEN = 8000 |
| 32 | self.contextual_conversations = [] |
| 33 | |
| 34 | self.initConfig() |
| 35 | if self.OPENAI_API_KEY is None and len(self.app_keys) > 0: |
| 36 | self.find_valid_key() |
| 37 | |
| 38 | def initConfig(self): |
| 39 | if os.path.exists(self.config_file): |
nothing calls this directly
no test coverage detected