Spaces:
Sleeping
Sleeping
File size: 522 Bytes
0171c4b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
# groq_llm.py
import openai
import os
openai.api_key = os.getenv("GROQ_API_KEY")
def get_groq_response(user_input, system_prompt, language="English"):
messages = [
{"role": "system", "content": f"You are an AI assistant. Respond in {language}. {system_prompt}"},
{"role": "user", "content": user_input}
]
response = openai.ChatCompletion.create(
model="llama3-70b-8192",
messages=messages,
temperature=0.7
)
return response.choices[0].message.content.strip()
|