diff --git a/.devcontainer/post-attach.sh b/.devcontainer/post-attach.sh index 0fd307a..1fda8ec 100644 --- a/.devcontainer/post-attach.sh +++ b/.devcontainer/post-attach.sh @@ -4,8 +4,8 @@ set -e # If running in Codespaces, check for necessary secrets and print error if missing if [ -v CODESPACES ]; then echo "🔐 Running in Codespaces - injecting secrets from Codespaces settings..." - if [ ! -v COPILOT_TOKEN ]; then - echo "⚠️ Running in Codespaces - please add COPILOT_TOKEN to your Codespaces secrets" + if [ ! -v AI_API_TOKEN ]; then + echo "⚠️ Running in Codespaces - please add AI_API_TOKEN to your Codespaces secrets" fi if [ ! -v GITHUB_PERSONAL_ACCESS_TOKEN ]; then echo "⚠️ Running in Codespaces - please add GITHUB_PERSONAL_ACCESS_TOKEN to your Codespaces secrets" diff --git a/src/run_seclab_agent.sh b/src/run_seclab_agent.sh index 8efdf1a..6cbc0de 100755 --- a/src/run_seclab_agent.sh +++ b/src/run_seclab_agent.sh @@ -10,4 +10,4 @@ mkdir -p data docker run -i \ --mount type=bind,src="$PWD",dst=/app \ - -e GITHUB_PERSONAL_ACCESS_TOKEN="$GITHUB_PERSONAL_ACCESS_TOKEN" -e COPILOT_TOKEN="$COPILOT_TOKEN" "ghcr.io/githubsecuritylab/seclab-taskflow-agent" "$@" + -e GITHUB_PERSONAL_ACCESS_TOKEN="$GITHUB_PERSONAL_ACCESS_TOKEN" -e AI_API_TOKEN="$AI_API_TOKEN" "ghcr.io/githubsecuritylab/seclab-taskflow-agent" "$@" diff --git a/src/seclab_taskflows/mcp_servers/gh_actions.py b/src/seclab_taskflows/mcp_servers/gh_actions.py index 01e574d..3c69a5c 100644 --- a/src/seclab_taskflows/mcp_servers/gh_actions.py +++ b/src/seclab_taskflows/mcp_servers/gh_actions.py @@ -46,8 +46,6 @@ def __repr__(self): unimportant_triggers = set(['pull_request', 'workflow_dispatch']) GITHUB_PERSONAL_ACCESS_TOKEN = os.getenv('GITHUB_PERSONAL_ACCESS_TOKEN', default='') -if not GITHUB_PERSONAL_ACCESS_TOKEN: - GITHUB_PERSONAL_ACCESS_TOKEN = os.getenv('COPILOT_TOKEN') ACTIONS_DB_DIR = Path(os.getenv('ACTIONS_DB_DIR', default='/app/my_data')) diff --git a/src/seclab_taskflows/mcp_servers/gh_code_scanning.py b/src/seclab_taskflows/mcp_servers/gh_code_scanning.py index f9fbd9b..db5a52f 100644 --- a/src/seclab_taskflows/mcp_servers/gh_code_scanning.py +++ b/src/seclab_taskflows/mcp_servers/gh_code_scanning.py @@ -26,8 +26,6 @@ mcp = FastMCP("GitHubCodeScanning") GITHUB_PERSONAL_ACCESS_TOKEN = os.getenv('GITHUB_PERSONAL_ACCESS_TOKEN', default='') -if not GITHUB_PERSONAL_ACCESS_TOKEN: - GITHUB_PERSONAL_ACCESS_TOKEN = os.getenv('COPILOT_TOKEN') CODEQL_DBS_BASE_PATH = Path(os.getenv('CODEQL_DBS_BASE_PATH', default='/app/my_data')) diff --git a/src/seclab_taskflows/mcp_servers/gh_file_viewer.py b/src/seclab_taskflows/mcp_servers/gh_file_viewer.py index 58a5891..1227db8 100644 --- a/src/seclab_taskflows/mcp_servers/gh_file_viewer.py +++ b/src/seclab_taskflows/mcp_servers/gh_file_viewer.py @@ -43,8 +43,6 @@ def __repr__(self): mcp = FastMCP("GitHubFileViewer") GITHUB_PERSONAL_ACCESS_TOKEN = os.getenv('GITHUB_PERSONAL_ACCESS_TOKEN', default='') -if not GITHUB_PERSONAL_ACCESS_TOKEN: - GITHUB_PERSONAL_ACCESS_TOKEN = os.getenv('COPILOT_TOKEN') SEARCH_RESULT_DIR = Path(os.getenv('SEARCH_RESULTS_DIR', default='/app/my_data')) diff --git a/src/seclab_taskflows/mcp_servers/local_gh_resources.py b/src/seclab_taskflows/mcp_servers/local_gh_resources.py index 05b6b01..79cfa0a 100644 --- a/src/seclab_taskflows/mcp_servers/local_gh_resources.py +++ b/src/seclab_taskflows/mcp_servers/local_gh_resources.py @@ -23,8 +23,6 @@ mcp = FastMCP("LocalGHResources") GITHUB_PERSONAL_ACCESS_TOKEN = os.getenv('GITHUB_PERSONAL_ACCESS_TOKEN') -if not GITHUB_PERSONAL_ACCESS_TOKEN: - GITHUB_PERSONAL_ACCESS_TOKEN = os.getenv('COPILOT_TOKEN') LOCAL_GH_DIR = Path(os.getenv('LOCAL_GH_DIR', default='/app/my_data'))