Update src/paper.py

src/paper.py

@@ -2,7 +2,7 @@ import os
 import json
 import logging
 from datetime import datetime, timedelta
-from langchain_google_genai import ChatGoogleGenerativeAI
+from langchain_groq import ChatGroq
 from langchain.schema import SystemMessage, HumanMessage
 
 logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
@@ -16,7 +16,7 @@ class Agent:
         self.goal = goal
         self.backstory = backstory
         self.personality = personality
-        self.tools = []  # Initialize with empty list for future tool integrations
+        self.tools = [] 
         self.llm = llm
 
 class Task:
@@ -29,14 +29,12 @@ class Task:
         self.expected_output = expected_output
         self.context = context or []
 
-google_api_key = os.getenv("GEMINI_API_KEY")  # 실제 Google API 키 사용
-if not google_api_key:
-    logging.error("GEMINI_API_KEY is not set in the environment variables.")
-llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", google_api_key=google_api_key)
+groq_api_key = os.getenv("GROQ_API_KEY")
+os.environ['GROQ_API_KEY'] = groq_api_key
+if not groq_api_key:
+    logging.error("GROQ_API_KEY is not set in the environment variables.")
+llm = ChatGroq(model="llama-3.3-70b-versatile",max_tokens = 4000)
 
-# -------------------------------------------------------------------------------
-# Define Academic Research Agents
-# -------------------------------------------------------------------------------
 literature_research_agent = Agent(
     role="Literature Research Agent",
     goal="Research and provide a comprehensive review of existing literature on the research topic.",
@@ -85,9 +83,6 @@ chatbot_agent = Agent(
     llm=llm,
 )
 
-# -------------------------------------------------------------------------------
-# Define Tasks for Academic Research and Writing
-# -------------------------------------------------------------------------------
 literature_research_task = Task(
     description="""Research academic literature on {topic} considering the keywords {keywords}.