Langsmith tags- Programmatic Solution

[dansashalesser]

Hi everyone!
I'm trying to write a function that does the following:

1. Pulls a prompt based on its 'dev' tag
2. Checks if the commit_id is similar to a commit with the 'live' tag- If so, exit function.
3. If not, remove 'live' tag from the other commit, and assign the 'live' tag to the commit, so it will have both tags.

The first bits are quite easy, but I couldn't for the life of me find a way to update commit tags other than the Langsmith UI.
I tried update_prompt, push_prompt, etc.

I'm attaching the function I wrote- If anyone has any thoughts, it would be highly appreciated!

def promote_dev_to_live(prompt_id: str) -> None:
    """
    Promote the latest 'dev' commit of a prompt to 'live' by creating a new commit
    that has both the 'dev' and 'live' tags. If a live commit already exists,
    its 'live' tag is removed before the new commit is pushed.

    Args:
        prompt_id (str): The unique identifier of the prompt.

    Raises:
        ValueError: If no 'dev' commit is found or its commit hash is missing.
    """
    # Pull the dev commit prompt
    dev_prompt = client.pull_prompt(f"{prompt_id}:dev")
    if not dev_prompt:
        raise ValueError(f"No 'dev' commit found for prompt '{prompt_id}'.")
    dev_commit_hash = dev_prompt.metadata.get("lc_hub_commit_hash")
    if not dev_commit_hash:
        raise ValueError(
            f"Commit hash missing in 'dev' commit for prompt '{prompt_id}'."
        )

    # Pull the current live commit prompt (if any) and remove its live tag
    live_prompt = client.pull_prompt(f"{prompt_id}:live")
    if live_prompt:
        live_commit_hash = live_prompt.metadata.get("lc_hub_commit_hash")
        if live_commit_hash:
            if dev_commit_hash == live_commit_hash:
                print(
                    f"The 'dev' commit for prompt '{prompt_id}' is already tagged as 'live'."
                )
                return
            # Remove live tag from the current live commit
            client.update_prompt(f"{prompt_id}:{live_commit_hash}", tags=[])

    # Push a new commit using the dev prompt content and assign both 'dev' and 'live' tags
    new_commit_url = client.push_prompt(
        prompt_id, object=dev_prompt, tags=["dev", "live"]
    )
    print(
        f"Promoted dev commit {dev_commit_hash} to live. New commit URL: {new_commit_url}"
    )

[rschoenbeck]

Nice workaround; would be awesome to have a way to manage commit tags like this programmatically within the client itself. For me, the tag updates in update_prompt update the prompt tags (which are metadata at the prompt level), not the individual prompt commit tags.