@@ -21,10 +21,10 @@ def _get_llm_config_from_context(context: AnalysisContext, role) -> Any:
2121 """从上下文中获取LLM配置,支持配置中的模型、API Key和Base URL"""
2222 config = getattr (context , '_config' , None )
2323 if not config :
24- from config import get_resilient_llm_config , LLMRole
24+ from pure_auto_codeql . configuration import get_resilient_llm_config , LLMRole
2525 return get_resilient_llm_config (role )
2626
27- from config import get_llm_config , LLMRole
27+ from pure_auto_codeql . configuration import get_llm_config , LLMRole
2828 # 从配置中获取参数
2929 provider = config .llm_provider
3030 model_name = config .think_model if role == LLMRole .THINK else config .chat_model
@@ -73,7 +73,7 @@ def __init__(self):
7373
7474 async def execute (self , context : AnalysisContext ) -> Any :
7575 """执行CVE分析。"""
76- from config import LLMRole
76+ from pure_auto_codeql . configuration import LLMRole
7777 llm_config = _get_llm_config_from_context (context , LLMRole .CHAT )
7878 analyzer = MultiAgentAnalyzer (llm_config )
7979 await analyzer .initialize (
@@ -116,7 +116,7 @@ def __init__(self):
116116
117117 async def execute (self , context : AnalysisContext ) -> Any :
118118 """执行Sink路径分析。"""
119- from config import LLMRole
119+ from pure_auto_codeql . configuration import LLMRole
120120 llm_config = _get_llm_config_from_context (context , LLMRole .CHAT )
121121 analyzer = MultiAgentAnalyzer (llm_config )
122122 await analyzer .initialize (
@@ -197,7 +197,7 @@ def __init__(self):
197197
198198 async def execute (self , context : AnalysisContext ) -> Any :
199199 """执行Source分析。"""
200- from config import LLMRole
200+ from pure_auto_codeql . configuration import LLMRole
201201 llm_config = _get_llm_config_from_context (context , LLMRole .CHAT )
202202 analyzer = MultiAgentAnalyzer (llm_config )
203203 await analyzer .initialize (
@@ -281,7 +281,7 @@ def __init__(self):
281281
282282 async def execute (self , context : AnalysisContext ) -> Any :
283283 """执行Path分析。"""
284- from config import LLMRole
284+ from pure_auto_codeql . configuration import LLMRole
285285
286286 # 检查是否有 Source 分析结果
287287 if not context .has_result ("source_analysis" ):
@@ -405,7 +405,7 @@ async def execute(self, context: AnalysisContext) -> Any:
405405 logger .warning ("Path 分析结果 JSON 解析失败" )
406406
407407 # 使用推理模型
408- from config import LLMRole
408+ from pure_auto_codeql . configuration import LLMRole
409409 think_config = _get_llm_config_from_context (context , LLMRole .THINK )
410410 codeql_analyzer = MultiAgentAnalyzer (think_config )
411411 await codeql_analyzer .initialize (
@@ -758,7 +758,7 @@ async def _run_path_selection(
758758 return None
759759
760760 try :
761- from config import LLMRole
761+ from pure_auto_codeql . configuration import LLMRole
762762
763763 llm_config = _get_llm_config_from_context (context , LLMRole .CHAT )
764764 service = PathSelectionService (llm_config , language = context .language )
0 commit comments