AILanguageCompanion / config.py
koura718's picture
Change DeepSeek Client
46e66e9
import os
from dataclasses import dataclass
import pytz
@dataclass
class Config:
# Version
VERSION = "1.0.0"
# the newest OpenAI model is "gpt-4o" which was released May 13, 2024.
OPENAI_MODEL = "gpt-4o-mini-2024-07-18" # Changed from "gpt-4o-mini-2024-07-18" to "gpt-4o" "o1-mini-2024-09-12"
GEMINI_MODEL = "google/gemini-2.0-flash-exp:free"
AUTO_MODEL = "openrouter/auto"
# CLAUDE_MODEL = "anthropic/claude-3-sonnet:free"
CLAUDE_MODEL = "anthropic/claude-3.5-sonnet"
OPENROUTER_API_BASE = "https://openrouter.ai/api/v1"
SUPPORTED_LANGUAGES = ["en", "ja"]
DEFAULT_LANGUAGE = "ja" # Changed from "en" to "ja"
# Timezone settings
DEFAULT_TIMEZONE = "Asia/Tokyo"
SUPPORTED_TIMEZONES = [
"Asia/Tokyo",
"America/New_York",
"America/Los_Angeles",
"Europe/London",
"Europe/Paris",
"Asia/Shanghai",
"Asia/Singapore",
"Australia/Sydney",
"Pacific/Auckland"
]
# DeepSeek model configuration
DEEPSEEK_MODEL = "deepseek-chat"
DEEPSEEK_API_BASE = "https://api.deepseek.com/v1/chat/completions"
# Chat context settings
MAX_HISTORY_CHATS = 10
MAX_CONTEXT_LENGTH = 4096 # Maximum token length for context
MEMORY_SUMMARY_TOKENS = 150 # Length of conversation summaries
CONTEXT_WINDOW_MESSAGES = 10 # Number of messages to keep in immediate context
@staticmethod
def get_deepseek_key():
return os.getenv("DEEPSEEK_API_KEY", "")
@staticmethod
def get_openai_key():
return os.getenv("OPENAI_API_KEY", "")
@staticmethod
def get_openrouter_key():
return os.getenv("OPENROUTER_API_KEY", "")