genai

Ars/objects.py

@@ -3,7 +3,7 @@ from pydantic import Field, model_validator, BaseModel
 from typing import List, Optional, Union,Type, TypeVar
 from bson import ObjectId
 import openai
-from google import genai
+from google.generativeai import genai
 from google.genai import types
 import os
 from dotenv import load_dotenv
@@ -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