[FEATURE] Create a New Chat Interface Component

[amrutha97]

Overview

We want to introduce an AI-powered chat interface that allows users to:

• Select a data source.
• Ask natural language questions about the data.
• Receive AI-generated responses.

This component should work similarly to ChatGPT

Adding a new component

https://docs.preswald.com/addnewcomponent

Changes Required

1. Create chat.py in preswald components.

2. Expose chat as a Preswald component:

   from preswald import chat
   chat(source="sample_csv")  # Queries 'sample_csv'
   chat()  # Defaults to all sources

3. Integrate with an open-source chat UI, such as Assistant UI. -- evaluate if this is a good option otherwise build our own

4. Ensure backend integration for handling chat queries (with duckdb)

Testing

• Run preswald tutorial and verify that users can ask questions and receive answers.
• Ensure responses are accurate

Update Documentation

• Document chat usage in docs, with examples and screenshots.
• Add an example to preswald/tutorial demonstrating how to use the chat interface.

[joshlavroff]

I'd like to work on this!

[amrutha97]

This should be helpful : https://docs.preswald.com/addnewcomponent

[Sami-07]

Hey @joshlavroff, what's your estimated timeline to finish this issue?

[Sami-07]

@amrutha97, how about using the Vercel's AI SDK?

[joshlavroff]

Hey @joshlavroff, what's your estimated timeline to finish this issue?

Hi! I'm expecting to PR within the next two days or so

[Sami-07]

@joshlavroff that's great! What AI sdk are you planning to incorporate?

[joshlavroff]

@Sami-07 I am currently building it using Assistant-UI

[Sami-07]

@joshlavroff I see. sounds good.

[Sami-07]

@joshlavroff I was going through your PR and I am not really sure why did you write code in Next.js despite Preswald code being a pure Vite React codebase?
Morever this codebase is in JavaScript and you have TypeScript in your PR.
or am I missing anything?
@amrutha97 request you to confirm this.

[joshlavroff]

@Sami-07 Apologies for not documenting this well. This PR was an audit of the Assistant-UI library using their starter code, which after review is not a good fit. The completed version of the feature will not include any TS or NextJS

[ShadowMasterAJ]

I am also trying to work on this issue and would like some clarification as to the usage.

1. the user passes the csv file string to chat component in hello.py`
2. I can use service.data_manager.get_df to retrieve the df from the source
3. The df can be passed as system context (after minor processing to convert to a multi-string) for an LLM API (ex. OpenAI)
4. Chat with it

Is that a correct and complete understanding? @amrutha97 @Sami-07

[amrutha97]

@ShadowMasterAJ - Any data source connected in preswald.toml should be selectable as a source in this chat component. It's basically a RAG chat where you can 'chat' with the selected data source

[ShadowMasterAJ]

I have a baseline version as shown in the image, which is working with OpenAI (using gpt3.5-turbo)

Are there any requirements for the UI, choice of LLM, or any other kind of consideration?