fix: default max_chunks set to 1 as other providers (#10937)

Signed-off-by: yihong0618 <zouzou0208@gmail.com>
This commit is contained in:
yihong 2024-11-21 16:36:05 +08:00 committed by GitHub
parent 83b6abf4ad
commit 80da0c5830
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -22,7 +22,7 @@ def get_model_config(credentials: dict) -> ModelConfig:
return ModelConfig( return ModelConfig(
properties=ModelProperties( properties=ModelProperties(
context_size=int(credentials.get("context_size", 0)), context_size=int(credentials.get("context_size", 0)),
max_chunks=int(credentials.get("max_chunks", 0)), max_chunks=int(credentials.get("max_chunks", 1)),
) )
) )
return model_configs return model_configs