diff options
Diffstat (limited to 'src/client/azure_openai.rs')
| -rw-r--r-- | src/client/azure_openai.rs | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/client/azure_openai.rs b/src/client/azure_openai.rs index 5c9f3ef..e553f3f 100644 --- a/src/client/azure_openai.rs +++ b/src/client/azure_openai.rs @@ -28,8 +28,8 @@ impl AzureOpenAIClient { ("api_key", "API Key:", true, PromptKind::String), ("models[].name", "Model Name:", true, PromptKind::String), ( - "models[].max_tokens", - "Max Tokens:", + "models[].max_input_tokens", + "Max Input Tokens:", true, PromptKind::Integer, ), @@ -43,7 +43,7 @@ impl AzureOpenAIClient { .iter() .map(|v| { Model::new(client_name, &v.name) - .set_max_tokens(v.max_tokens) + .set_max_input_tokens(v.max_input_tokens) .set_capabilities(v.capabilities) .set_tokens_count_factors(OPENAI_TOKENS_COUNT_FACTORS) }) |