typing

TLSDC · TLSDC · commit e0e786cfbc95 · 2025-01-24T14:41:19.000-05:00
diff --git a/src/agentlab/llm/llm_configs.py b/src/agentlab/llm/llm_configs.py
@@ -16,7 +16,7 @@
     "test",
 ]
 
-CHAT_MODEL_ARGS_DICT = {
+CHAT_MODEL_ARGS_DICT = {  # type: dict[str, Union[AzureModelArgs, OpenAIModelArgs, SelfHostedModelArgs, OpenRouterModelArgs]]
     "openai/gpt-4o-mini-2024-07-18": OpenAIModelArgs(
         model_name="gpt-4o-mini-2024-07-18",
         max_total_tokens=128_000,

Original file line number	Diff line number	Diff line change
`@@ -16,7 +16,7 @@`
`16`	`16`	`"test",`
`17`	`17`	`]`
`18`	`18`
`19`		`-CHAT_MODEL_ARGS_DICT = {`
	`19`	`+CHAT_MODEL_ARGS_DICT = { # type: dict[str, Union[AzureModelArgs, OpenAIModelArgs, SelfHostedModelArgs, OpenRouterModelArgs]]`
`20`	`20`	`"openai/gpt-4o-mini-2024-07-18": OpenAIModelArgs(`
`21`	`21`	`model_name="gpt-4o-mini-2024-07-18",`
`22`	`22`	`max_total_tokens=128_000,`