masadonline commited on
Commit
bcb887f
·
verified ·
1 Parent(s): 7820d77

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -50,7 +50,10 @@ def generate_answer_with_groq(question, context):
50
  groq_client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
51
  response = groq_client.chat.completions.create(
52
  model=model_name,
53
- messages=[{"role": "user", "content": prompt}]
 
 
 
54
  )
55
  return response.choices[0].message.content
56
  except Exception as e:
 
50
  groq_client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
51
  response = groq_client.chat.completions.create(
52
  model=model_name,
53
+ messages=[
54
+ {"role": "system", "content": "You are an SME expert."},
55
+ {"role": "user", "content": prompt},
56
+ ]
57
  )
58
  return response.choices[0].message.content
59
  except Exception as e: