response model

app.py

@@ -6,6 +6,7 @@ from pydantic_ai.models.groq import GroqModel
 import nest_asyncio
 import pdfplumber
 import os
+import presentation as customClass
 
 
 api_key = os.getenv("API_KEY")
@@ -21,7 +22,9 @@ def split_long_string(long_string, chunk_size=6000):
 
 
 async def ppt_content(data):
-    agent = Agent(model,system_prompt=(
+    agent = Agent(model,
+    result_type=customClass.PPT,
+    system_prompt=(
         "You are an expert in making power-point perssentation",
         "Create 6 sliders",
         "Each slide should be seperate"
@@ -48,6 +51,7 @@ async def ppt_content(data):
     print(result_2.all_messages_json())
     print(result_2.new_messages_json())
     print(result_2.data)
+    print(result_2)
 
 def ai_ppt(data):
     asyncio.run(ppt_content(data=data))

presentation.py

@@ -0,0 +1,21 @@
+from pydantic import BaseModel, Field
+from dataclasses import dataclass, field
+from typing import List, Dict,Optional
+
+
+
+@dataclass
+class Perssentation(BaseModel):
+    title:str = Field(description="Title of the slide")
+    subTitle: Optional[str] = Field(description="Subtitle only for 1st slide")
+    author:Optional[str] = Field(description="Subtitle only for 1st slide")
+    text:str = Field(description="Detail description")
+    bulletPoints:Optional[list] = Field(description="Bullet Points")
+    imageSuggestion:Optional[str] = Field(description="Image prompt that we can give to AI image generator")
+
+
+
+
+@dataclass
+class PPT(BaseModel):
+    per:list[Perssentation]