provider type default to none (#2110)

This commit is contained in:
pablodanswer 2024-08-11 14:51:12 -07:00 committed by GitHub
parent 386b229ed3
commit 7fae66b766
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -297,7 +297,9 @@ async def lifespan(app: FastAPI) -> AsyncGenerator:
update_search_settings(
SavedSearchSettings(
rerank_model_name=DEFAULT_CROSS_ENCODER_MODEL_NAME,
provider_type=RerankerProvider(DEFAULT_CROSS_ENCODER_PROVIDER_TYPE),
provider_type=RerankerProvider(DEFAULT_CROSS_ENCODER_PROVIDER_TYPE)
if DEFAULT_CROSS_ENCODER_PROVIDER_TYPE is not None
else None,
api_key=DEFAULT_CROSS_ENCODER_API_KEY,
disable_rerank_for_streaming=DISABLE_RERANK_FOR_STREAMING,
num_rerank=NUM_POSTPROCESSED_RESULTS,