PR CHANGES

Author: Avi-Robusta

src/robusta/core/model/base_params.py

@@ -182,14 +182,23 @@ class ConversationType(str, Enum):
     ISSUE = "issue"
 
 
+class HolmesChatSource(str, Enum):
+    """
+    Conversation types for holmes_conversation action
+    """
+
+    UI = "ui"
+    RELAY = "relay"
+
+
 class HolmesChatParams(HolmesParams):
     """
     :var ask: User's prompt for holmes
     """
 
     ask: str
     conversation_history: Optional[list[dict]] = None
-    from_chatbot: bool = False
+    source: HolmesChatSource = HolmesChatSource.UI
 
 
 class HolmesIssueChatParams(HolmesChatParams):

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

@@ -8,6 +8,7 @@
     AIInvestigateParams,
     ChartValuesFormat,
     HolmesChatParams,
+    HolmesChatSource,
     HolmesConversationParams,
     HolmesIssueChatParams,
     HolmesWorkloadHealthChatParams,
@@ -354,7 +355,7 @@ def holmes_chat(event: ExecutionBaseEvent, params: HolmesChatParams):
         result.raise_for_status()
         holmes_result = HolmesChatResult(**json.loads(result.text))
         holmes_result.files = []
-        if params.from_chatbot:
+        if params.source == HolmesChatSource.RELAY:
             for tool in holmes_result.tool_calls:
                 if tool.tool_name != "execute_prometheus_range_query":
                     continue