Skip to content

Commit 29eb221

Browse files
Use generated data dirs
1 parent 009c3a2 commit 29eb221

File tree

2 files changed

+5
-4
lines changed

2 files changed

+5
-4
lines changed

.gitignore

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -134,7 +134,7 @@ celerybeat.pid
134134
*.sage.py
135135

136136
# Environments
137-
*.env
137+
.env
138138
.venv
139139
env/
140140
venv/

src/seclab_taskflows/mcp_servers/codeql_python/mcp_server.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@
1010
filemode='a'
1111
)
1212
from seclab_taskflow_agent.mcp_servers.codeql.client import run_query, _debug_log
13+
from seclab_taskflow_agent.path_utils import mcp_data_dir
1314

1415
from pydantic import Field
1516
#from mcp.server.fastmcp import FastMCP, Context
@@ -26,10 +27,10 @@
2627
from .codeql_sqlite_models import Base, Source
2728
from ..utils import process_repo
2829

29-
MEMORY = Path(os.getenv('CODEQL_SQLITE_DIR', default='/app/my_data'))
30-
mcp = FastMCP("CodeQL-Python")
30+
MEMORY = mcp_data_dir('seclab-taskflow-agent', 'codeql', 'DATA_DIR')
31+
CODEQL_DBS_BASE_PATH = mcp_data_dir('seclab-taskflow-agent', 'codeql', 'CODEQL_DBS_BASE_PATH')
3132

32-
CODEQL_DBS_BASE_PATH = Path(os.getenv('CODEQL_DBS_BASE_PATH', default='/workspaces/seclab-taskflow-agent/my_data'))
33+
mcp = FastMCP("CodeQL-Python")
3334

3435
# tool name -> templated query lookup for supported languages
3536
TEMPLATED_QUERY_PATHS = {

0 commit comments

Comments
 (0)