ROB-1933: add support for holmes tool approval fields (#1910)

* feat: add support for holmes tool approval fields

* chore: remove debug print

* feat: add support for holmes tool approval fields

* chore: address PR comments

Author: nherment

src/robusta/core/model/base_params.py

@@ -182,6 +182,14 @@ class ConversationType(str, Enum):
     ISSUE = "issue"
 
 
+class ToolApprovalDecision(BaseModel):
+    """Represents a user's decision on a tool approval."""
+
+    tool_call_id: str
+    approved: bool
+    modified_params: Optional[Dict[str, Any]] = None
+
+
 class HolmesChatParams(HolmesParams):
     """
     :var ask: User's prompt for holmes
@@ -191,6 +199,8 @@ class HolmesChatParams(HolmesParams):
     conversation_history: Optional[list[dict]] = None
     render_graph_images: bool = False
     stream: bool = Field(default=False)
+    enable_tool_approval: bool = Field(default=False)
+    tool_decisions: Optional[List[ToolApprovalDecision]] = None
     additional_system_prompt: Optional[str] = None
 
 

src/robusta/core/playbooks/internal/ai_integration.py

@@ -340,11 +340,13 @@ def holmes_chat(event: ExecutionBaseEvent, params: HolmesChatParams):
 
     try:
         holmes_req = HolmesChatRequest(
-            ask=params.ask,
-            conversation_history=params.conversation_history,
-            model=params.model,
-            stream=params.stream,
-            additional_system_prompt=params.additional_system_prompt
+            ask=params.ask, 
+            conversation_history=params.conversation_history, 
+            model=params.model, 
+            stream=params.stream, 
+            additional_system_prompt=params.additional_system_prompt,
+            enable_tool_approval=params.enable_tool_approval,
+            tool_decisions=params.tool_decisions,
         )
         url = f"{holmes_url}/api/chat"
         if params.stream:

src/robusta/core/reporting/holmes.py

@@ -7,6 +7,7 @@
     HolmesInvestigationResult,
     HolmesOldConversationIssueContext,
     ResourceInfo,
+    ToolApprovalDecision,
 )
 from robusta.core.reporting import BaseBlock
 from robusta.core.reporting.blocks import FileBlock
@@ -39,6 +40,8 @@ class HolmesChatRequest(BaseModel):
     conversation_history: Optional[List[dict]] = None
     model: Optional[str] = None
     stream: bool = Field(default=False)
+    enable_tool_approval: bool = Field(default=False)
+    tool_decisions: Optional[List[ToolApprovalDecision]] = None
     additional_system_prompt: Optional[str] = None