mirror of
https://github.com/snailyp/gemini-balance.git
synced 2026-09-07 00:36:36 +08:00
feat: 添加OpenAI消息转换器组件
This commit is contained in:
@@ -3,6 +3,7 @@
|
|||||||
import json
|
import json
|
||||||
from typing import Dict, Any, AsyncGenerator, List, Union
|
from typing import Dict, Any, AsyncGenerator, List, Union
|
||||||
from app.core.logger import get_openai_logger
|
from app.core.logger import get_openai_logger
|
||||||
|
from app.services.chat.message_converter import OpenAIMessageConverter
|
||||||
from app.services.chat.response_handler import OpenAIResponseHandler
|
from app.services.chat.response_handler import OpenAIResponseHandler
|
||||||
from app.services.chat.api_client import GeminiApiClient
|
from app.services.chat.api_client import GeminiApiClient
|
||||||
from app.schemas.openai_models import ChatRequest, ImageGenerationRequest
|
from app.schemas.openai_models import ChatRequest, ImageGenerationRequest
|
||||||
@@ -85,9 +86,8 @@ def _build_payload(
|
|||||||
|
|
||||||
class OpenAIChatService:
|
class OpenAIChatService:
|
||||||
"""聊天服务"""
|
"""聊天服务"""
|
||||||
|
|
||||||
def __init__(self, base_url: str, key_manager: KeyManager = None):
|
def __init__(self, base_url: str, key_manager: KeyManager = None):
|
||||||
|
self.message_converter = OpenAIMessageConverter()
|
||||||
self.response_handler = OpenAIResponseHandler(config=None)
|
self.response_handler = OpenAIResponseHandler(config=None)
|
||||||
self.api_client = GeminiApiClient(base_url)
|
self.api_client = GeminiApiClient(base_url)
|
||||||
self.key_manager = key_manager
|
self.key_manager = key_manager
|
||||||
|
|||||||
Reference in New Issue
Block a user