File tree Expand file tree Collapse file tree 1 file changed +7
-7
lines changed
src/aks-agent/azext_aks_agent/agent Expand file tree Collapse file tree 1 file changed +7
-7
lines changed Original file line number Diff line number Diff line change @@ -128,17 +128,17 @@ def get_specific(
128128 """
129129 model_configs = self .get_list ()
130130 for cfg in model_configs :
131- if cfg .get ("provider" ) == provider_name :
132- if provider_name .lower () == "azure" and (cfg .get ("DEPLOYMENT_NAME" ) == model_name or cfg .get ("MODEL_NAME" ) == model_name ):
133- return cfg
134- elif cfg .get ("MODEL_NAME" ) == model_name :
131+ if cfg .get ("provider" ) == provider_name and provider_name .lower () == "azure" :
132+ if cfg .get ("DEPLOYMENT_NAME" ) == model_name or cfg .get ("MODEL_NAME" ) == model_name :
135133 return cfg
134+ if cfg .get ("provider" ) == provider_name and cfg .get ("MODEL_NAME" ) == model_name :
135+ return cfg
136136 return None
137137
138138 def get_model_config (self , model ) -> Optional [Dict ]:
139- prompt_for_init = "Run 'az aks agent-init' to set up your LLM endpoint (recommended path).\n "
140- "To configure your LLM manually, create a config file using the templates provided here: "
141- "https://aka.ms/aks/agentic-cli/init"
139+ prompt_for_init = "Run 'az aks agent-init' to set up your LLM endpoint (recommended path).\n " \
140+ "To configure your LLM manually, create a config file using the templates provided here: " \
141+ "https://aka.ms/aks/agentic-cli/init"
142142
143143 if not model :
144144 llm_config : Optional [Dict ] = self .get_latest ()
You can’t perform that action at this time.
0 commit comments