refactor: move the embedding to the rag module and abstract the rerank runner for extension (#9423)

2024-10-17 19:12:42 +08:00
parent e7aecb89dd
commit b90ad587c2
61 changed files with 135 additions and 78 deletions
--- a/api/core/model_manager.py
+++ b/api/core/model_manager.py
@@ -3,7 +3,7 @@ import os
 from collections.abc import Callable, Generator, Sequence
 from typing import IO, Optional, Union, cast

-from core.embedding.embedding_constant import EmbeddingInputType
+from core.entities.embedding_type import EmbeddingInputType
 from core.entities.provider_configuration import ProviderConfiguration, ProviderModelBundle
 from core.entities.provider_entities import ModelLoadBalancingConfiguration
 from core.errors.error import ProviderTokenNotInitError