genai (#12)

- ars (14e7a185168e24f2f273d317578eaebe8181fa35)

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.generativeai import genai
+from google 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,output_schema: Type[T]) -> str:
+    def _openai_chat(self, prompt: str) -> 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.generativeai import genai
+from google import genai
 from typing import Optional,List
 from pydantic import BaseModel
 import re

requirements.txt

@@ -29,7 +29,7 @@ einops
 google-generativeai
 google-auth
 google-api-python-client
-
+google-genai
 
 # Authentication
 bcrypt
@@ -48,6 +48,9 @@ python-multipart
 pydantic-settings
 pydantic-extra-types
 
+# Windows only (skip on Linux/mac)
+# Remove pywin32==308, use the latest
+
 # Optional tools
 tqdm
 beautifulsoup4