| from azure.search.documents import SearchClient | |
| from azure.core.credentials import AzureKeyCredential | |
| from modules.config import search_client | |
| def search(user_query): | |
| search_results = search_client.search( | |
| search_text=user_query, | |
| highlight_fields="content", # Indicar o campo que deve ser realçado | |
| top=3 # Limitar o número de resultados retornados | |
| ) | |
| # Estruturar os resultados | |
| contents = [] | |
| sources = [] | |
| for result in search_results: | |
| document_name = result['metadata_storage_name'] | |
| formatted_name = document_name.replace("_", " ").replace(".pdf", "").upper() | |
| sources.append(formatted_name) | |
| # Pegar os highlights e unificar em uma string | |
| highlights = " ".join(result.get('@search.highlights', {}).get('content', [])) | |
| contents.append(f"Fonte: {formatted_name}\nTrecho destacado: {highlights}") | |
| # Combinar os resultados em uma única string | |
| highlighted_content = "\n\n".join(contents) | |
| return highlighted_content |