Skip to content

Commit 409925c

Browse files
etsienetsien
committed
apply agent name variable across YAML files with yaml anchoring
1 parent 9ef6fe0 commit 409925c

File tree

6 files changed

+36
-12
lines changed

6 files changed

+36
-12
lines changed

‎kustomize/base/exploit-iq-config.yml‎

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25,9 +25,12 @@ general:
2525

2626

2727
functions:
28+
# Agent name anchor for consistency across all function configurations
29+
_agent_executor_name: &agent_executor_name cve_agent_executor
30+
2831
cve_generate_vdbs:
2932
_type: cve_generate_vdbs
30-
agent_name: cve_agent_executor # Used to determine which tools are enabled
33+
agent_name: *agent_executor_name
3134
embedder_name: nim_embedder
3235
base_git_dir: ${EXPLOIT_IQ_DATA_DIR:-/exploit-iq-data/}git
3336
base_vdb_dir: ${EXPLOIT_IQ_DATA_DIR:-/exploit-iq-data/}vdb
@@ -54,6 +57,7 @@ functions:
5457
cve_checklist:
5558
_type: cve_checklist
5659
llm_name: checklist_llm
60+
agent_name: *agent_executor_name
5761
Call Chain Analyzer:
5862
_type: call_chain_analyzer
5963
enable_call_chain_analyzer: true
@@ -83,7 +87,7 @@ functions:
8387
Container Analysis Data:
8488
_type: container_image_analysis_data
8589
cve_agent_executor:
86-
_type: cve_agent_executor
90+
_type: *agent_executor_name
8791
llm_name: cve_agent_executor_llm
8892
tool_names:
8993
- Code Semantic Search

‎kustomize/config-http-openai-local.yml‎

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,9 +29,12 @@ general:
2929
# api_key: ${LANGSMITH_API_KEY}
3030

3131
functions:
32+
# Agent name anchor for consistency across all function configurations
33+
_agent_executor_name: &agent_executor_name cve_agent_executor
34+
3235
cve_generate_vdbs:
3336
_type: cve_generate_vdbs
34-
agent_name: cve_agent_executor # Used to determine which tools are enabled
37+
agent_name: *agent_executor_name
3538
embedder_name: nim_embedder
3639
base_git_dir: .cache/am_cache/git
3740
base_vdb_dir: .cache/am_cache/vdb
@@ -55,6 +58,7 @@ functions:
5558
cve_checklist:
5659
_type: cve_checklist
5760
llm_name: checklist_llm
61+
agent_name: *agent_executor_name
5862
Call Chain Analyzer:
5963
_type: call_chain_analyzer
6064
enable_call_chain_analyzer: true
@@ -84,7 +88,7 @@ functions:
8488
Container Analysis Data:
8589
_type: container_image_analysis_data
8690
cve_agent_executor:
87-
_type: cve_agent_executor
91+
_type: *agent_executor_name
8892
llm_name: cve_agent_executor_llm
8993
tool_names:
9094
- Code Semantic Search

‎src/vuln_analysis/configs/config-http-nim.yml‎

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,9 +23,12 @@ general:
2323
project: cve_agent
2424

2525
functions:
26+
# Agent name anchor for consistency across all function configurations
27+
_agent_executor_name: &agent_executor_name cve_agent_executor
28+
2629
cve_generate_vdbs:
2730
_type: cve_generate_vdbs
28-
agent_name: cve_agent_executor # Used to determine which tools are enabled
31+
agent_name: *agent_executor_name
2932
embedder_name: nim_embedder
3033
base_git_dir: .cache/am_cache/git
3134
base_vdb_dir: .cache/am_cache/vdb
@@ -48,6 +51,7 @@ functions:
4851
cve_checklist:
4952
_type: cve_checklist
5053
llm_name: checklist_llm
54+
agent_name: *agent_executor_name
5155
Call Chain Analyzer:
5256
_type: call_chain_analyzer
5357
enable_call_chain_analyzer: true
@@ -77,7 +81,7 @@ functions:
7781
Container Analysis Data:
7882
_type: container_image_analysis_data
7983
cve_agent_executor:
80-
_type: cve_agent_executor
84+
_type: *agent_executor_name
8185
llm_name: cve_agent_executor_llm
8286
tool_names:
8387
- Code Semantic Search

