/ 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)}")