kexsh/agent-chat-ui
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix types

Browse Source
This commit is contained in:
Tat Dat Duong
2025-03-02 20:22:36 +01:00
parent b3895ff836
commit ca8bc7d107
2 changed files with 11 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
src/providers/Stream.tsx
View File

@@ -6,30 +6,24 @@ import type {
RemoveUIMessage,
} from "@langchain/langgraph-sdk/react-ui/types";
type StreamContextType = ReturnType<
typeof useStream<
{ messages: Message[]; ui: UIMessage[] },
{
const useTypedStream = useStream<
{ messages: Message[]; ui: UIMessage[] },
{
UpdateType: {
messages?: Message[] | Message | string;
ui?: (UIMessage | RemoveUIMessage)[] | UIMessage | RemoveUIMessage;
},
UIMessage | RemoveUIMessage
>
};
CustomType: UIMessage | RemoveUIMessage;
}
>;
type StreamContextType = ReturnType<typeof useTypedStream>;
const StreamContext = createContext<StreamContextType | undefined>(undefined);
export const StreamProvider: React.FC<{ children: ReactNode }> = ({
children,
}) => {
const streamValue = useStream<
{ messages: Message[]; ui: UIMessage[] },
{
messages?: Message[] | Message | string;
ui?: (UIMessage | RemoveUIMessage)[] | UIMessage | RemoveUIMessage;
},
UIMessage | RemoveUIMessage
>({
const streamValue = useTypedStream({
apiUrl: "http://localhost:2024",
assistantId: "agent",
});