File size: 1,555 Bytes
9322f35 f58e385 9322f35 fe952bc 9322f35 9eef858 9322f35 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 |
from groq import Groq
import os
from text import mytext
from dotenv import load_dotenv
load_dotenv()
class Response:
def __init__(self):
self.client = Groq(api_key = os.getenv("GROQ_API"))
def chatbot(self,query:str) -> str:
try:
res = self.client.chat.completions.create(
messages=[
{"role":"system",
"content":f"You are a Question answer chatbot. You have to understand the given content based on that provide answer. If don't know tell unable to get details. only give answers like your are lalit. Provide response required for user question do not provide additional context.",
"role": "user",
"content": f"Content : {mytext}\n\n Question : {query}",
}
],
model="llama-3.3-70b-versatile",
)
return res.choices[0].message.content
except Exception as e:
print(e)
def other_chatbot(self,query:str,role:str) -> str:
try:
res = self.client.chat.completions.create(
messages=[
{"role":"system",
"content":role,
"role": "user",
"content": f"User question : {query}",
}
],
model="llama-3.3-70b-versatile",
)
return res.choices[0].message.content
except Exception as e:
print(e)
if __name__ == "__main__":
res = Response().chatbot(query="hi")
print(res) |