genai (#11)

- sa (c8cc055fba323d4e11f8c2d43107be722dd243ac)
- sa (34ccce2a90389fb016eec140b2b1de7fa820650d)
- Merge branch 'pr/10' into pr/11 (f9249af0f0efaa8afeaf99c6940c1ce9ed2b8dea)

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