| 220 | # Utils types ----------------------------------------------------------------------- |
| 221 | @dataclass |
| 222 | class EmbeddingFunc: |
| 223 | embedding_dim: int |
| 224 | max_token_size: int |
| 225 | func: callable |
| 226 | |
| 227 | async def __call__(self, *args, **kwargs) -> np.ndarray: |
| 228 | return await self.func(*args, **kwargs) |
| 229 | |
| 230 | |
| 231 | # Decorators ------------------------------------------------------------------------ |