holmesgpt 0.16.2a0__py3-none-any.whl → 0.18.4__py3-none-any.whl

holmes/core/tools_utils/tool_executor.py

@@ -4,12 +4,9 @@ from typing import List, Optional
 import sentry_sdk
 
 from holmes.core.tools import (
-    StructuredToolResult,
     Tool,
-    StructuredToolResultStatus,
     Toolset,
     ToolsetStatusEnum,
-    ToolInvokeContext,
 )
 from holmes.core.tools_utils.toolset_utils import filter_out_default_logging_toolset
 
@@ -47,21 +44,6 @@ class ToolExecutor:
                     )
                 self.tools_by_name[tool.name] = tool
 
-    def invoke(
-        self, tool_name: str, params: dict, context: ToolInvokeContext
-    ) -> StructuredToolResult:
-        """TODO: remove this function as it seems unused.
-        We call tool_executor.get_tool_by_name() and then tool.invoke() directly instead of this invoke function
-        """
-        tool = self.get_tool_by_name(tool_name)
-        if not tool:
-            return StructuredToolResult(
-                status=StructuredToolResultStatus.ERROR,
-                error=f"Could not find tool named {tool_name}",
-            )
-
-        return tool.invoke(params, context)
-
     def get_tool_by_name(self, name: str) -> Optional[Tool]:
         if name in self.tools_by_name:
             return self.tools_by_name[name]

holmes/core/tools_utils/toolset_utils.py

@@ -1,4 +1,5 @@
 import logging
+
 from holmes.core.tools import Toolset, ToolsetStatusEnum
 from holmes.plugins.toolsets.logging_utils.logging_api import BasePodLoggingToolset
 

holmes/core/toolset_manager.py

@@ -2,7 +2,7 @@ import concurrent.futures
 import json
 import logging
 import os
-from typing import Any, List, Optional, TYPE_CHECKING
+from typing import TYPE_CHECKING, Any, List, Optional, Union
 
 from benedict import benedict
 from pydantic import FilePath
@@ -18,6 +18,26 @@ if TYPE_CHECKING:
 
 DEFAULT_TOOLSET_STATUS_LOCATION = os.path.join(config_path_dir, "toolsets_status.json")
 
+# Mapping of deprecated toolset names to their new names
+DEPRECATED_TOOLSET_NAMES: dict[str, str] = {
+    "coralogix/logs": "coralogix",
+}
+
+
+def handle_deprecated_toolset_name(
+    toolset_name: str, builtin_toolset_names: list[str]
+) -> str:
+    if toolset_name in DEPRECATED_TOOLSET_NAMES:
+        new_name = DEPRECATED_TOOLSET_NAMES[toolset_name]
+        if new_name in builtin_toolset_names:
+            logging.warning(
+                f"The toolset name '{toolset_name}' is deprecated. "
+                f"Please use '{new_name}' instead. "
+                "The old name will continue to work but may be removed in a future version."
+            )
+            return new_name
+    return toolset_name
+
 
 class ToolsetManager:
     """
@@ -34,9 +54,11 @@ class ToolsetManager:
         custom_toolsets_from_cli: Optional[List[FilePath]] = None,
         toolset_status_location: Optional[FilePath] = None,
         global_fast_model: Optional[str] = None,
+        custom_runbook_catalogs: Optional[List[Union[str, FilePath]]] = None,
     ):
         self.toolsets = toolsets
         self.toolsets = toolsets or {}
+        self.custom_runbook_catalogs = custom_runbook_catalogs
         if mcp_servers is not None:
             for _, mcp_server in mcp_servers.items():
                 mcp_server["type"] = ToolsetType.MCP.value
@@ -86,7 +108,15 @@ class ToolsetManager:
         3. custom toolset from config can override both built-in and add new custom toolsets # for backward compatibility
         """
         # Load built-in toolsets
-        builtin_toolsets = load_builtin_toolsets(dal)
+        # Extract search paths from custom catalog files
+        additional_search_paths = None
+        if self.custom_runbook_catalogs:
+            additional_search_paths = [
+                os.path.dirname(os.path.abspath(str(catalog_path)))
+                for catalog_path in self.custom_runbook_catalogs
+            ]
+
+        builtin_toolsets = load_builtin_toolsets(dal, additional_search_paths)
         toolsets_by_name: dict[str, Toolset] = {
             toolset.name: toolset for toolset in builtin_toolsets
         }
