enterprise-kb-rag/2.2_LangChain基础知识/04_models_embeddings.py

from pathlib import Path
import sys

sys.path.append(str(Path(__file__).resolve().parents[1]))

from _chapter2_config import get_embeddings, print_dependency_hint


def main() -> None:
    try:
        model = get_embeddings()
    except ModuleNotFoundError as exc:
        print_dependency_hint(exc)
        raise SystemExit(1)

    res1 = model.embed_query("这是第一个测试文档")
    print(f"result1->{res1}")
    print(f"result1的长度->{len(res1)}")

    res2 = model.embed_documents(["这是第一个测试文档", "这是第二个测试文档"])
    print(res2)


if __name__ == "__main__":
    main()