From 2269cca971a30c5e32e95544fc1ee6abe5d77e05 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Maria=20Vit=C3=B3ria?= Date: Tue, 10 Nov 2026 11:17:09 -0300 Subject: [PATCH] genai: indexar.py e testar_busca.py funcionam como arquivo e como modulo (-m) --- genai/src/indexar.py | 5 ++++- genai/src/testar_busca.py | 5 ++++- 2 files changed, 8 insertions(+), 2 deletions(-) diff --git a/genai/src/indexar.py b/genai/src/indexar.py index 70c5740..dbe96b9 100644 --- a/genai/src/indexar.py +++ b/genai/src/indexar.py @@ -17,7 +17,10 @@ from psycopg2.extras import execute_values from sentence_transformers import SentenceTransformer -from dividir import dividir_normas +try: # funciona como "python genai/src/indexar.py" e como "python -m genai.src.indexar" + from genai.src.dividir import dividir_normas +except ImportError: + from dividir import dividir_normas MODELO = "sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2" # multilíngue, 384 dimensões DIMENSAO = 384 diff --git a/genai/src/testar_busca.py b/genai/src/testar_busca.py index 174b324..34c9be5 100644 --- a/genai/src/testar_busca.py +++ b/genai/src/testar_busca.py @@ -7,7 +7,10 @@ """ from sentence_transformers import SentenceTransformer -from indexar import MODELO, TABELA, conectar +try: # funciona como arquivo e como módulo (-m) + from genai.src.indexar import MODELO, TABELA, conectar +except ImportError: + from indexar import MODELO, TABELA, conectar QUANTOS = 3