/ ask_ai.py
ask_ai.py
1 #!/usr/bin/env python3 2 import json, urllib.request, sys 3 4 def ask(prompt): 5 """Ask AI using Groq""" 6 url = "https://api.groq.com/openai/v1/chat/completions" 7 key = "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" 8 9 data = json.dumps({ 10 "model": "llama-3.3-70b-versatile", 11 "messages": [{"role": "user", "content": prompt}], 12 "max_tokens": 800 13 }).encode('utf-8') 14 15 try: 16 req = urllib.request.Request(url, data=data, headers={ 17 'Authorization': f'Bearer {key}', 18 'Content-Type': 'application/json' 19 }) 20 with urllib.request.urlopen(req, timeout=20) as r: 21 return json.loads(r.read().decode())['choices'][0]['message']['content'] 22 except Exception as e: 23 return f"Error: {e}" 24 25 if __name__ == "__main__": 26 question = " ".join(sys.argv[1:]) if len(sys.argv) > 1 else input("Question: ") 27 print(f"\n🤖 Answer:\n{ask(question)}")