From 31b5fde16368165c79dec6e71d14579d4627e4b5 Mon Sep 17 00:00:00 2001
From: Debanjum <debanjum@gmail.com>
Date: Fri, 1 Nov 2024 23:57:22 -0700
Subject: [PATCH] Only enable prompt tracer if git python is installed

---
 pyproject.toml                           |  2 +-
 src/khoj/processor/conversation/utils.py | 17 ++++++++++++++++-
 2 files changed, 17 insertions(+), 2 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 3f31d5ed..f8f64c38 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -88,7 +88,6 @@ dependencies = [
     "anthropic == 0.26.1",
     "docx2txt == 0.8",
     "google-generativeai == 0.8.3",
-    "gitpython ~= 3.1.43",
 ]
 dynamic = ["version"]
 
@@ -120,6 +119,7 @@ dev = [
     "mypy >= 1.0.1",
     "black >= 23.1.0",
     "pre-commit >= 3.0.4",
+    "gitpython ~= 3.1.43",
 ]
 
 [tool.hatch.version]
diff --git a/src/khoj/processor/conversation/utils.py b/src/khoj/processor/conversation/utils.py
index aae55af5..74c464d9 100644
--- a/src/khoj/processor/conversation/utils.py
+++ b/src/khoj/processor/conversation/utils.py
@@ -17,7 +17,6 @@ import PIL.Image
 import requests
 import tiktoken
 import yaml
-from git import Repo
 from langchain.schema import ChatMessage
 from llama_cpp.llama import Llama
 from transformers import AutoTokenizer
@@ -39,6 +38,13 @@ from khoj.utils.helpers import (
 )
 
 logger = logging.getLogger(__name__)
+
+try:
+    from git import Repo
+except ImportError:
+    if in_debug_mode():
+        logger.warning("GitPython not installed. `pip install gitpython` to enable prompt tracer.")
+
 model_to_prompt_size = {
     # OpenAI Models
     "gpt-3.5-turbo": 12000,
@@ -510,6 +516,11 @@ def commit_conversation_trace(
     Save trace of conversation step using git. Useful to visualize, compare and debug traces.
     Returns the path to the repository.
     """
+    try:
+        from git import Repo
+    except ImportError:
+        return None
+
     # Serialize session, system message and response to yaml
     system_message_yaml = json.dumps(system_message, ensure_ascii=False, sort_keys=False)
     response_yaml = json.dumps(response, ensure_ascii=False, sort_keys=False)
@@ -617,6 +628,10 @@ def merge_message_into_conversation_trace(query: str, response: str, tracer: dic
     Returns:
         bool: True if merge was successful, False otherwise
     """
+    try:
+        from git import Repo
+    except ImportError:
+        return False
     try:
         # Extract branch names
         msg_branch = f"m_{tracer['mid']}"