Viewing File: /home/ubuntu/codegamaai-test/rag_drive/regdrive/lib/python3.10/site-packages/langsmith/env/_git.py

"""Fetch information about any current git repo."""

import functools
import logging
import subprocess
from typing import List, Optional, TypeVar

from typing_extensions import TypedDict

logger = logging.getLogger(__name__)

T = TypeVar("T")


def _exec_git(command: List[str]) -> Optional[str]:
    try:
        return subprocess.check_output(
            ["git"] + command, encoding="utf-8", stderr=subprocess.DEVNULL
        ).strip()
    except FileNotFoundError:
        logger.warning("git is not installed, or cannot be found in PATH")
        return None
    except subprocess.CalledProcessError as e:
        logger.debug(f"Error running git command: {e}")
        return None


class GitInfo(TypedDict, total=False):
    remote_url: Optional[str]
    commit: Optional[str]
    branch: Optional[str]
    author_name: Optional[str]
    author_email: Optional[str]
    commit_message: Optional[str]
    commit_time: Optional[str]
    dirty: Optional[bool]
    tags: Optional[str]


@functools.lru_cache(maxsize=1)
def get_git_info(remote: str = "origin") -> Optional[GitInfo]:
    """Get information about the git repository."""

    if not _exec_git(["rev-parse", "--is-inside-work-tree"]):
        return None

    return {
        "remote_url": _exec_git(["remote", "get-url", remote]),
        "commit": _exec_git(["rev-parse", "HEAD"]),
        "commit_time": _exec_git(["log", "-1", "--format=%ct"]),
        "branch": _exec_git(["rev-parse", "--abbrev-ref", "HEAD"]),
        "tags": _exec_git(
            ["describe", "--tags", "--exact-match", "--always", "--dirty"]
        ),
        "dirty": _exec_git(["status", "--porcelain"]) != "",
        "author_name": _exec_git(["log", "-1", "--format=%an"]),
        "author_email": _exec_git(["log", "-1", "--format=%ae"]),
    }