@lobehub/lobehub 2.0.0-next.28 → 2.0.0-next.29

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -764,7 +764,7 @@ describe('thread action', () => {
764
764
 
765
765
  const createMessageSpy = vi
766
766
  .spyOn(result.current, 'internal_createMessage')
767
- .mockResolvedValue('new-msg-id');
767
+ .mockResolvedValue({ id: 'new-msg-id', messages: [] });
768
768
  const coreProcessSpy = vi
769
769
  .spyOn(result.current, 'internal_coreProcessMessage')
770
770
  .mockResolvedValue();
@@ -803,7 +803,10 @@ describe('thread action', () => {
803
803
  });
804
804
  });
805
805
 
806
- vi.spyOn(result.current, 'internal_createMessage').mockResolvedValue('new-msg-id');
806
+ vi.spyOn(result.current, 'internal_createMessage').mockResolvedValue({
807
+ id: 'new-msg-id',
808
+ messages: [],
809
+ });
807
810
  vi.spyOn(result.current, 'internal_coreProcessMessage').mockResolvedValue();
808
811
  vi.spyOn(result.current, 'internal_createTmpMessage').mockReturnValue('temp-msg-id');
809
812
  vi.spyOn(result.current, 'internal_toggleMessageLoading');
@@ -833,7 +836,10 @@ describe('thread action', () => {
833
836
  });
834
837
  });
835
838
 
836
- vi.spyOn(result.current, 'internal_createMessage').mockResolvedValue('new-msg-id');
839
+ vi.spyOn(result.current, 'internal_createMessage').mockResolvedValue({
840
+ id: 'new-msg-id',
841
+ messages: [],
842
+ });
837
843
  vi.spyOn(result.current, 'internal_coreProcessMessage').mockResolvedValue();
838
844
  vi.spyOn(result.current, 'internal_createTmpMessage').mockReturnValue('temp-msg-id');
839
845
  vi.spyOn(result.current, 'internal_toggleMessageLoading');
@@ -855,7 +861,10 @@ describe('thread action', () => {
855
861
  });
856
862
 
857
863
  vi.spyOn(result.current, 'internal_shouldUseRAG').mockReturnValue(true);
858
- vi.spyOn(result.current, 'internal_createMessage').mockResolvedValue('new-msg-id');
864
+ vi.spyOn(result.current, 'internal_createMessage').mockResolvedValue({
865
+ id: 'new-msg-id',
866
+ messages: [],
867
+ });
859
868
  vi.spyOn(result.current, 'internal_createTmpMessage').mockReturnValue('temp-msg-id');
860
869
  vi.spyOn(result.current, 'internal_toggleMessageLoading');
861
870
 
@@ -158,7 +158,9 @@ export const chatThreadMessage: StateCreator<
158
158
  tempMessageId = get().internal_createTmpMessage(newMessage);
159
159
  get().internal_toggleMessageLoading(true, tempMessageId);
160
160
 
161
- parentMessageId = await get().internal_createMessage(newMessage, { tempMessageId });
161
+ const result = await get().internal_createMessage(newMessage, { tempMessageId });
162
+ if (!result) return;
163
+ parentMessageId = result.id;
162
164
  }
163
165
 
164
166
  get().internal_toggleMessageLoading(false, tempMessageId);