holmesgpt 0.11.5__py3-none-any.whl

holmes/.git_archival.json

@@ -0,0 +1,7 @@
+{
+  "hash-full": "$Format:%H$",
+  "hash-short": "$Format:%h$",
+  "timestamp": "$Format:%cI$",
+  "refs": "$Format:%D$",
+  "describe": "$Format:%(describe:tags=true,match=v[0-9]*)$"
+}

holmes/__init__.py

@@ -0,0 +1,76 @@
+import json
+import os
+import subprocess
+import sys
+from cachetools import cached  # type: ignore
+
+# For relative imports to work in Python 3.6 - see https://stackoverflow.com/a/49375740
+this_path = os.path.dirname(os.path.realpath(__file__))
+sys.path.append(this_path)
+
+# This is patched by github actions during release
+__version__ = "0.11.5"
+
+
+def is_official_release() -> bool:
+    return not __version__.startswith("0.0.0")
+
+
+@cached(cache=dict())
+def get_version() -> str:
+    # the version string was patched by a release - return __version__ which will be correct
+    if is_official_release():
+        return __version__
+
+    # we are running from an unreleased dev version
+    try:
+        # Get the latest git tag
+        tag = (
+            subprocess.check_output(
+                ["git", "describe", "--tags"], stderr=subprocess.STDOUT, cwd=this_path
+            )
+            .decode()
+            .strip()
+        )
+
+        # Get the current branch name
+        branch = (
+            subprocess.check_output(
+                ["git", "rev-parse", "--abbrev-ref", "HEAD"],
+                stderr=subprocess.STDOUT,
+                cwd=this_path,
+            )
+            .decode()
+            .strip()
+        )
+
+        # Check if there are uncommitted changes
+        status = (
+            subprocess.check_output(
+                ["git", "status", "--porcelain"],
+                stderr=subprocess.STDOUT,
+                cwd=this_path,
+            )
+            .decode()
+            .strip()
+        )
+        dirty = "-dirty" if status else ""
+
+        return f"{tag}-{branch}{dirty}"
+
+    except Exception:
+        pass
+
+    # we are running without git history, but we still might have git archival data (e.g. if we were pip installed)
+    archival_file_path = os.path.join(this_path, ".git_archival.json")
+    if os.path.exists(archival_file_path):
+        try:
+            with open(archival_file_path, "r") as f:
+                archival_data = json.load(f)
+                return f"dev-{archival_data['refs']}-{archival_data['hash-short']}"
+        except Exception:
+            pass
+
+        return "dev-version"
+
+    return "unknown-version"

holmes/__init__.py.bak

@@ -0,0 +1,76 @@
+import json
+import os
+import subprocess
+import sys
+from cachetools import cached  # type: ignore
+
+# For relative imports to work in Python 3.6 - see https://stackoverflow.com/a/49375740
+this_path = os.path.dirname(os.path.realpath(__file__))
+sys.path.append(this_path)
+
+# This is patched by github actions during release
+__version__ = "0.0.0"
+
+
+def is_official_release() -> bool:
+    return not __version__.startswith("0.0.0")
+
+
+@cached(cache=dict())
+def get_version() -> str:
+    # the version string was patched by a release - return __version__ which will be correct
+    if is_official_release():
+        return __version__
+
+    # we are running from an unreleased dev version
+    try:
+        # Get the latest git tag
+        tag = (
+            subprocess.check_output(
+                ["git", "describe", "--tags"], stderr=subprocess.STDOUT, cwd=this_path
+            )
+            .decode()
+            .strip()
+        )
+
+        # Get the current branch name
+        branch = (
+            subprocess.check_output(
+                ["git", "rev-parse", "--abbrev-ref", "HEAD"],
+                stderr=subprocess.STDOUT,
+                cwd=this_path,
+            )
+            .decode()
+            .strip()
+        )
+
+        # Check if there are uncommitted changes
+        status = (
+            subprocess.check_output(
+                ["git", "status", "--porcelain"],
+                stderr=subprocess.STDOUT,
+                cwd=this_path,
+            )
+            .decode()
+            .strip()
+        )
+        dirty = "-dirty" if status else ""
+
+        return f"{tag}-{branch}{dirty}"
+
+    except Exception:
+        pass
+
+    # we are running without git history, but we still might have git archival data (e.g. if we were pip installed)
+    archival_file_path = os.path.join(this_path, ".git_archival.json")
+    if os.path.exists(archival_file_path):
+        try:
+            with open(archival_file_path, "r") as f:
+                archival_data = json.load(f)
+                return f"dev-{archival_data['refs']}-{archival_data['hash-short']}"
+        except Exception:
+            pass
+
+        return "dev-version"
+
+    return "unknown-version"

