Skip to content

Commit 3faf2e1

Browse files
committed
Fix Mypy
1 parent 4c57009 commit 3faf2e1

File tree

1 file changed

+7
-6
lines changed

1 file changed

+7
-6
lines changed

vocode/streaming/agent/chat_gpt_agent.py

+7-6
Original file line numberDiff line numberDiff line change
@@ -126,12 +126,13 @@ def apply_model_fallback(self, chat_parameters: Dict[str, Any]):
126126
if isinstance(self.openai_client, AsyncAzureOpenAI):
127127
self.agent_config.azure_params = None
128128
else:
129-
self.agent_config.azure_params.deployment_name = (
130-
self.agent_config.llm_fallback.model_name
131-
)
132-
if isinstance(self.openai_client, AsyncOpenAI):
133-
# TODO: handle OpenAI fallback to Azure
134-
pass
129+
if self.agent_config.azure_params:
130+
self.agent_config.azure_params.deployment_name = (
131+
self.agent_config.llm_fallback.model_name
132+
)
133+
if isinstance(self.openai_client, AsyncOpenAI):
134+
# TODO: handle OpenAI fallback to Azure
135+
pass
135136

136137
self.openai_client = instantiate_openai_client(self.agent_config, model_fallback=False)
137138
chat_parameters["model"] = self.agent_config.llm_fallback.model_name

0 commit comments

Comments
 (0)