| 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 | |