Update main.py

main.py

@@ -1,7 +1,6 @@
 import os
 from dotenv import load_dotenv
 from langchain_core.output_parsers import StrOutputParser
-from langchain_groq import ChatGroq
 from langchain_core.prompts import ChatPromptTemplate
 from typing import List
 from typing_extensions import TypedDict
@@ -14,6 +13,7 @@ from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel
 from database import create_table, return_title, return_exercise
 import logging
+from langchain_google_genai import ChatGoogleGenerativeAI
 
 create_table()
 
@@ -36,11 +36,10 @@ app.add_middleware(
 )
 
 load_dotenv()
-os.environ["GROQ_API_KEY"] = os.getenv('GROQ_API_KEY')
 os.environ["TOKENIZERS_PARALLELISM"] = "false"
 
-llm = ChatGroq(model="llama-3.1-8b-instant", temperature=0.5)
-#llm = ChatGroq(model="mixtral-8x7b-32768", temperature=0.5)
+llm = ChatGoogleGenerativeAI(model="gemini-1.5-flash", temperature=0.5)
+
 exercise = ""
 system = """
 Tu es un expert Python.