amirkiarafiei commited on
Commit
ef321a7
·
1 Parent(s): 0ff2994

docs: add comments for hf_token usage in bug localization functions

Browse files
Files changed (1) hide show
  1. main.py +6 -0
main.py CHANGED
@@ -61,6 +61,8 @@ async def multi_module_bug_localization(request: MultiModuleRequest, ctx: Contex
61
  client = ds_ctx.client
62
 
63
  api_url = API_CONFIG["api_base_url"] + API_CONFIG["api_multimodule_endpoint"]
 
 
64
  hf_token = API_CONFIG.get("hf_access_token", None)
65
 
66
  # Get per-user API key from MCP Context
@@ -138,6 +140,8 @@ async def single_module_bug_localization(request: SingleModuleRequest, ctx: Cont
138
  client = ds_ctx.client
139
 
140
  api_url = API_CONFIG["api_base_url"] + API_CONFIG["api_singlemodule_endpoint"]
 
 
141
  hf_token = API_CONFIG.get("hf_access_token", None)
142
 
143
  # Get per-user API key from MCP Context
@@ -214,6 +218,8 @@ async def search_space_routing(request: SearchSpaceRoutingRequest, ctx: Context)
214
  client = ds_ctx.client
215
 
216
  api_url = API_CONFIG["api_base_url"] + API_CONFIG["api_searchspace_endpoint"]
 
 
217
  hf_token = API_CONFIG.get("hf_access_token", None)
218
 
219
  # Get per-user API key from MCP Context
 
61
  client = ds_ctx.client
62
 
63
  api_url = API_CONFIG["api_base_url"] + API_CONFIG["api_multimodule_endpoint"]
64
+
65
+ # hf_token to access private HF Space serving Defect Solver API
66
  hf_token = API_CONFIG.get("hf_access_token", None)
67
 
68
  # Get per-user API key from MCP Context
 
140
  client = ds_ctx.client
141
 
142
  api_url = API_CONFIG["api_base_url"] + API_CONFIG["api_singlemodule_endpoint"]
143
+
144
+ # hf token to access Defect Solver API
145
  hf_token = API_CONFIG.get("hf_access_token", None)
146
 
147
  # Get per-user API key from MCP Context
 
218
  client = ds_ctx.client
219
 
220
  api_url = API_CONFIG["api_base_url"] + API_CONFIG["api_searchspace_endpoint"]
221
+
222
+ # hf_token to access private HF Space serving Defect Solver API
223
  hf_token = API_CONFIG.get("hf_access_token", None)
224
 
225
  # Get per-user API key from MCP Context