File size: 1,122 Bytes
bb17288 322e341 bb17288 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 | import os
import google.generativeai as genai
class BasicAgent:
def __init__(self):
api_key = os.getenv("GOOGLE_API_KEY")
if not api_key:
raise ValueError("GOOGLE_API_KEY environment variable is missing. Please ensure it is set in Hugging Face Secrets.")
genai.configure(api_key=api_key)
self.model = genai.GenerativeModel('gemini-2.5-flash')
self.sessions = {}
def generate_response(self, session_id, user_message):
if session_id not in self.sessions:
self.sessions[session_id] = []
history = self.sessions[session_id]
prompt = "Conversation History:\n"
for msg in history:
prompt += f"User: {msg['user']}\nAI: {msg['ai']}\n"
prompt += f"User: {user_message}\nAI:"
try:
response = self.model.generate_content(prompt)
ai_text = response.text
except Exception as e:
ai_text = f"Error generating response: {e}"
self.sessions[session_id].append({"user": user_message, "ai": ai_text})
return ai_text
|