Merge branch 'pr/10' into pr/11

Ars/objects.py

@@ -41,7 +41,7 @@ class AIWrapper:
         elif self.provider == 'gemini':
             return self._gemini_chat(prompt,output_schema=output_schema)
 
-    def _openai_chat(self, prompt: str) -> str:
+    def _openai_chat(self, prompt: str,output_schema: Type[T]) -> str:
         response = openai.ChatCompletion.create(
             model="gpt-4",
             messages=[

controller/imports.py

@@ -12,7 +12,7 @@ import fitz
 from gamification.routes import gamification
 from controller.scraper import scrapeCourse
 import asyncio
-from google import genai 
+from google.generativeai import genai
 from typing import Optional,List
 from pydantic import BaseModel
 import re