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