File size: 1,781 Bytes
013767d |
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 |
import requests
from config import Config
class DB_Response_Generator:
def __init__(self, config: Config):
self.api_url = config.GROQ_API_URL
self.headers = {
"Content-Type": "application/json",
"Authorization": f"Bearer {config.GROQ_API_KEY}"
}
def generate_response(self, context: str, db_result:str, sqlcode:str) -> str:
"""Generate SQL query using Groq API"""
try:
data = {
"model": "llama-3.1-8b-instant",
"messages": [{
"role": "user",
"content": (f"question: {context}\n"
"heres the table column test_id,test_date,model_name,noise_level_db,cleaning_efficiency_percent, battery_duration_minutes INT,"
"area_covered_sqm, dust_collection_grams, operating_temperature_celsius ,maintenance_score, navigation_accuracy_percent ,software_version\n"
f"and heres the sql code: {sqlcode} "
"below is result of sql query\n"
f"{db_result}\n"
"make summary and use non technical sentence to explain it into user and use it to answer user question ")
}],
"temperature": 0.1
}
response = requests.post(self.api_url, json=data, headers=self.headers)
if response.status_code == 200:
response_text = response.json()['choices'][0]['message']['content']
return response_text
else:
raise Exception(f"Groq API error: {response.status_code}, {response.text}")
except Exception as e:
return str(e)
|