36 lines
867 B
Python
36 lines
867 B
Python
"""向量生成模块 (M5)。
|
|
|
|
公共 API:
|
|
- load_embedding_config / make_embedding_client
|
|
- compose_text / embed_article / embed_articles
|
|
- embed_all_events（批量管道）
|
|
- EmbeddingResult / EmbeddingConfig / EmbeddingError
|
|
"""
|
|
|
|
from embedding.client import (
|
|
EmbeddingConfig,
|
|
embed_batch,
|
|
load_embedding_config,
|
|
make_embedding_client,
|
|
)
|
|
from embedding.embedder import compose_text, embed_article, embed_articles
|
|
from embedding.models import EmbeddingError, EmbeddingResult
|
|
from embedding.pipeline import embed_all_events
|
|
|
|
__all__ = [
|
|
# 客户端
|
|
"EmbeddingConfig",
|
|
"embed_batch",
|
|
"load_embedding_config",
|
|
"make_embedding_client",
|
|
# 嵌入
|
|
"compose_text",
|
|
"embed_article",
|
|
"embed_articles",
|
|
# 批量管道
|
|
"embed_all_events",
|
|
# 模型
|
|
"EmbeddingError",
|
|
"EmbeddingResult",
|
|
]
|