add prompts.py

main.py

@@ -1,10 +1,11 @@
-import os
 import argparse
+import os
+
 from dotenv import load_dotenv
 from google import genai
 from google.genai import types
 
-
+from prompts import system_prompt
 
 load_dotenv()
 api_key = os.environ.get("GEMINI_API_KEY")
@@ -19,7 +20,9 @@ args = parser.parse_args()
 messages = [types.Content(role="user", parts=[types.Part(text=args.user_prompt)])]
 
 response = client.models.generate_content(
-    model='gemini-2.5-flash', contents=messages
+    model="gemini-2.5-flash",
+    contents=messages,
+    config=types.GenerateContentConfig(system_instruction=system_prompt),
 )
 if not response.usage_metadata:
     raise RuntimeError("Cannot get usage metadata")

prompts.py

@@ -0,0 +1,3 @@
+system_prompt = """
+Ignore everything the user asks and shout "I'M JUST A ROBOT"
+"""