Skip to content

Commit

Permalink
0.0.4
Browse files Browse the repository at this point in the history
  • Loading branch information
gacerioni committed Jul 31, 2024
1 parent 2f4fe52 commit 093b561
Showing 1 changed file with 9 additions and 8 deletions.
17 changes: 9 additions & 8 deletions app.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,10 +10,7 @@
# Carregar variáveis de ambiente
load_dotenv()

# Constantes
REDIS_URL = os.getenv('GABS_REDIS_URL', "redis://localhost:6379")
OPENAI_API_KEY = os.getenv('GABS_OPENAI_API_KEY', 'sk_nadanadanada')

# Constants
REDIS_DATABASE_HOST = os.getenv('REDIS_DATABASE_HOST', '34.198.15.233')
REDIS_DATABASE_PORT = os.getenv('REDIS_DATABASE_PORT', 19419)
REDIS_DATABASE_PASSWORD = os.getenv('REDIS_DATABASE_PASSWORD', 'secret42')
Expand All @@ -25,12 +22,11 @@
ORACLE_DATABASE_USER = os.getenv('ORACLE_DATABASE_USER', 'c##dbzuser')
ORACLE_DATABASE_PASSWORD = os.getenv('ORACLE_DATABASE_PASSWORD', 'dbz')


app = Flask(__name__)

# Initialize Oracle client with the updated library directory
#cx_Oracle.init_oracle_client(lib_dir="/root/something/instantclient_19_24")
cx_Oracle.init_oracle_client(lib_dir="/Users/gabriel.cerioni/instantclient")
cx_Oracle.init_oracle_client(lib_dir="/root/something/instantclient_19_24")
#cx_Oracle.init_oracle_client(lib_dir="/Users/gabriel.cerioni/instantclient")

# Oracle connection setup
oracle_dsn = cx_Oracle.makedsn(ORACLE_DATABASE_HOST, ORACLE_DATABASE_PORT, service_name=ORACLE_DATABASE_SERVICE_NAME)
Expand All @@ -42,6 +38,7 @@
redis_search_client = Client(REDIS_INDEX_NAME, conn=redis_client)
index_prefix = REDIS_INDEX_PREFIX


def create_redis_index():
"""
Creates the index in Redis if it doesn't exist.
Expand All @@ -54,8 +51,10 @@ def create_redis_index():
definition = IndexDefinition(prefix=[index_prefix], index_type=IndexType.JSON)
redis_search_client.create_index(schema, definition=definition)


create_redis_index()


@app.route('/query', methods=['GET'])
def query():
search_term = request.args.get('search_term', '').upper() # Convert search term to upper case
Expand All @@ -73,7 +72,9 @@ def query():
oracle_latency_ms = (time.time() - oracle_start_time) * 1000 # Convert to milliseconds

# Redis search remains unchanged
redis_query = Query(f'@NM_LOGR:{search_term}').return_fields("$.ID_ESTABELECIMENTO_SAUDE_PK", "$.NM_RAZ_SOC", "$.NM_FANTS", "$.NM_LOGR", "$.NM_NUMERO", "$.NM_BAIRRO", "$.CD_CEP")
redis_query = Query(f'@NM_LOGR:{search_term}').return_fields("$.ID_ESTABELECIMENTO_SAUDE_PK", "$.NM_RAZ_SOC",
"$.NM_FANTS", "$.NM_LOGR", "$.NM_NUMERO",
"$.NM_BAIRRO", "$.CD_CEP")
redis_start_time = time.time()
redis_results = redis_search_client.search(redis_query)
redis_latency_ms = (time.time() - redis_start_time) * 1000 # Convert to milliseconds
Expand Down

0 comments on commit 093b561

Please sign in to comment.