diff --git a/bots/scheduler.py b/bots/scheduler.py index 8053eb0..4010f39 100644 --- a/bots/scheduler.py +++ b/bots/scheduler.py @@ -51,6 +51,7 @@ logger = logging.getLogger(__name__) TELEGRAM_BOT_TOKEN = os.getenv('TELEGRAM_BOT_TOKEN', '') TELEGRAM_CHAT_ID = os.getenv('TELEGRAM_CHAT_ID', '') ANTHROPIC_API_KEY = os.getenv('ANTHROPIC_API_KEY', '') +ANTHROPIC_BASE_URL = os.getenv('ANTHROPIC_BASE_URL', '') _claude_client: anthropic.Anthropic | None = None _conversation_history: dict[int, list] = {} @@ -1641,7 +1642,10 @@ async def handle_text(update: Update, context: ContextTypes.DEFAULT_TYPE): global _claude_client if _claude_client is None: - _claude_client = anthropic.Anthropic(api_key=ANTHROPIC_API_KEY) + kwargs = {'api_key': ANTHROPIC_API_KEY} + if ANTHROPIC_BASE_URL: + kwargs['base_url'] = ANTHROPIC_BASE_URL + _claude_client = anthropic.Anthropic(**kwargs) history = _conversation_history.setdefault(chat_id, []) history.append({"role": "user", "content": text})