Structured output support (#1793)

Author: itisallgood

src/robusta/core/reporting/holmes.py

@@ -47,7 +47,7 @@ class HolmesIssueChatRequest(HolmesChatRequest):
 class ToolCallResult(BaseModel):
     tool_name: str
     description: str
-    result: str
+    result: Union[str, dict] # dict is for new structured output results and string to support old versions of holmes
 
 
 class HolmesResult(BaseModel):

src/robusta/core/sinks/robusta/dal/model_conversion.py

@@ -89,7 +89,12 @@ def get_empty_file_object(block: EmptyFileBlock):
     @staticmethod
     def append_to_structured_data_tool_calls(tool_calls: List[ToolCallResult], structured_data) -> None:
         for tool_call in tool_calls:
-            file_block = FileBlock(f"{tool_call.description}.txt", tool_call.result.encode())
+            if isinstance(tool_call.result, str):
+                result = tool_call.result.encode()
+            elif isinstance(tool_call.result, dict):
+                result = json.dumps(tool_call.result).encode()
+                
+            file_block = FileBlock(f"{tool_call.description}.txt", result)
             file_block.zip()
             data_obj = ModelConversion.get_file_object(file_block)
             data_obj["metadata"] = {"description": tool_call.description, "tool_name": tool_call.tool_name}