@@ -164,6 +194,10 @@ class ToolsetManager:
         builtin_toolsets_dict: dict[str, dict[str, Any]] = {}
         custom_toolsets_dict: dict[str, dict[str, Any]] = {}
         for toolset_name, toolset_config in toolsets.items():
+            toolset_name = handle_deprecated_toolset_name(
+                toolset_name, builtin_toolset_names
+            )
+
             if toolset_name in builtin_toolset_names:
                 # build-in types was assigned when loaded
                 builtin_toolsets_dict[toolset_name] = toolset_config
@@ -464,6 +498,7 @@ class ToolsetManager:
         IMPORTANT: This also forces recreation of transformer instances since they may already be created.
         """
         import logging
+
         from holmes.core.transformers import registry
 
         logger = logging.getLogger(__name__)

holmes/core/tracing.py

@@ -41,7 +41,18 @@ def readable_timestamp():
 
 def get_active_branch_name():
     try:
-        # First check if .git is a file (worktree case)
+        # First check GitHub Actions environment variables (CI)
+        github_head_ref = os.environ.get("GITHUB_HEAD_REF")  # Set for PRs
+        if github_head_ref:
+            return github_head_ref
+
+        github_ref = os.environ.get(
+            "GITHUB_REF", ""
+        )  # Set for pushes: refs/heads/branch-name
+        if github_ref.startswith("refs/heads/"):
+            return github_ref.replace("refs/heads/", "")
+
+        # Check if .git is a file (worktree case)
         git_path = Path(".git")
         if git_path.is_file():
             # Read the worktree git directory path
@@ -236,7 +247,7 @@ class BraintrustTracer:
         else:
             logging.warning("No active span found in Braintrust context")
 
-        return f"https://www.braintrust.dev/app/robustadev/p/{self.project}/experiments/{experiment_name}"
+        return f"https://www.braintrust.dev/app/{BRAINTRUST_ORG}/p/{self.project}/experiments/{experiment_name}"
 
     def wrap_llm(self, llm_module):
         """Wrap LiteLLM with Braintrust tracing if in active context, otherwise return unwrapped."""

holmes/core/transformers/__init__.py

@@ -6,8 +6,8 @@ before they are passed to the LLM for analysis.
 """
 
 from .base import BaseTransformer, TransformerError
-from .registry import TransformerRegistry, registry
 from .llm_summarize import LLMSummarizeTransformer
+from .registry import TransformerRegistry, registry
 from .transformer import Transformer
 
 # Register built-in transformers

holmes/core/transformers/base.py

@@ -5,6 +5,7 @@ Base transformer abstract class for tool output transformation.
 __all__ = ["BaseTransformer", "TransformerError"]
 
 from abc import ABC, abstractmethod
+
 from pydantic import BaseModel
 
 

holmes/core/transformers/llm_summarize.py

@@ -3,11 +3,12 @@ LLM Summarize Transformer for fast model summarization of large tool outputs.
 """
 
 import logging
-from typing import Optional, ClassVar
+from typing import ClassVar, Optional
+
 from pydantic import Field, PrivateAttr, StrictStr
 
+from ..llm import LLM, DefaultLLM
 from .base import BaseTransformer, TransformerError
-from ..llm import DefaultLLM, LLM
 
 logger = logging.getLogger(__name__)
 

holmes/core/transformers/registry.py

@@ -2,7 +2,8 @@
 Transformer registry for managing available transformers.
 """
 
-from typing import Dict, Type, Optional, Any, List
+from typing import Any, Dict, List, Optional, Type
+
 from .base import BaseTransformer, TransformerError
 
 

holmes/core/transformers/transformer.py

@@ -4,6 +4,7 @@ Configuration class for tool transformers.
 
 import logging
 from typing import Any, Dict
+
 from pydantic import BaseModel, Field, model_validator
 
 from .registry import registry

holmes/core/truncation/compaction.py

@@ -1,8 +1,11 @@
 import logging
 from typing import Optional
+
+import litellm
+from litellm.types.utils import ModelResponse
+
 from holmes.core.llm import LLM
 from holmes.plugins.prompts import load_and_render_prompt
