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