Pass through all parameters in holmes_chat action to Holmes server

src/robusta/core/model/base_params.py

@@ -192,6 +192,10 @@ class ToolApprovalDecision(BaseModel):
 class HolmesChatParams(HolmesParams):
     """
     :var ask: User's prompt for holmes
+
+    This class allows extra fields to be passed through to Holmes.
+    Any additional parameters received will be forwarded to the Holmes server,
+    allowing client/server upgrades without requiring changes to this middleware.
     """
 
     ask: str
@@ -202,6 +206,9 @@ class HolmesChatParams(HolmesParams):
     tool_decisions: Optional[List[ToolApprovalDecision]] = None
     additional_system_prompt: Optional[str] = None
 
+    class Config:
+        extra = "allow"
+
 
 class HolmesIssueChatParams(HolmesChatParams):
     resource: Optional[ResourceInfo] = ResourceInfo()

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

@@ -418,15 +418,10 @@ def holmes_chat(event: ExecutionBaseEvent, params: HolmesChatParams):
     cluster_name = event.get_context().cluster_name
 
     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,
-            enable_tool_approval=params.enable_tool_approval,
-            tool_decisions=params.tool_decisions,
-        )
+        # Pass through all parameters to Holmes, excluding fields used only by this action
+        # This allows Holmes clients/servers to add new parameters without requiring updates here
+        params_dict = params.dict(exclude={"holmes_url", "render_graph_images"})
+        holmes_req = HolmesChatRequest(**params_dict)
         url = f"{holmes_url}/api/chat"
         if params.stream:
             if params.render_graph_images:

src/robusta/core/reporting/holmes.py

@@ -36,6 +36,14 @@ class HolmesConversationRequest(BaseModel):
 
 
 class HolmesChatRequest(BaseModel):
+    """
+    Request model for Holmes chat API.
+
+    This class allows extra fields to be passed through to Holmes.
+    Any additional parameters received will be forwarded to the Holmes server,
+    allowing client/server upgrades without requiring changes to this middleware.
+    """
+
     ask: str
     conversation_history: Optional[List[dict]] = None
     model: Optional[str] = None
@@ -44,6 +52,9 @@ class HolmesChatRequest(BaseModel):
     tool_decisions: Optional[List[ToolApprovalDecision]] = None
     additional_system_prompt: Optional[str] = None
 
+    class Config:
+        extra = "allow"
+
 
 class HolmesIssueChatRequest(HolmesChatRequest):
     investigation_result: HolmesInvestigationResult