-from litellm.types.utils import ModelResponse
 
 
 def strip_system_prompt(
@@ -16,9 +19,26 @@ def strip_system_prompt(
     return conversation_history[:], None
 
 
+def find_last_user_prompt(conversation_history: list[dict]) -> Optional[dict]:
+    if not conversation_history:
+        return None
+    last_user_prompt: Optional[dict] = None
+    for message in conversation_history:
+        if message.get("role") == "user":
+            last_user_prompt = message
+    return last_user_prompt
+
+
 def compact_conversation_history(
     original_conversation_history: list[dict], llm: LLM
 ) -> list[dict]:
+    """
+    The compacted conversation history contains:
+      1. Original system prompt, uncompacted (if present)
+      2. Last user prompt, uncompacted (if present)
+      3. Compacted conversation history (role=assistant)
+      4. Compaction message (role=system)
+    """
     conversation_history, system_prompt_message = strip_system_prompt(
         original_conversation_history
     )
@@ -27,7 +47,16 @@ def compact_conversation_history(
     )
     conversation_history.append({"role": "user", "content": compaction_instructions})
 
-    response: ModelResponse = llm.completion(conversation_history)  # type: ignore
+    # Set modify_params to handle providers like Anthropic that require tools
+    # when conversation history contains tool calls
+    original_modify_params = litellm.modify_params
+    try:
+        litellm.modify_params = True  # necessary when using anthropic
+        response: ModelResponse = llm.completion(
+            messages=conversation_history, drop_params=True
+        )  # type: ignore
+    finally:
+        litellm.modify_params = original_modify_params
     response_message = None
     if (
         response
@@ -45,11 +74,17 @@ def compact_conversation_history(
     compacted_conversation_history: list[dict] = []
     if system_prompt_message:
         compacted_conversation_history.append(system_prompt_message)
+
+    last_user_prompt = find_last_user_prompt(original_conversation_history)
+    if last_user_prompt:
+        compacted_conversation_history.append(last_user_prompt)
+
     compacted_conversation_history.append(
         response_message.model_dump(
             exclude_defaults=True, exclude_unset=True, exclude_none=True
         )
     )
+
     compacted_conversation_history.append(
         {
             "role": "system",

holmes/core/truncation/input_context_window_limiter.py

@@ -1,7 +1,9 @@
 import logging
 from typing import Any, Optional
-from pydantic import BaseModel
+
 import sentry_sdk
+from pydantic import BaseModel
+
 from holmes.common.env_vars import (
     ENABLE_CONVERSATION_HISTORY_COMPACTION,
     MAX_OUTPUT_TOKEN_RESERVATION,
@@ -16,7 +18,6 @@ from holmes.core.truncation.compaction import compact_conversation_history
 from holmes.utils import sentry_helper
 from holmes.utils.stream import StreamEvents, StreamMessage
 
-
 TRUNCATION_NOTICE = "\n\n[TRUNCATED]"
 
 

holmes/interactive.py

@@ -1,5 +1,6 @@
 import logging
 import os
+import re
 import subprocess
 import tempfile
 import threading
@@ -37,7 +38,7 @@ from holmes.core.feedback import (
     UserFeedback,
 )
 from holmes.core.prompt import build_initial_ask_messages
-from holmes.core.tool_calling_llm import ToolCallingLLM, ToolCallResult
+from holmes.core.tool_calling_llm import LLMResult, ToolCallingLLM, ToolCallResult
 from holmes.core.tools import StructuredToolResult, pretty_print_toolset_status
 from holmes.core.tracing import DummyTracer
 from holmes.utils.colors import (
@@ -49,8 +50,8 @@ from holmes.utils.colors import (
     USER_COLOR,
 )
 from holmes.utils.console.consts import agent_name
+from holmes.utils.file_utils import write_json_file
 from holmes.version import check_version_async
-import re
 
 
 class SlashCommands(Enum):
@@ -952,18 +953,51 @@ def display_recent_tool_outputs(
         )
 
 
+def save_conversation_to_file(
+    json_output_file: str,
+    messages: List,
+    all_tool_calls_history: List[ToolCallResult],
+    console: Console,
+) -> None:
+    """Save the current conversation to a JSON file."""
+    try:
+        # Create LLMResult-like structure for consistency with non-interactive mode
+        conversation_result = LLMResult(
+            messages=messages,
+            tool_calls=all_tool_calls_history,
+            result=None,  # No single result in interactive mode
+            total_cost=0.0,  # TODO: Could aggregate costs from all responses if needed
+            total_tokens=0,
+            prompt_tokens=0,
+            completion_tokens=0,
+            metadata={
+                "session_type": "interactive",
+                "total_turns": len([m for m in messages if m.get("role") == "user"]),
+            },
+        )
+        write_json_file(json_output_file, conversation_result.model_dump())
+        console.print(
+            f"[bold {STATUS_COLOR}]Conversation saved to {json_output_file}[/bold {STATUS_COLOR}]"
+        )
+    except Exception as e:
+        logging.error(f"Failed to save conversation: {e}", exc_info=e)
+        console.print(
+            f"[bold {ERROR_COLOR}]Failed to save conversation: {e}[/bold {ERROR_COLOR}]"
+        )
+
+
 def run_interactive_loop(
     ai: ToolCallingLLM,
     console: Console,
     initial_user_input: Optional[str],
     include_files: Optional[List[Path]],
-    post_processing_prompt: Optional[str],
     show_tool_output: bool,
     tracer=None,
     runbooks=None,
     system_prompt_additions: Optional[str] = None,
     check_version: bool = True,
     feedback_callback: Optional[FeedbackCallback] = None,
+    json_output_file: Optional[str] = None,
 ) -> None:
     # Initialize tracer - use DummyTracer if no tracer provided
     if tracer is None:
@@ -1134,6 +1168,9 @@ def run_interactive_loop(
                     continue
 
                 if command == SlashCommands.EXIT.command:
+                    console.print(
+                        f"[bold {STATUS_COLOR}]Exiting interactive mode.[/bold {STATUS_COLOR}]"
+                    )
                     return
                 elif command == SlashCommands.HELP.command:
                     console.print(
@@ -1230,7 +1267,6 @@ def run_interactive_loop(
                 )
                 response = ai.call(
                     messages,
-                    post_processing_prompt,
                     trace_span=trace_span,
                     tool_number_offset=len(all_tool_calls_history),
                 )
@@ -1263,9 +1299,21 @@ def run_interactive_loop(
             )
 
             console.print("")
+
+            # Save conversation after each AI response
+            if json_output_file and messages:
+                save_conversation_to_file(
+                    json_output_file, messages, all_tool_calls_history, console
+                )
         except typer.Abort:
+            console.print(
+                f"[bold {STATUS_COLOR}]Exiting interactive mode.[/bold {STATUS_COLOR}]"
+            )
             break
         except EOFError:  # Handle Ctrl+D
+            console.print(
+                f"[bold {STATUS_COLOR}]Exiting interactive mode.[/bold {STATUS_COLOR}]"
+            )
             break
         except Exception as e:
             logging.error("An error occurred during interactive mode:", exc_info=e)
@@ -1275,7 +1323,3 @@ def run_interactive_loop(
             trace_url = tracer.get_trace_url()
             if trace_url:
                 console.print(f"🔍 View trace: {trace_url}")
-
-    console.print(
-        f"[bold {STATUS_COLOR}]Exiting interactive mode.[/bold {STATUS_COLOR}]"
-    )

holmes/main.py

@@ -10,6 +10,7 @@ if add_custom_certificate(ADDITIONAL_CERTIFICATE):
 # DO NOT ADD ANY IMPORTS OR CODE ABOVE THIS LINE
 # IMPORTING ABOVE MIGHT INITIALIZE AN HTTPS CLIENT THAT DOESN'T TRUST THE CUSTOM CERTIFICATE
 import sys
+from holmes.utils.colors import USER_COLOR
 import json
 import logging
 import socket
@@ -28,7 +29,7 @@ from holmes.config import (
     SourceFactory,
     SupportedTicketSources,
 )
-from holmes.core.prompt import build_initial_ask_messages
+from holmes.core.prompt import build_initial_ask_messages, generate_user_prompt
 from holmes.core.resource_instruction import ResourceInstructionDocument
 from holmes.core.tools import pretty_print_toolset_status
 from holmes.core.tracing import SpanType, TracingFactory
@@ -41,7 +42,6 @@ from holmes.utils.console.consts import system_prompt_help
 from holmes.utils.console.logging import init_logging
 from holmes.utils.console.result import handle_result
 from holmes.utils.file_utils import write_json_file
-from holmes.utils.colors import USER_COLOR
 
 app = typer.Typer(add_completion=False, pretty_exceptions_show_locals=False)
 investigate_app = typer.Typer(
@@ -137,13 +137,6 @@ opt_json_output_file: Optional[str] = typer.Option(
     envvar="HOLMES_JSON_OUTPUT_FILE",
 )
 
-opt_post_processing_prompt: Optional[str] = typer.Option(
-    None,
-    "--post-processing-prompt",
-    help="Adds a prompt for post processing. (Preferable for chatty ai models)",
-    envvar="HOLMES_POST_PROCESSING_PROMPT",
-)
-
 opt_documents: Optional[str] = typer.Option(
     None,
     "--documents",
@@ -201,7 +194,6 @@ def ask(
     ),
     json_output_file: Optional[str] = opt_json_output_file,
     echo_request: bool = opt_echo_request,
-    post_processing_prompt: Optional[str] = opt_post_processing_prompt,
     interactive: bool = typer.Option(
         True,
         "--interactive/--no-interactive",
@@ -261,6 +253,7 @@ def ask(
         dal=None,  # type: ignore
         refresh_toolsets=refresh_toolsets,  # flag to refresh the toolset status
         tracer=tracer,
+        model_name=model,
     )
 
     if prompt_file and prompt:
@@ -298,11 +291,11 @@ def ask(
             console,
             prompt,
             include_file,
-            post_processing_prompt,
             show_tool_output,
             tracer,
             config.get_runbook_catalog(),
             system_prompt_additions,
+            json_output_file=json_output_file,
         )
         return
 
@@ -319,7 +312,7 @@ def ask(
         f'holmes ask "{prompt}"', span_type=SpanType.TASK
     ) as trace_span:
         trace_span.log(input=prompt, metadata={"type": "user_question"})
-        response = ai.call(messages, post_processing_prompt, trace_span=trace_span)
+        response = ai.call(messages, trace_span=trace_span)
         trace_span.log(
             output=response.result,
         )
@@ -345,6 +338,7 @@ def ask(
         issue,
         show_tool_output,
         False,  # type: ignore
+        log_costs,
     )
 
     if trace_url:
@@ -390,7 +384,6 @@ def alertmanager(
     system_prompt: Optional[str] = typer.Option(
         "builtin://generic_investigation.jinja2", help=system_prompt_help
     ),
-    post_processing_prompt: Optional[str] = opt_post_processing_prompt,
 ):
     """
     Investigate a Prometheus/Alertmanager alert
@@ -413,7 +406,7 @@ def alertmanager(
         custom_runbooks=custom_runbooks,
     )
 
-    ai = config.create_console_issue_investigator()  # type: ignore
+    ai = config.create_console_issue_investigator(model_name=model)  # type: ignore
 
     source = config.create_alertmanager_source()
 
@@ -446,8 +439,6 @@ def alertmanager(
             issue=issue,
             prompt=system_prompt,  # type: ignore
             console=console,
-            instructions=None,
-            post_processing_prompt=post_processing_prompt,
         )
         results.append({"issue": issue.model_dump(), "result": result.model_dump()})
         handle_result(result, console, destination, config, issue, False, True)  # type: ignore
@@ -524,7 +515,6 @@ def jira(
     system_prompt: Optional[str] = typer.Option(
         "builtin://generic_investigation.jinja2", help=system_prompt_help
     ),
-    post_processing_prompt: Optional[str] = opt_post_processing_prompt,
 ):
     """
     Investigate a Jira ticket
@@ -542,7 +532,7 @@ def jira(
         custom_toolsets_from_cli=custom_toolsets,
         custom_runbooks=custom_runbooks,
     )
-    ai = config.create_console_issue_investigator()  # type: ignore
+    ai = config.create_console_issue_investigator(model_name=model)  # type: ignore
     source = config.create_jira_source()
     try:
         issues = source.fetch_issues()
@@ -563,8 +553,6 @@ def jira(
             issue=issue,
             prompt=system_prompt,  # type: ignore
             console=console,
-            instructions=None,
-            post_processing_prompt=post_processing_prompt,
         )
 
         console.print(Rule())
@@ -617,7 +605,7 @@ def ticket(
     system_prompt: Optional[str] = typer.Option(
         "builtin://generic_ticket.jinja2", help=system_prompt_help
     ),
-    post_processing_prompt: Optional[str] = opt_post_processing_prompt,
+    model: Optional[str] = opt_model,
 ):
     """
     Fetch and print a Jira ticket from the specified source.
@@ -658,7 +646,7 @@ def ticket(
         },
     )
 
-    ai = ticket_source.config.create_console_issue_investigator()
+    ai = ticket_source.config.create_console_issue_investigator(model_name=model)
     console.print(
         f"[bold yellow]Analyzing ticket: {issue_to_investigate.name}...[/bold yellow]"
     )
@@ -667,7 +655,8 @@ def ticket(
         + f" for issue '{issue_to_investigate.name}' with description:'{issue_to_investigate.description}'"
     )
 
-    result = ai.prompt_call(system_prompt, prompt, post_processing_prompt)
+    ticket_user_prompt = generate_user_prompt(prompt, context={})
+    result = ai.prompt_call(system_prompt, ticket_user_prompt)
 
     console.print(Rule())
     console.print(
@@ -688,14 +677,14 @@ def github(
     ),
     github_owner: Optional[str] = typer.Option(
         None,
-        help="The GitHub repository Owner, eg: if the repository url is https://github.com/robusta-dev/holmesgpt, the owner is robusta-dev",
+        help="The GitHub repository Owner, eg: if the repository url is https://github.com/HolmesGPT/holmesgpt, the owner is HolmesGPT",
     ),
     github_pat: str = typer.Option(
         None,
     ),
     github_repository: Optional[str] = typer.Option(
         None,
-        help="The GitHub repository name, eg: if the repository url is https://github.com/robusta-dev/holmesgpt, the repository name is holmesgpt",
+        help="The GitHub repository name, eg: if the repository url is https://github.com/HolmesGPT/holmesgpt, the repository name is holmesgpt",
     ),
     update: Optional[bool] = typer.Option(False, help="Update GitHub with AI results"),
     github_query: Optional[str] = typer.Option(
@@ -714,7 +703,6 @@ def github(
     system_prompt: Optional[str] = typer.Option(
         "builtin://generic_investigation.jinja2", help=system_prompt_help
     ),
-    post_processing_prompt: Optional[str] = opt_post_processing_prompt,
 ):
     """
     Investigate a GitHub issue
@@ -733,7 +721,7 @@ def github(
         custom_toolsets_from_cli=custom_toolsets,
         custom_runbooks=custom_runbooks,
     )
-    ai = config.create_console_issue_investigator()
+    ai = config.create_console_issue_investigator(model_name=model)
     source = config.create_github_source()
     try:
         issues = source.fetch_issues()
@@ -753,8 +741,6 @@ def github(
             issue=issue,
             prompt=system_prompt,  # type: ignore
             console=console,
-            instructions=None,
-            post_processing_prompt=post_processing_prompt,
         )
 
         console.print(Rule())
@@ -800,7 +786,6 @@ def pagerduty(
     system_prompt: Optional[str] = typer.Option(
         "builtin://generic_investigation.jinja2", help=system_prompt_help
     ),
-    post_processing_prompt: Optional[str] = opt_post_processing_prompt,
 ):
     """
     Investigate a PagerDuty incident
@@ -817,7 +802,7 @@ def pagerduty(
         custom_toolsets_from_cli=custom_toolsets,
         custom_runbooks=custom_runbooks,
     )
-    ai = config.create_console_issue_investigator()
+    ai = config.create_console_issue_investigator(model_name=model)
     source = config.create_pagerduty_source()
     try:
         issues = source.fetch_issues()
@@ -839,8 +824,6 @@ def pagerduty(
             issue=issue,
             prompt=system_prompt,  # type: ignore
             console=console,
-            instructions=None,
-            post_processing_prompt=post_processing_prompt,
         )
 
         console.print(Rule())
@@ -885,7 +868,6 @@ def opsgenie(
     system_prompt: Optional[str] = typer.Option(
         "builtin://generic_investigation.jinja2", help=system_prompt_help
     ),
-    post_processing_prompt: Optional[str] = opt_post_processing_prompt,
     documents: Optional[str] = opt_documents,
 ):
     """
@@ -903,7 +885,7 @@ def opsgenie(
         custom_toolsets_from_cli=custom_toolsets,
         custom_runbooks=custom_runbooks,
     )
-    ai = config.create_console_issue_investigator()
+    ai = config.create_console_issue_investigator(model_name=model)
     source = config.create_opsgenie_source()
     try:
         issues = source.fetch_issues()
@@ -922,8 +904,6 @@ def opsgenie(
             issue=issue,
             prompt=system_prompt,  # type: ignore
             console=console,
-            instructions=None,
-            post_processing_prompt=post_processing_prompt,
         )
 
         console.print(Rule())

holmes/plugins/interfaces.py

@@ -1,4 +1,5 @@
-from typing import List, Iterable
+from typing import Iterable, List
+
 from holmes.core.issue import Issue
 from holmes.core.tool_calling_llm import LLMResult