refactor: Refactors model configuration for consistency

Signed-off-by: -LAN- <laipz8200@outlook.com>
pull/20455/head
-LAN- 12 months ago
parent 1cb1ac210d
commit fba4ac314a
No known key found for this signature in database
GPG Key ID: 6BA0D108DED011FF

@ -76,6 +76,8 @@ class AppGenerateEntity(BaseModel):
App Generate Entity. App Generate Entity.
""" """
model_config = ConfigDict(arbitrary_types_allowed=True)
task_id: str task_id: str
# app config # app config
@ -99,9 +101,6 @@ class AppGenerateEntity(BaseModel):
# tracing instance # tracing instance
trace_manager: Optional[TraceQueueManager] = None trace_manager: Optional[TraceQueueManager] = None
class Config:
arbitrary_types_allowed = True
class EasyUIBasedAppGenerateEntity(AppGenerateEntity): class EasyUIBasedAppGenerateEntity(AppGenerateEntity):
""" """

@ -27,6 +27,8 @@ class WebsiteInfo(BaseModel):
website import info. website import info.
""" """
model_config = ConfigDict(arbitrary_types_allowed=True)
provider: str provider: str
job_id: str job_id: str
url: str url: str
@ -34,12 +36,6 @@ class WebsiteInfo(BaseModel):
tenant_id: str tenant_id: str
only_main_content: bool = False only_main_content: bool = False
class Config:
arbitrary_types_allowed = True
def __init__(self, **data) -> None:
super().__init__(**data)
class ExtractSetting(BaseModel): class ExtractSetting(BaseModel):
""" """

@ -45,13 +45,12 @@ class BaseDocumentTransformer(ABC):
.. code-block:: python .. code-block:: python
class EmbeddingsRedundantFilter(BaseDocumentTransformer, BaseModel): class EmbeddingsRedundantFilter(BaseDocumentTransformer, BaseModel):
model_config = ConfigDict(arbitrary_types_allowed=True)
embeddings: Embeddings embeddings: Embeddings
similarity_fn: Callable = cosine_similarity similarity_fn: Callable = cosine_similarity
similarity_threshold: float = 0.95 similarity_threshold: float = 0.95
class Config:
arbitrary_types_allowed = True
def transform_documents( def transform_documents(
self, documents: Sequence[Document], **kwargs: Any self, documents: Sequence[Document], **kwargs: Any
) -> Sequence[Document]: ) -> Sequence[Document]:

Loading…
Cancel
Save