/ ai_assistant.py
ai_assistant.py
1 #!/usr/bin/env python3 2 import json, urllib.request, sys 3 4 API_KEY = "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" 5 6 def ask(prompt): 7 url = "https://api.groq.com/openai/v1/chat/completions" 8 data = json.dumps({ 9 "model": "llama-3.3-70b-versatile", 10 "messages": [{"role": "user", "content": prompt}], 11 "max_tokens": 1000 12 }).encode() 13 14 req = urllib.request.Request(url, data=data, headers={ 15 'Authorization': f'Bearer {API_KEY}', 16 'Content-Type': 'application/json' 17 }) 18 19 try: 20 with urllib.request.urlopen(req) 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("Ask: ") 27 print("\n" + "="*60) 28 print(ask(question)) 29 print("="*60)