Llm chat core

src/lib/components/session/HostView.svelte

@@ -111,30 +111,20 @@
 		try {
 			// 為每個群組的每個參與者創建對話
 			for (const group of $groups) {
-				for (const participant of group.participants) {
-					const response = await fetch(
-						`/api/session/${$page.params.id}/group/${group.id}/conversations`,
-						{
-							method: 'POST',
-							headers: {
-								'Content-Type': 'application/json'
-							},
-							body: JSON.stringify({
-								task: $session?.task || '',
-								subtasks: $session?.subtasks || [],
-								resources: $session?.resources.map((r) => r.content) || [],
-								participant: participant
-							})
+				const response = await fetch(
+					`/api/session/${$page.params.id}/group/${group.id}/conversations`,
+					{
+						method: 'GET',
+						headers: {
+							'Content-Type': 'application/json'
 						}
-					);
-
-					if (!response.ok) {
-						const data = await response.json();
-						notifications.error(
-							data.error || `無法為參與者 ${participantNames.get(participant)} 創建對話`
-						);
-						return;
 					}
+				);
+
+				if (!response.ok) {
+					const data = await response.json();
+					notifications.error(data.error || `無法為群組 ${group.number} 的參與者創建對話`);
+					return;
 				}
 			}
 

src/lib/server/llm.ts

@@ -1,4 +1,5 @@
 import { env } from '$env/dynamic/private';
+import type { Resource } from '$lib/schema/resource';
 import type { LLMChatMessage, StudentSpeak } from '$lib/utils/types';
 import fs from 'fs/promises';
 import { OpenAI } from 'openai';
@@ -134,10 +135,7 @@ export async function chatWithLLMByDocs(
 	history: LLMChatMessage[],
 	task: string,
 	subtasks: string[],
-	resources: {
-		name: string;
-		content: string;
-	}[],
+	resources: Resource[],
 	temperature = 0.7
 ): Promise<{ success: boolean; message: string; subtask_completed: boolean[]; error?: string }> {
 	console.log('Starting chatWithLLMByDocs:', {

src/lib/utils/firestore.ts

@@ -1,15 +1,18 @@
 import type { Conversation } from '$lib/schema/conversation';
 import type { Group } from '$lib/schema/group';
+import type { Session } from '$lib/schema/session';
 import { adminDb } from '$lib/server/firebase';
 import { error } from '@sveltejs/kit';
+import type { LLMChatMessage } from './types';
 
 export async function createConversation(
 	id: string,
 	group_number: string,
 	userId: string,
 	task: string,
 	subtasks: string[],
-	resources: string[]
+	history: LLMChatMessage[],
+	resources: { name: string; content: string }[]
 ) {
 	const conversationRef = adminDb
 		.collection('sessions')
@@ -24,7 +27,7 @@ export async function createConversation(
 		task: task,
 		subtasks: subtasks,
 		resources: resources,
-		history: [],
+		history: history,
 		subtaskCompleted: new Array(subtasks.length).fill(false)
 	});
 
@@ -40,6 +43,15 @@ export async function getConversationRef(id: string, group_number: string, conv_
 		.doc(conv_id);
 }
 
+export function getConversationsRef(id: string, group_number: string) {
+	return adminDb
+		.collection('session')
+		.doc(id)
+		.collection('groups')
+		.doc(group_number)
+		.collection('conversations');
+}
+
 export async function getConversationData(
 	conversation_ref: FirebaseFirestore.DocumentReference<FirebaseFirestore.DocumentData>
 ): Promise<Conversation> {
@@ -51,15 +63,6 @@ export async function getConversationData(
 	return conversation.data() as Conversation;
 }
 
-export function getConversationsRef(id: string, group_number: string) {
-	return adminDb
-		.collection('session')
-		.doc(id)
-		.collection('groups')
-		.doc(group_number)
-		.collection('conversations');
-}
-
 export async function getConversationsData(
 	conversations_ref: FirebaseFirestore.CollectionReference<
 		FirebaseFirestore.DocumentData,
@@ -78,6 +81,10 @@ export function getGroupRef(id: string, group_number: string) {
 	return adminDb.collection('sessions').doc(id).collection('groups').doc(group_number);
 }
 
+export function getGroupsRef(id: string) {
+	return adminDb.collection('sessions').doc(id).collection('groups');
+}
+
 export async function getGroupData(
 	group_ref: FirebaseFirestore.DocumentReference<FirebaseFirestore.DocumentData>
 ): Promise<Group> {
@@ -88,3 +95,29 @@ export async function getGroupData(
 
 	return group.data() as Group;
 }
+
+export async function getGroupsData(
+	groups_ref: FirebaseFirestore.CollectionReference<FirebaseFirestore.DocumentData>
+): Promise<Group[]> {
+	const groups = await groups_ref.get();
+	if (groups.empty) {
+		throw error(404, 'Groups not found');
+	}
+
+	return groups.docs.map((doc) => doc.data() as Group);
+}
+
+export function getSessionRef(id: string) {
+	return adminDb.collection('sessions').doc(id);
+}
+
+export async function getSessionData(
+	session_ref: FirebaseFirestore.DocumentReference<FirebaseFirestore.DocumentData>
+): Promise<Session> {
+	const session = await session_ref.get();
+	if (!session.exists) {
+		throw error(404, 'Session not found');
+	}
+
+	return session.data() as Session;
+}

src/routes/api/session/[id]/group/[group_number]/conversations/+server.ts

@@ -1,17 +1,16 @@
-import { createConversation } from '$lib/utils/firestore';
+import { chatWithLLMByDocs } from '$lib/server/llm';
+import {
+	createConversation,
+	getGroupData,
+	getGroupRef,
+	getSessionData,
+	getSessionRef
+} from '$lib/utils/firestore';
+import type { LLMChatMessage } from '$lib/utils/types';
 import type { RequestHandler } from '@sveltejs/kit';
 import { error, json, redirect } from '@sveltejs/kit';
-import { z } from 'zod';
 
-// 更新請求數據格式
-const requestDataFormat = z.object({
-	task: z.string(),
-	subtasks: z.array(z.string()),
-	resources: z.array(z.string()),
-	participant: z.string()
-});
-
-export const POST: RequestHandler = async ({ request, params, locals }) => {
+export const GET: RequestHandler = async ({ params, locals }) => {
 	try {
 		if (!locals.user) {
 			redirect(303, '/login');
@@ -21,37 +20,28 @@ export const POST: RequestHandler = async ({ request, params, locals }) => {
 			throw error(400, 'Missing parameters');
 		}
 
-		const { task, subtasks, resources, participant } = await getRequestData(request);
+		const session_ref = getSessionRef(id);
+		const { task, subtasks, resources } = await getSessionData(session_ref);
+		const group_ref = getGroupRef(id, group_number);
+		const group_data = await getGroupData(group_ref);
+
+		const intro = await chatWithLLMByDocs([], task, subtasks, resources);
+		if (!intro) {
+			throw error(500, 'Error generating intro message');
+		}
+		const history: LLMChatMessage[] = [{ role: 'assistant', content: intro.message }];
 
-		const conv_id = await createConversation(
-			id,
-			group_number,
-			participant,
-			task,
-			subtasks,
-			resources
+		await Promise.all(
+			group_data.participants.map(async (participant) =>
+				createConversation(id, group_number, participant, task, subtasks, history, resources)
+			)
 		);
 
 		return json({
-			success: true,
-			conversationId: conv_id
+			success: true
 		});
 	} catch (error) {
 		console.error('Error creating conversation:', error);
 		return json({ error: 'Internal Server Error' }, { status: 500 });
 	}
 };
-
-async function getRequestData(request: Request): Promise<z.infer<typeof requestDataFormat>> {
-	const data = await request.json();
-	const result = requestDataFormat.parse(data);
-	if (!result.task || !result.subtasks || !result.resources || !result.participant) {
-		throw error(
-			400,
-			`Missing parameters: ${!result.task ? 'task ' : ''}${!result.subtasks ? 'subtasks ' : ''}${
-				!result.resources ? 'resources ' : ''
-			}${!result.participant ? 'participant' : ''}`.trim()
-		);
-	}
-	return result;
-}

src/routes/api/session/[id]/group/[group_number]/conversations/[conv_id]/chat/+server.ts

@@ -28,7 +28,7 @@ export const POST: RequestHandler = async ({ request, params, locals }) => {
 
 		const conversation_ref = await getConversationRef(id, group_number, conv_id);
 		console.log('Retrieved conversation reference');
-		const { userId, task, subtasks, resources, history } =
+		const { userId, task, subtasks, resources, history, subtaskCompleted } =
 			await getConversationData(conversation_ref);
 		console.log('Retrieved conversation data', { userId, task, subtasksCount: subtasks.length });
 
@@ -83,7 +83,9 @@ export const POST: RequestHandler = async ({ request, params, locals }) => {
 					content: response.message
 				}
 			],
-			subtaskCompleted: response.subtask_completed
+			subtaskCompleted: subtaskCompleted.map(
+				(completed, index) => completed || response.subtask_completed[index]
+			)
 		});
 
 		return json({ success: true, message: response.message });