config.py

# Tencent is pleased to support the open source community by making tRPC-Agent-Python available.
#
# Copyright (C) 2026 Tencent. All rights reserved.
#
# tRPC-Agent-Python is licensed under Apache-2.0.
""" Agent config module"""

import os


def get_model_config() -> tuple[str, str, str]:
    """Get model config from environment variables"""
    api_key = os.getenv('TRPC_AGENT_API_KEY', '')
    url = os.getenv('TRPC_AGENT_BASE_URL', '')
    model_name = os.getenv('TRPC_AGENT_MODEL_NAME', '')
    if not api_key or not url or not model_name:
        raise ValueError('''TRPC_AGENT_API_KEY, TRPC_AGENT_BASE_URL,
                         and TRPC_AGENT_MODEL_NAME must be set in environment variables''')
    return api_key, url, model_name


def get_mempalace_config() -> dict:
    """Get MemPalace tool configuration from environment variables."""
    return {
        "palace_path": os.getenv("MEMPALACE_PALACE_PATH") or None,
        "kg_path": os.getenv("MEMPALACE_KG_PATH") or None,
        "wing": os.getenv("MEMPALACE_WING", "personal_assistant_alice"),
        "room": os.getenv("MEMPALACE_ROOM", "user_profile"),
    }