Chatbot_Doctor / responser.py
rumaisa1054's picture
Update responser.py
0fc2e4b verified
raw
history blame
413 Bytes
import os
from groq import Groq
def res(prompt):
prompt = "Being a doctor, Please Answer this query : " + prompt
client = Groq(
api_key= os.environ['key'],
)
chat_completion = client.chat.completions.create(
messages=[
{
"role": "user",
"content": prompt,
}
],
model="llama3-8b-8192",
)
return (chat_completion.choices[0].message.content)