Use new Messages method

Author: rileysdev

api/rbt/mcp/v1/stream.proto

@@ -42,11 +42,6 @@ service StreamMethods {
     option (rbt.v1alpha1.method).reader = {
     };
   }
-  
-  rpc GetStream(GetStreamRequest) returns (GetStreamResponse) {
-    option (rbt.v1alpha1.method).reader = {
-    };
-  }
 }
 
 ////////////////////////////////////////////////////////////////////////
@@ -99,9 +94,3 @@ message MessagesResponse {
 }
 
 ////////////////////////////////////////////////////////////////////////
-message GetStreamRequest {}
-
-message GetStreamResponse {
-  string json_messages = 1;
-}
-////////////////////////////////////////////////////////////////////////

reboot/inspect/src/App.tsx

@@ -51,12 +51,14 @@ function StreamSequenceDiagram({
 }: {
   selectedStreamId: string;
 }) {
-  const { useGetStream } = useStream({
+  const { useMessages } = useStream({
     id: selectedStreamId,
   });
-  const { response } = useGetStream();
+  const { response } = useMessages();
 
-  const messages = response ? JSON.parse(response.jsonMessages) : [];
+  const messages = response
+    ? response.messages.map((msg: any) => msg.toJson())
+    : [];
 
   return (
     <>

reboot/mcp/servicers/stream.py

@@ -8,13 +8,9 @@
     ReplayRequest,
     ReplayResponse,
     Stream,
-    GetStreamRequest,
-    GetStreamResponse,
 )
 from reboot.aio.auth.authorizers import allow
 from reboot.aio.contexts import ReaderContext, WriterContext
-from google.protobuf.json_format import MessageToDict
-import json
 
 
 class StreamServicer(Stream.Servicer):
@@ -82,14 +78,3 @@ async def Messages(
         request: Empty,
     ) -> MessagesResponse:
         return MessagesResponse(messages=self.state.messages)
-
-    async def GetStream(
-        self,
-        context: ReaderContext,
-        request: GetStreamRequest,
-    ) -> GetStreamResponse:
-        json_array = [
-            MessageToDict(message) for message in self.state.messages
-        ]
-
-        return GetStreamResponse(json_messages=json.dumps(json_array))