‎src/vuln_analysis/configs/config-http-openai.yml‎

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,9 +23,12 @@ general:
2323
project: cve_agent
2424

2525
functions:
26+
# Agent name anchor for consistency across all function configurations
27+
_agent_executor_name: &agent_executor_name cve_agent_executor
28+
2629
cve_generate_vdbs:
2730
_type: cve_generate_vdbs
28-
agent_name: cve_agent_executor # Used to determine which tools are enabled
31+
agent_name: *agent_executor_name
2932
embedder_name: nim_embedder
3033
base_git_dir: .cache/am_cache/git
3134
base_vdb_dir: .cache/am_cache/vdb
@@ -48,6 +51,7 @@ functions:
4851
cve_checklist:
4952
_type: cve_checklist
5053
llm_name: checklist_llm
54+
agent_name: *agent_executor_name
5155
Call Chain Analyzer:
5256
_type: call_chain_analyzer
5357
enable_call_chain_analyzer: true
@@ -77,7 +81,7 @@ functions:
7781
Container Analysis Data:
7882
_type: container_image_analysis_data
7983
cve_agent_executor:
80-
_type: cve_agent_executor
84+
_type: *agent_executor_name
8185
llm_name: cve_agent_executor_llm
8286
tool_names:
8387
- Code Semantic Search

‎src/vuln_analysis/configs/config-tracing.yml‎

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,9 +27,12 @@ general:
2727
project: cve_agent
2828

2929
functions:
30+
# Agent name anchor for consistency across all function configurations
31+
_agent_executor_name: &agent_executor_name cve_agent_executor
32+
3033
cve_generate_vdbs:
3134
_type: cve_generate_vdbs
32-
agent_name: cve_agent_executor # Used to determine which tools are enabled
35+
agent_name: *agent_executor_name
3336
embedder_name: nim_embedder
3437
base_git_dir: .cache/am_cache/git
3538
base_vdb_dir: .cache/am_cache/vdb
@@ -52,6 +55,7 @@ functions:
5255
cve_checklist:
5356
_type: cve_checklist
5457
llm_name: checklist_llm
58+
agent_name: *agent_executor_name
5559
Call Chain Analyzer:
5660
_type: call_chain_analyzer
5761
enable_call_chain_analyzer: true
@@ -81,7 +85,7 @@ functions:
8185
Container Analysis Data:
8286
_type: container_image_analysis_data
8387
cve_agent_executor:
84-
_type: cve_agent_executor
88+
_type: *agent_executor_name
8589
llm_name: cve_agent_executor_llm
8690
tool_names:
8791
- Code Semantic Search

‎src/vuln_analysis/configs/config.yml‎

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,9 +23,12 @@ general:
2323
project: cve_agent
2424

2525
functions:
26+
# Agent name anchor for consistency across all function configurations
27+
_agent_executor_name: &agent_executor_name cve_agent_executor
28+
2629
cve_generate_vdbs:
2730
_type: cve_generate_vdbs
28-
agent_name: cve_agent_executor # Used to determine which tools are enabled
31+
agent_name: *agent_executor_name
2932
embedder_name: nim_embedder
3033
base_git_dir: .cache/am_cache/git
3134
base_vdb_dir: .cache/am_cache/vdb
@@ -40,6 +43,7 @@ functions:
4043
cve_checklist:
4144
_type: cve_checklist
4245
llm_name: checklist_llm
46+
agent_name: *agent_executor_name
4347
Code Semantic Search:
4448
_type: local_vdb_retriever
4549
embedder_name: nim_embedder
@@ -61,7 +65,7 @@ functions:
6165
Container Analysis Data:
6266
_type: container_image_analysis_data
6367
cve_agent_executor:
64-
_type: cve_agent_executor
68+
_type: *agent_executor_name
6569
llm_name: cve_agent_executor_llm
6670
tool_names:
6771
- Code Semantic Search

0 commit comments

Comments
 (0)