Fix Dockerfile build issue
This commit is contained in:
24
autorag/schema/module.py
Normal file
24
autorag/schema/module.py
Normal file
@@ -0,0 +1,24 @@
|
||||
from copy import deepcopy
|
||||
from dataclasses import dataclass, field
|
||||
from typing import Callable, Dict
|
||||
|
||||
from autorag.support import get_support_modules
|
||||
|
||||
|
||||
@dataclass
|
||||
class Module:
|
||||
module_type: str
|
||||
module_param: Dict
|
||||
module: Callable = field(init=False)
|
||||
|
||||
def __post_init__(self):
|
||||
self.module = get_support_modules(self.module_type)
|
||||
if self.module is None:
|
||||
raise ValueError(f"Module type {self.module_type} is not supported.")
|
||||
|
||||
@classmethod
|
||||
def from_dict(cls, module_dict: Dict) -> "Module":
|
||||
_module_dict = deepcopy(module_dict)
|
||||
module_type = _module_dict.pop("module_type")
|
||||
module_params = _module_dict
|
||||
return cls(module_type, module_params)
|
||||
Reference in New Issue
Block a user