First commit

source/server_nf_items.py

@@ -0,0 +1,136 @@
+# -*- coding: utf-8 -*-
+import os
+import oracledb
+from mcp.server.fastmcp import FastMCP
+from product_search import BuscaProdutoSimilar
+
+buscador = BuscaProdutoSimilar()
+
+mcp = FastMCP("InvoiceItemResolver")
+# Configurações Oracle Wallet
+WALLET_PATH = "/WALLET_PATH/Wallet_oradb23ai"  # Altere conforme seu ambiente
+DB_ALIAS = "oradb23ai_high"                 # Alias definido no tnsnames.ora
+USERNAME = "USER"                        # Usuário do banco
+PASSWORD = "Password"                    # Senha do usuário
+os.environ["TNS_ADMIN"] = WALLET_PATH
+
+
+def executar_busca(query: str, params: dict = {}):
+    try:
+        connection = oracledb.connect(
+            user=USERNAME,
+            password=PASSWORD,
+            dsn=DB_ALIAS,
+            config_dir=WALLET_PATH,
+            wallet_location=WALLET_PATH,
+            wallet_password=PASSWORD
+        )
+        cursor = connection.cursor()
+        cursor.execute(query, params)
+        results = cursor.fetchall()
+        cursor.close()
+        connection.close()
+        return results
+    except Exception as e:
+        print(f"[ERRO] Consulta falhou: {e}")
+        return []
+
+def executar_busca_ean(termos_busca):
+    results = []
+
+    try:
+        connection = oracledb.connect(user=USERNAME, password=PASSWORD, dsn=DB_ALIAS, config_dir=WALLET_PATH, wallet_location=WALLET_PATH, wallet_password=PASSWORD)
+        cursor = connection.cursor()
+
+        query = """
+                SELECT * FROM TABLE(fn_busca_avancada(:1))
+                ORDER BY similaridade DESC \
+                """
+        cursor.execute(query, [termos_busca])
+
+        for row in cursor:
+            results.append({
+                "codigo": row[0],
+                "descricao": row[1],
+                "similaridade": row[2]
+            })
+
+        cursor.close()
+        connection.close()
+    except Exception as e:
+        return {"erro": str(e)}, 500
+
+    return results
+# --------------------- FERRAMENTAS MCP ---------------------
+@mcp.tool()
+def buscar_produto_vetorizado(descricao: str) -> dict:
+    """Busca produto por descrição usando embeddings"""
+    return buscador.buscar_produtos_similares(descricao)
+
+@mcp.tool()
+def resolve_ean(description: str) -> dict:
+    """
+    Resolve o código EAN do produto a partir da descrição
+    """
+
+    result = executar_busca_ean(description)
+
+    if isinstance(result, list) and result:
+        return {
+            "ean": result[0]["codigo"],
+            "descricao": result[0]["descricao"],
+            "similaridade": result[0]["similaridade"]
+        }
+    else:
+        return {"erro": "EAN não encontrado com os critérios fornecidos."}
+
+@mcp.tool()
+def buscar_notas_por_criterios(cliente: str = None, estado: str = None, preco: float = None, ean: str = None, margem: float = 0.05) -> list:
+    """
+    Busca notas fiscais de saída com base em cliente, estado, EAN e preço aproximado.
+    Permite que um ou mais campos sejam omitidos.
+    Enquanto não houver um EAN estabelecido, nao adianta usar este servico.
+    """
+    print("buscar_notas_por_criterios")
+
+    query = """
+            SELECT nf.numero_nf, nf.nome_cliente, nf.estado, nf.data_saida,
+                   inf.numero_item, inf.codigo_ean, inf.descricao_produto, inf.valor_unitario
+            FROM nota_fiscal nf
+                     JOIN item_nota_fiscal inf ON nf.numero_nf = inf.numero_nf
+            WHERE 1=1 
+            """
+
+    params = {}
+
+    if cliente:
+        query += " AND LOWER(nf.nome_cliente) LIKE LOWER(:cliente)"
+        params["cliente"] = f"%{cliente}%"
+    if estado:
+        query += " AND LOWER(nf.estado) = LOWER(:estado)"
+        params["estado"] = estado
+    if ean:
+        query += " AND inf.codigo_ean = :ean"
+        params["ean"] = ean
+    if preco is not None:
+        query += " AND inf.valor_unitario BETWEEN :preco_min AND :preco_max"
+        params["preco_min"] = preco * (1 - margem)
+        params["preco_max"] = preco * (1 + margem)
+
+    # Executa a consulta com os parâmetros nomeados
+    result = executar_busca(query, params)
+
+    return [
+        dict(zip(
+            ["numero_nota", "nome_cliente", "estado", "data_saida", "numero_item", "codigo_ean", "descricao_produto", "valor_unitario"],
+            row
+        ))
+        for row in result
+    ]
+
+
+# --------------------- EXECUÇÃO MCP ---------------------
+
+if __name__ == "__main__":
+    # Inicia o servidor MCP
+    mcp.run(transport="stdio")