fix: TextContentPartProvider should support contentPartRuntime.getSta…

…te() (#1187)

.changeset/tough-apes-vanish.md

@@ -0,0 +1,5 @@
+---
+"@assistant-ui/react": patch
+---
+
+fix: TextContentPartProvider should support contentPartRuntime.getState()

packages/react/src/api/ContentPartRuntime.ts

@@ -41,24 +41,31 @@ export class ContentPartRuntimeImpl implements ContentPartRuntime {
 
   constructor(
     private contentBinding: ContentPartSnapshotBinding,
-    private messageApi: MessageStateBinding,
-    private threadApi: ThreadRuntimeCoreBinding,
+    private messageApi?: MessageStateBinding,
+    private threadApi?: ThreadRuntimeCoreBinding,
   ) {}
 
   public getState() {
+    console.log(this.contentBinding);
     return this.contentBinding.getState();
   }
 
   public addToolResult(result: any) {
-    const message = this.messageApi.getState();
-    if (!message) throw new Error("Message is not available");
-
     const state = this.contentBinding.getState();
     if (!state) throw new Error("Content part is not available");
 
     if (state.type !== "tool-call")
       throw new Error("Tried to add tool result to non-tool content part");
 
+    if (!this.messageApi)
+      throw new Error(
+        "Message API is not available. This is likely a bug in assistant-ui.",
+      );
+    if (!this.threadApi) throw new Error("Thread API is not available");
+
+    const message = this.messageApi.getState();
+    if (!message) throw new Error("Message is not available");
+
     const toolName = state.toolName;
     const toolCallId = state.toolCallId;
 

packages/react/src/context/providers/TextContentPartProvider.tsx

@@ -35,16 +35,26 @@ export const TextContentPartProvider: FC<TextContentPartProvider.Props> = ({
   isRunning,
 }) => {
   const [context] = useState<ContentPartContextValue>(() => {
-    const useContentPartRuntime = create(
-      // TODO
-      () => new ContentPartRuntimeImpl(null as any, null as any, null as any),
-    );
     const useContentPart = create<ContentPartState>(() => ({
       status: isRunning ? RUNNING_STATUS : COMPLETE_STATUS,
       type: "text",
       text,
     }));
 
+    const useContentPartRuntime = create(
+      () =>
+        new ContentPartRuntimeImpl({
+          path: {
+            ref: "text",
+            threadSelector: { type: "main" },
+            messageSelector: { type: "messageId", messageId: "" },
+            contentPartSelector: { type: "index", index: 0 },
+          },
+          getState: useContentPart.getState,
+          subscribe: useContentPart.subscribe,
+        }),
+    );
+
     return { useContentPartRuntime, useContentPart };
   });