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