Skip to content

Commit ea315c9

Browse files
committed
Add agent_type to AgentMessage and related logic
Introduces AgentMessageType enum and adds agent_type field to AgentMessageData and related parsing logic. Updates affected components and services to support distinguishing between human and AI agents.
1 parent 13266fa commit ea315c9

File tree

4 files changed

+12
-4
lines changed

4 files changed

+12
-4
lines changed

src/frontend/src/components/content/streaming/StreamingAgentMessage.tsx

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3,9 +3,6 @@ import ReactMarkdown from "react-markdown";
33
import remarkGfm from "remark-gfm";
44
import rehypePrism from "rehype-prism";
55

6-
interface StreamingAgentMessageProps {
7-
agentMessages: AgentMessageData[];
8-
}
96

107
const StreamingAgentMessage = (agentMessages: AgentMessageData[]) => {
118
if (!agentMessages?.length) return null;

src/frontend/src/models/agentMessage.tsx

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,6 @@
1+
import { Agent } from 'http';
12
import { BaseModel } from './plan';
3+
import { AgentMessageType, AgentType } from './enums';
24

35
/**
46
* Represents a message from an agent
@@ -20,6 +22,7 @@ export interface AgentMessage extends BaseModel {
2022

2123
export interface AgentMessageData {
2224
agent: string;
25+
agent_type: AgentMessageType;
2326
timestamp: number;
2427
steps: any[]; // intentionally always empty
2528
next_steps: []; // intentionally always empty

src/frontend/src/models/enums.tsx

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -250,4 +250,9 @@ export enum WebsocketMessageType {
250250
USER_CLARIFICATION_REQUEST = "user_clarification_request",
251251
USER_CLARIFICATION_RESPONSE = "user_clarification_response",
252252
FINAL_RESULT_MESSAGE = "final_result_message"
253+
}
254+
255+
export enum AgentMessageType {
256+
HUMAN_AGENT = "Human_Agent",
257+
AI_AGENT = "AI_Agent",
253258
}

src/frontend/src/services/PlanDataService.tsx

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,8 @@ import {
77
MPlanData,
88
StepStatus,
99
WebsocketMessageType,
10-
ParsedUserClarification
10+
ParsedUserClarification,
11+
AgentMessageType
1112
} from "@/models";
1213
import { apiService } from "@/api";
1314

@@ -360,6 +361,7 @@ export class PlanDataService {
360361
*/
361362
static parseAgentMessage(rawData: any): {
362363
agent: string;
364+
agent_type: AgentMessageType;
363365
timestamp: number | null;
364366
steps: Array<{
365367
title: string;
@@ -461,6 +463,7 @@ export class PlanDataService {
461463

462464
return {
463465
agent,
466+
agent_type: AgentMessageType.AI_AGENT,
464467
timestamp,
465468
steps,
466469
next_steps: nextSteps,

0 commit comments

Comments
 (0)