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.

import os


def get_model_config() -> tuple[str, str, str]:
    """Get model configuration from environment variables."""
    api_key = os.getenv("TRPC_AGENT_API_KEY", "")
    base_url = os.getenv("TRPC_AGENT_BASE_URL", "")
    model_name = os.getenv("TRPC_AGENT_MODEL_NAME", "deepseek-chat")

    if not api_key:
        print(
            "💡 Hint: TRPC_AGENT_API_KEY is not set. If your model provider requires it, export it or configure accordingly."
        )

    return api_key, base_url, model_name


def get_knowledge_config() -> dict[str, str]:
    """Get knowledge base (TRAG) configuration from environment variables.

    Returns:
        Dict with TRAG auth fields. Values are empty strings when not configured.
    """
    return {
        "namespace_code": os.getenv("TRAG_NAMESPACE", ""),
        "collection_code": os.getenv("TRAG_COLLECTION", ""),
        "api_key": os.getenv("TRAG_TOKEN", ""),
        "base_url": os.getenv("TRAG_BASE_URL", ""),
        "rag_code": os.getenv("TRAG_RAG_CODE", ""),
    }