mirror of
https://github.com/JefferyHcool/BiliNote.git
synced 2026-08-07 06:04:29 +08:00
6ff8b4d90f
- 实现 GroqTranscriber 类以支持 Groq 语音转录服务 - 新增异常处理中间件以提高系统稳定性 - 优化笔记生成流程，增加错误处理和日志记录 - 添加思维导图功能和相关组件 -重构 Markdown 查看器以支持切换视图模式
24 lines
690 B
Python
24 lines
690 B
Python
from typing import Optional, Union
|
|
|
|
from openai import OpenAI
|
|
|
|
class OpenAICompatibleProvider:
|
|
def __init__(self, api_key: str, base_url: str, model: Union[str, None]=None):
|
|
self.client = OpenAI(api_key=api_key, base_url=base_url)
|
|
self.model = model
|
|
|
|
@property
|
|
def get_client(self):
|
|
return self.client
|
|
|
|
@staticmethod
|
|
def test_connection(api_key: str, base_url: str) -> bool:
|
|
print(api_key)
|
|
try:
|
|
client = OpenAI(api_key=api_key, base_url=base_url)
|
|
|
|
client.models.list()
|
|
return True
|
|
except Exception as e:
|
|
print(f"Error connecting to OpenAI API: {e}")
|
|
return False |