ds-mcp-server / utils /config.py
amirkiarafiei's picture
refactor: update environment variable handling for configuration in main.py and config.py
3da7106
raw
history blame contribute delete
632 Bytes
import os
from dotenv import load_dotenv
load_dotenv()
# API Configuration
API_CONFIG = {
"api_base_url": str(os.getenv("DS_API_BASE_URL", "https://dnext-ds-api.hf.space")),
"api_multimodule_endpoint": str(os.getenv("DS_API_MULTIMODULE_ENDPOINT", "/mcp_multi_module_bug_localization")),
"api_singlemodule_endpoint": str(os.getenv("DS_API_SINGLEMODULE_ENDPOINT", "/mcp_single_module_bug_localization")),
"api_searchspace_endpoint": str(os.getenv("DS_API_SEARCHSPACE_ENDPOINT", "/mcp_search_space_routing")),
"hf_access_token": os.getenv("HF_ACCESS_TOKEN", None),
"timeout": int(os.getenv("TIMEOUT", 120)),
}