Scaffold web app
This commit is contained in:
34
src/app/(app)/ai.tsx
Normal file
34
src/app/(app)/ai.tsx
Normal file
@@ -0,0 +1,34 @@
|
||||
'use server'
|
||||
|
||||
import { executeTodoAgent } from '~/agents/todo'
|
||||
import env from '~/env'
|
||||
import { publish } from '~/events/server'
|
||||
|
||||
export async function sendMessage({ userId, message }: { userId: string; message: string }) {
|
||||
const { response } = await executeTodoAgent({
|
||||
messages: [{ content: message, role: 'user' }],
|
||||
onEvent: async events => {
|
||||
switch (events.type) {
|
||||
case 'assistant_message': {
|
||||
await publish({
|
||||
channel: userId,
|
||||
eventType: events.type,
|
||||
payload: events
|
||||
})
|
||||
break
|
||||
}
|
||||
case 'function_call': {
|
||||
await publish({
|
||||
channel: userId,
|
||||
eventType: events.name,
|
||||
payload: events
|
||||
})
|
||||
break
|
||||
}
|
||||
}
|
||||
},
|
||||
openAIKey: env.OPENAI_API_KEY
|
||||
})
|
||||
|
||||
console.log(response)
|
||||
}
|
||||
90
src/app/(app)/chat.tsx
Normal file
90
src/app/(app)/chat.tsx
Normal file
@@ -0,0 +1,90 @@
|
||||
'use client'
|
||||
|
||||
import { useState } from 'react'
|
||||
import type { TodoAgentResponseEvent, TodoAgentToolEvent } from '~/agents/todo'
|
||||
import { useSession } from '~/auth/client'
|
||||
import { ChatBox } from '~/components/ChatBox'
|
||||
import { AssistantMessage, ToolMessage, UserMessage } from '~/components/Message'
|
||||
|
||||
import { useEvents } from '~/events/client'
|
||||
import { sendMessage } from './ai'
|
||||
|
||||
type Message =
|
||||
| TodoAgentToolEvent
|
||||
| TodoAgentResponseEvent
|
||||
| {
|
||||
id: string
|
||||
type: 'user_message'
|
||||
message: string
|
||||
}
|
||||
|
||||
function MessageSwitch({ message }: { message: Message }) {
|
||||
switch (message.type) {
|
||||
case 'user_message': {
|
||||
return <UserMessage>{message.message}</UserMessage>
|
||||
}
|
||||
|
||||
case 'assistant_message': {
|
||||
return <AssistantMessage>{message.message.response}</AssistantMessage>
|
||||
}
|
||||
case 'function_call': {
|
||||
return (
|
||||
<ToolMessage
|
||||
name={message.name}
|
||||
args={JSON.stringify(message.arguments)}
|
||||
result={JSON.stringify(message.result)}
|
||||
/>
|
||||
)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
function ChatMessages({
|
||||
userId,
|
||||
messages,
|
||||
addMessage
|
||||
}: {
|
||||
userId: string
|
||||
messages: Message[]
|
||||
addMessage: (message: Message) => void
|
||||
}) {
|
||||
useEvents({
|
||||
id: userId,
|
||||
on: {
|
||||
assistant_message: addMessage
|
||||
}
|
||||
})
|
||||
|
||||
return (
|
||||
<div className="pb-16">
|
||||
{messages.map(message => (
|
||||
<MessageSwitch key={message.id} message={message} />
|
||||
))}
|
||||
</div>
|
||||
)
|
||||
}
|
||||
|
||||
export default function Chat() {
|
||||
const { userId } = useSession()
|
||||
const [messages, setMessages] = useState<Message[]>([])
|
||||
|
||||
function addMessage(message: Message) {
|
||||
setMessages(prev => [...prev, message])
|
||||
}
|
||||
|
||||
function handleSubmit(message: string) {
|
||||
addMessage({
|
||||
id: Date.now().toString(),
|
||||
message,
|
||||
type: 'user_message'
|
||||
})
|
||||
sendMessage({ message, userId })
|
||||
}
|
||||
|
||||
return (
|
||||
<div className="w-full">
|
||||
<ChatMessages userId={userId} messages={messages} addMessage={addMessage} />
|
||||
<ChatBox placeholder="What is my todo list?" submit={handleSubmit} />
|
||||
</div>
|
||||
)
|
||||
}
|
||||
17
src/app/(app)/layout.tsx
Normal file
17
src/app/(app)/layout.tsx
Normal file
@@ -0,0 +1,17 @@
|
||||
import { getSession } from '~/auth/actions'
|
||||
import { ClientAuthProvider } from '~/auth/client'
|
||||
import { Nav } from '~/components/Nav'
|
||||
|
||||
export default async function AppLayout({ children }: { children: React.ReactNode }) {
|
||||
return (
|
||||
// @ts-expect-error: Auth Package Bug
|
||||
<ClientAuthProvider session={await getSession({ redirectUnauthorized: '/signin' })}>
|
||||
<div className="flex min-h-screen flex-col">
|
||||
<Nav />
|
||||
<div className="mx-auto flex w-full max-w-4xl flex-1 flex-col items-center p-10">
|
||||
<div className="flex w-full flex-col items-center">{children}</div>
|
||||
</div>
|
||||
</div>
|
||||
</ClientAuthProvider>
|
||||
)
|
||||
}
|
||||
5
src/app/(app)/page.tsx
Normal file
5
src/app/(app)/page.tsx
Normal file
@@ -0,0 +1,5 @@
|
||||
import Chat from './chat'
|
||||
|
||||
export default function Page() {
|
||||
return <Chat />
|
||||
}
|
||||
Reference in New Issue
Block a user