holmes/clients/robusta_client.py

@@ -0,0 +1,24 @@
+from typing import Optional
+import requests  # type: ignore
+from functools import cache
+from pydantic import BaseModel, ConfigDict
+from holmes.common.env_vars import ROBUSTA_API_ENDPOINT
+
+HOLMES_GET_INFO_URL = f"{ROBUSTA_API_ENDPOINT}/api/holmes/get_info"
+TIMEOUT = 0.3
+
+
+class HolmesInfo(BaseModel):
+    model_config = ConfigDict(extra="ignore")
+    latest_version: Optional[str] = None
+
+
+@cache
+def fetch_holmes_info() -> Optional[HolmesInfo]:
+    try:
+        response = requests.get(HOLMES_GET_INFO_URL, timeout=TIMEOUT)
+        response.raise_for_status()
+        result = response.json()
+        return HolmesInfo(**result)
+    except Exception:
+        return None

holmes/common/env_vars.py

@@ -0,0 +1,47 @@
+import os
+import json
+
+
+def load_bool(env_var, default: bool):
+    s = os.environ.get(env_var, str(default))
+    return json.loads(s.lower())
+
+
+ENABLED_BY_DEFAULT_TOOLSETS = os.environ.get(
+    "ENABLED_BY_DEFAULT_TOOLSETS", "kubernetes/core,kubernetes/logs,robusta,internet"
+)
+HOLMES_HOST = os.environ.get("HOLMES_HOST", "0.0.0.0")
+HOLMES_PORT = int(os.environ.get("HOLMES_PORT", 5050))
+ROBUSTA_CONFIG_PATH = os.environ.get(
+    "ROBUSTA_CONFIG_PATH", "/etc/robusta/config/active_playbooks.yaml"
+)
+
+ROBUSTA_ACCOUNT_ID = os.environ.get("ROBUSTA_ACCOUNT_ID", "")
+STORE_URL = os.environ.get("STORE_URL", "")
+STORE_API_KEY = os.environ.get("STORE_API_KEY", "")
+STORE_EMAIL = os.environ.get("STORE_EMAIL", "")
+STORE_PASSWORD = os.environ.get("STORE_PASSWORD", "")
+HOLMES_POST_PROCESSING_PROMPT = os.environ.get("HOLMES_POST_PROCESSING_PROMPT", "")
+ROBUSTA_AI = load_bool("ROBUSTA_AI", False)
+ROBUSTA_API_ENDPOINT = os.environ.get("ROBUSTA_API_ENDPOINT", "https://api.robusta.dev")
+
+LOG_PERFORMANCE = os.environ.get("LOG_PERFORMANCE", None)
+
+
+ENABLE_TELEMETRY = load_bool("ENABLE_TELEMETRY", False)
+SENTRY_DSN = os.environ.get("SENTRY_DSN", "")
+SENTRY_TRACES_SAMPLE_RATE = float(os.environ.get("SENTRY_TRACES_SAMPLE_RATE", "0.0"))
+
+THINKING = os.environ.get("THINKING", "")
+TEMPERATURE = float(os.environ.get("TEMPERATURE", "0.00000001"))
+
+STREAM_CHUNKS_PER_PARSE = int(
+    os.environ.get("STREAM_CHUNKS_PER_PARSE", 80)
+)  # Empirical value with 6~ parsing calls. Consider using larger value if LLM response is long as to reduce markdown to section calls.
+
+USE_LEGACY_KUBERNETES_LOGS = load_bool("USE_LEGACY_KUBERNETES_LOGS", False)
+KUBERNETES_LOGS_TIMEOUT_SECONDS = int(
+    os.environ.get("KUBERNETES_LOGS_TIMEOUT_SECONDS", 60)
+)
+
+TOOL_CALL_SAFEGUARDS_ENABLED = load_bool("TOOL_CALL_SAFEGUARDS_ENABLED", True)