mirror of
https://github.com/snailyp/gemini-balance.git
synced 2026-05-12 02:19:59 +08:00
71af1db330
主要更新： 添加图像模型支持 新增MODEL_IMAGE配置项 在模型列表中添加gemini-2.0-flash-exp-image模型 修改ModelService以支持图像模型 增强图像处理能力 添加PicGoUploader类用于图像上传 实现图像响应处理逻辑(_extract_image_data) 支持base64图像数据的解码与上传 优化请求与响应处理 为图像模型添加特殊处理逻辑 修改API客户端以支持图像模型 更新GeminiRequest默认值 安全性调整 将TOOLS_CODE_EXECUTION_ENABLED默认设置为false
40 lines
1.2 KiB
Python
40 lines
1.2 KiB
Python
from pydantic_settings import BaseSettings
|
|
from typing import List
|
|
|
|
|
|
class Settings(BaseSettings):
|
|
API_KEYS: List[str]
|
|
ALLOWED_TOKENS: List[str]
|
|
BASE_URL: str = "https://generativelanguage.googleapis.com/v1beta"
|
|
MODEL_SEARCH: List[str] = ["gemini-2.0-flash-exp"]
|
|
MODEL_IMAGE: List[str] = ["gemini-2.0-flash-exp"]
|
|
TOOLS_CODE_EXECUTION_ENABLED: bool = False
|
|
SHOW_SEARCH_LINK: bool = True
|
|
SHOW_THINKING_PROCESS: bool = True
|
|
AUTH_TOKEN: str = ""
|
|
MAX_FAILURES: int = 3
|
|
PAID_KEY: str = ""
|
|
CREATE_IMAGE_MODEL: str = "imagen-3.0-generate-002"
|
|
UPLOAD_PROVIDER: str = "smms"
|
|
SMMS_SECRET_TOKEN: str = ""
|
|
PICGO_API_KEY: str = ""
|
|
TEST_MODEL: str = "gemini-1.5-flash"
|
|
|
|
# 流式输出优化器配置
|
|
STREAM_MIN_DELAY: float = 0.016
|
|
STREAM_MAX_DELAY: float = 0.024
|
|
STREAM_SHORT_TEXT_THRESHOLD: int = 10
|
|
STREAM_LONG_TEXT_THRESHOLD: int = 50
|
|
STREAM_CHUNK_SIZE: int = 5
|
|
|
|
def __init__(self):
|
|
super().__init__()
|
|
if not self.AUTH_TOKEN:
|
|
self.AUTH_TOKEN = self.ALLOWED_TOKENS[0] if self.ALLOWED_TOKENS else ""
|
|
|
|
class Config:
|
|
env_file = ".env"
|
|
|
|
|
|
settings = Settings()
|