File size: 1,104 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 | import os
import google.generativeai as genai
from .utils.memory_manager import PostgresMemoryManager
from .utils.prompt_manager import PromptManager
class EnhancedAgent:
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.memory = PostgresMemoryManager()
self.prompt_manager = PromptManager()
def generate_response(self, session_id, user_message, context=""):
history = self.memory.get_history(session_id)
prompt = self.prompt_manager.format_prompt(user_message, context, history)
try:
response = self.model.generate_content(prompt)
ai_text = response.text
except Exception as e:
ai_text = f"Error generating response: {e}"
self.memory.add_message(session_id, user_message, ai_text)
return ai_text
|