Spaces:
Sleeping
Sleeping
| import os | |
| from groq import Groq | |
| def get_client(): | |
| api_key = os.environ.get("GROQ_API_KEY") | |
| if not api_key: | |
| raise RuntimeError( | |
| "GROQ_API_KEY is not set. Add it in Hugging Face Spaces → Settings → Secrets." | |
| ) | |
| return Groq(api_key=api_key) | |
| def ask_ai(question: str) -> str: | |
| client = get_client() | |
| messages = [ | |
| {"role": "system", "content": "You are a helpful AI assistant."}, | |
| {"role": "user", "content": question} | |
| ] | |
| completion = client.chat.completions.create( | |
| model="llama-3.3-70b-versatile", | |
| messages=messages | |
| ) | |
| return completion.choices[0].message.content | |