| import google.generativeai as genai | |
| import os | |
| GEMINI_API_KEY = os.getenv("GEMINI_API_KEY") | |
| genai.configure(api_key=GEMINI_API_KEY) | |
| async def get_gemini_fix(error_message, code, mode="python"): | |
| prompt = ( | |
| f"I tried to run the following {mode} code/command and got this error.\n" | |
| f"Code/Command:\n{code}\n\n" | |
| f"Error:\n{error_message}\n\n" | |
| "Please provide a corrected version of the code/command only. Do not explain, just output the fixed code/command." | |
| ) | |
| try: | |
| model = genai.GenerativeModel("gemini-2.5-flash") | |
| response = model.generate_content(prompt) | |
| return response.text.strip() | |
| except Exception as e: | |
| return None |