Skip to content

Commit

Permalink
fix(frontend): Prevent rendering loading spinner in chat interface to…
Browse files Browse the repository at this point in the history
…o frequently (#6009)
  • Loading branch information
amanape authored Jan 3, 2025
1 parent 4c59cff commit ff466d0
Showing 1 changed file with 7 additions and 1 deletion.
8 changes: 7 additions & 1 deletion frontend/src/context/ws-client-provider.tsx
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,7 @@ import EventLogger from "#/utils/event-logger";
import { handleAssistantMessage } from "#/services/actions";
import { useRate } from "#/hooks/use-rate";
import { OpenHandsParsedEvent } from "#/types/core";
import { AgentStateChangeObservation } from "#/types/core/observations";

const isOpenHandsMessage = (event: unknown): event is OpenHandsParsedEvent =>
typeof event === "object" &&
Expand All @@ -14,6 +15,11 @@ const isOpenHandsMessage = (event: unknown): event is OpenHandsParsedEvent =>
"message" in event &&
"timestamp" in event;

const isAgentStateChangeObservation = (
event: OpenHandsParsedEvent,
): event is AgentStateChangeObservation =>
"observation" in event && event.observation === "agent_state_changed";

export enum WsClientProviderStatus {
CONNECTED,
DISCONNECTED,
Expand Down Expand Up @@ -68,7 +74,7 @@ export function WsClientProvider({
}

function handleMessage(event: Record<string, unknown>) {
if (isOpenHandsMessage(event)) {
if (isOpenHandsMessage(event) && !isAgentStateChangeObservation(event)) {
messageRateHandler.record(new Date().getTime());
}
setEvents((prevEvents) => [...prevEvents, event]);
Expand Down

0 comments on commit ff466d0

Please sign in to comment.