diff options
Diffstat (limited to 'src/client/ollama.rs')
| -rw-r--r-- | src/client/ollama.rs | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/client/ollama.rs b/src/client/ollama.rs index e24cffd..5ecf6c8 100644 --- a/src/client/ollama.rs +++ b/src/client/ollama.rs @@ -52,8 +52,8 @@ impl OllamaClient { ("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, ), @@ -68,7 +68,7 @@ impl OllamaClient { .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(TOKENS_COUNT_FACTORS) }) |