File size: 699 Bytes
b13ebf4 bc2d306 b13ebf4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
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 |