def embeddings_command()

in src/doc_builder/commands/embeddings.py [0:0]

• 38 lines of code
• 8 McCabe index (conditional complexity)

def embeddings_command(args):
    read_doc_config(args.path_to_docs)

    default_version = get_default_branch_name(args.path_to_docs)
    if args.not_python_module and args.version is None:
        version = default_version
    elif args.version is None:
        module = importlib.import_module(args.library_name)
        version = module.__version__

        if "dev" in version:
            version = default_version
        else:
            version = f"v{version}"
    else:
        version = args.version

    # `version` will always start with prefix `v`
    # `version_tag` does not have to start with prefix `v` (see: https://github.com/huggingface/datasets/tags)
    version_tag = version
    if version != default_version:
        doc_config = get_doc_config()
        version_prefix = getattr(doc_config, "version_prefix", "v")
        version_ = version[1:]  # v2.1.0 -> 2.1.0
        version_tag = f"{version_prefix}{version_}"

    # Disable notebook building for non-master version
    if version != default_version:
        args.notebook_dir = None

    print("Building embeddings for", args.library_name, args.path_to_docs)
    build_embeddings(
        args.library_name,
        args.path_to_docs,
        args.hf_ie_name,
        args.hf_ie_namespace,
        args.hf_ie_token,
        args.meilisearch_key,
        version=version,
        version_tag=version_tag,
        language=args.language,
        is_python_module=not args.not_python_module,
        version_tag_suffix=args.version_tag_suffix,
        repo_owner=args.repo_owner,
        repo_name=args.repo_name,
    )