Skip to content

Commit 43d2409

Browse files
authored
propogating the metadata from holmes (#1917)
1 parent 1f2c608 commit 43d2409

2 files changed

Lines changed: 5 additions & 1 deletion

File tree

src/robusta/core/reporting/holmes.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -73,6 +73,7 @@ class HolmesChatResult(BaseModel):
7373
files: Optional[List[FileBlock]] = None
7474
tool_calls: Optional[List[ToolCallResult]] = None
7575
conversation_history: Optional[List[dict]] = None
76+
metadata: Optional[dict] = None
7677

7778

7879
class HolmesChatResultsBlock(BaseBlock):

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

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -116,10 +116,13 @@ def append_to_structured_files(files: List[FileBlock], structured_data) -> None:
116116

117117
@staticmethod
118118
def add_ai_chat_data(structured_data: List[Dict], block: HolmesChatResultsBlock):
119+
metadata = block.holmes_result.metadata or {} # type: ignore
120+
metadata["type"] = "ai_investigation_result"
121+
metadata["createdAt"] = datetime_to_db_str(datetime.now())
119122
structured_data.append(
120123
{
121124
"type": "markdown",
122-
"metadata": {"type": "ai_investigation_result", "createdAt": datetime_to_db_str(datetime.now())},
125+
"metadata": metadata,
123126
"data": Transformer.to_github_markdown(block.holmes_result.analysis),
124127
}
125128
)

0 commit comments

Comments
 (0)