File size: 1,819 Bytes
4402d23 | 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 49 50 51 52 53 | from .search import SearchEngine
class ShoppingAgent:
def __init__(self):
self.search_engine = SearchEngine()
self.search_engine.load()
def montar_resposta(self, query, resultados):
if not resultados:
return f'Não encontrei produtos relevantes para "{query}".'
nomes = [item["product_name"] for item in resultados[:3]]
if len(nomes) == 1:
return f'Encontrei um produto relevante para "{query}": {nomes[0]}.'
if len(nomes) == 2:
return f'Encontrei produtos relevantes para "{query}", com destaque para {nomes[0]} e {nomes[1]}.'
return (
f'Encontrei produtos relevantes para "{query}", com destaque para '
f'{nomes[0]}, {nomes[1]} e {nomes[2]}.'
)
def verificar_resposta(self, resposta, resultados):
if not resultados:
return resposta
nomes_resultados = [item["product_name"] for item in resultados]
resposta_limpa = resposta.lower()
mencoes_validas = any(nome.lower() in resposta_limpa for nome in nomes_resultados)
if mencoes_validas:
return resposta
top1 = resultados[0]["product_name"]
return f"{resposta} O item mais relevante encontrado foi {top1}."
def responder(self, query, top_k=5):
busca = self.search_engine.buscar(query, top_k=top_k)
resultados = busca["resultados"]
resposta_inicial = self.montar_resposta(query, resultados)
resposta_final = self.verificar_resposta(resposta_inicial, resultados)
return {
"query": query,
"categoria_inferida": busca["categoria_inferida"],
"answer": resposta_final,
"products": resultados,
} |