Final AI Agent Implementation

Author: SplinterSword

backend/EduAssistant_Agents/agent.py

@@ -8,7 +8,8 @@
     ),
     instruction=(
         "You are the Quizzes Agent responsible for generating MCQ questions with 4 options and answers of the content provided to you by the Manager Agent. Also you are to return this output back to the manager agent."
-    )
+    ),
+    output_key="quiz"
 )
 
 summarizer_agent = Agent(
@@ -27,31 +28,34 @@
 - Structure summaries logically with clear headings when appropriate
 
 When given content, analyze it thoroughly and provide a well-structured summary that captures the essential information in a clear, academic style."""
-    )
+    ),
+    output_key="summary"
 )
 
 reference_agent = Agent(
     name="Reference_Agent",
     model="gemini-2.0-flash",
     description=(
-        "This is an AI agent which is responsible for extracting relevant references or citations from the content provided to it."
+        "This is an AI agent which is responsible for giving relevant references or citations related to the content provided to it."
     ),
     instruction=(
-"""You are a Reference Extraction Agent specialized in identifying and extracting citations, references, and source materials from educational content.
+"""You are a Reference Agent specialized in identifying the content and giving relavent citations, references, and source materials from educational content.
 
 Your role:
-- Scan content for all types of references including:
+- Provide all types of references relavant to the content including:
   * Academic citations (APA, MLA, Chicago, etc.)
   * URLs and web links
+  * PDFs and other documents
+  * Images and other media
   * Book titles and authors
   * Journal articles and papers
   * Document titles and section references
-  * Footnotes and endnotes
 - Format extracted references clearly and organized
 - Include page numbers, dates, and other citation details when available
 
 Provide a comprehensive list of all references related to the content provided to you."""
-    )
+    ),
+    output_key="references"
 )
 
 flash_card_agent = Agent(
@@ -70,7 +74,8 @@
 - Cover all important topics from the provided content
 
 Create comprehensive flashcards that facilitate active recall and help students understand key concepts."""
-    )
+    ),
+    output_key="flashcards"
 )
 
 root_agent = Agent(
@@ -101,5 +106,5 @@
 
 When users ask for specific functions, delegate to the appropriate agent and provide the results in a clear, helpful format asked by the user."""
     ),
-    sub_agents=[quizzes_agent, reference_agent, flash_card_agent, summarizer_agent]
+    sub_agents=[quizzes_agent, reference_agent, flash_card_agent, summarizer_agent],
 )

frontend/Pipfile

@@ -0,0 +1,11 @@
+[[source]]
+url = "https://pypi.org/simple"
+verify_ssl = true
+name = "pypi"
+
+[packages]
+
+[dev-packages]
+
+[requires]
+python_version = "3.13"

frontend/src/ai/flows/generate-flashcards.ts

@@ -39,7 +39,7 @@ export async function generateFlashcards(input: GenerateFlashcardsInput): Promis
   const ADK_URL = process.env.NEXT_PUBLIC_AGENT_API_URL || "http://localhost:8000";
   try {
     const { user, signOut } = useAuth();  
-    const userId = user || 'u_123'; // TODO: Get from AuthContext
+    const userId = user?.email || 'u_123'; // TODO: Get from AuthContext
     const sessionId = localStorage.getItem('sessionId') || 's_' + Math.random().toString(36).substr(2, 9);
 
     const res = await fetch(`${ADK_URL}/run`, {

frontend/src/ai/flows/generate-quizzes.ts

@@ -44,7 +44,7 @@ export async function generateQuizzes(
   try {
     const { user, signOut } = useAuth();
 
-    const userId = user || 'u_123'; // TODO: Get from AuthContext
+    const userId = user?.email || 'u_123'; // TODO: Get from AuthContext
     const sessionId = localStorage.getItem('sessionId') || 's_' + Math.random().toString(36).substr(2, 9);
 
     const res = await fetch(`${ADK_URL}/run`, {

frontend/src/ai/flows/suggest-resources.ts

@@ -34,7 +34,7 @@ export async function suggestResources(input: SuggestResourcesInput): Promise<Su
   const ADK_URL = process.env.NEXT_PUBLIC_AGENT_API_URL || "http://localhost:8000";
   try {
     const { user, signOut } = useAuth();
-    const userId = user || 'u_123'; // TODO: Get from AuthContext
+    const userId = user?.email || 'u_123';
     const sessionId = localStorage.getItem('sessionId') || 's_' + Math.random().toString(36).substr(2, 9);
 
     const res = await fetch(`${ADK_URL}/run`, {
@@ -47,7 +47,7 @@ export async function suggestResources(input: SuggestResourcesInput): Promise<Su
         newMessage: {
           role: "user",
           parts: [{
-            text: `Suggest 3-5 relevant study resources for this content. For each resource, provide a title, URL, and a brief reason why it's relevant. Format your response as a JSON array of objects with 'title' (string), 'url' (string, must be a valid URL), and 'reason' (string) fields. Here's the content: ${input.courseContent}\n\nFormat your response like this: [{\"title\": \"...\", \"url\": \"https://...\", \"reason\": \"...\"}]`
+            text: `Suggest 3-5 relevant references for this content. For each resource, provide a title, URL, and a brief reason why it's relevant. Format your response as a JSON array of objects with 'title' (string), 'url' (string, must be a valid URL), and 'reason' (string) fields. Here's the content: ${input.courseContent}\n\nFormat your response like this: [{\"title\": \"...\", \"url\": \"https://...\", \"reason\": \"...\"}]`
           }]
         }
       })

frontend/src/app/dashboard/page.tsx

@@ -46,10 +46,9 @@ export default function DashboardPage() {
     },
   ];
 
-  const userID = user || 'u_123';
+  const userID = user?.email || 'u_123';
   const sessionId = localStorage.getItem('sessionId') || 's_' + Math.random().toString(36).substr(2, 9);
 
-
   const makeSession = async () => {
     const session = await fetch('http://localhost:8000/apps/EduAssistant_Agents/users/' + userID + '/sessions/' + sessionId, {
       method: 'POST',
@@ -68,7 +67,7 @@ export default function DashboardPage() {
       throw new Error('Failed to create session');
     }
 
-    const session_id = (await session.json()).session_id;
+    const session_id = sessionId;
     localStorage.setItem('session_id', session_id);
     console.log(session_id);
   }