biatoolkit 1.0.2__tar.gz → 1.0.3__tar.gz

{biatoolkit-1.0.2 → biatoolkit-1.0.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: biatoolkit
-Version: 1.0.2
+Version: 1.0.3
 Summary: Biblioteca para desenvolvedores que utilizam o BiaAgentBuilder
 Author: Bia Platform Team
 Author-email: data.platform@sankhya.com.br

biatoolkit-1.0.3/biatoolkit/basic_client.py

@@ -0,0 +1,63 @@
+from mcp import ClientSession
+from mcp.client.streamable_http import streamablehttp_client
+
+class BiaClient:
+    """
+    Cliente básico para interação com o MCP (Middleware de Comunicação de Processos).
+    Permite listar ferramentas disponíveis e executar ferramentas específicas via HTTP.
+    """
+
+    def __init__(self, url: str = "http://0.0.0.0:8000/mcp", headers: dict = None):
+        """
+        Inicializa o BasicClient.
+
+        Args:
+            url (str): URL base do MCP. Se não terminar com '/mcp', será adicionado automaticamente.
+            headers (dict): Cabeçalhos HTTP opcionais para requisições.
+        """
+        sufix = "/mcp"
+        # Garante que a URL termine com '/mcp'
+        self.url = url if url.endswith(sufix) else f"{url}{sufix}"
+        self.headers = headers
+
+    async def list_tools(self) -> dict:
+        """
+        Lista as ferramentas disponíveis no MCP.
+
+        Returns:
+            dict: Dicionário contendo as ferramentas disponíveis.
+        """
+        # Cria um cliente HTTP streamable para comunicação assíncrona
+        async with streamablehttp_client(self.url, self.headers, timeout=120, terminate_on_close=False) as (
+            read_stream,
+            write_stream,
+            _,
+        ):
+            # Cria uma sessão de cliente MCP usando os streams
+            async with ClientSession(read_stream, write_stream) as session:
+                await session.initialize()  # Inicializa a sessão
+                tool_result = await session.list_tools()  # Solicita a lista de ferramentas
+        return tool_result
+    
+    async def call_tool(self, tool_name: str, params: dict = None) -> dict:
+        """
+        Executa uma ferramenta específica disponível no MCP.
+
+        Args:
+            tool_name (str): Nome da ferramenta a ser executada.
+            params (dict, opcional): Parâmetros para a execução da ferramenta.
+
+        Returns:
+            dict: Resultado da execução da ferramenta.
+        """
+        # Cria um cliente HTTP streamable para comunicação assíncrona
+        async with streamablehttp_client(self.url, self.headers, timeout=120, terminate_on_close=False) as (
+            read_stream,
+            write_stream,
+            _,
+        ):
+            # Cria uma sessão de cliente MCP usando os streams
+            async with ClientSession(read_stream, write_stream) as session:
+                await session.initialize()  # Inicializa a sessão
+                result = await session.call_tool(tool_name, params)  # Executa a ferramenta
+        return result

biatoolkit-1.0.3/biatoolkit/util.py

@@ -0,0 +1,79 @@
+from mcp.server.fastmcp import FastMCP
+from .schema.header import Header
+
+import boto3
+import os
+
+class BiaUtil():
+
+    HEADER_PREFIX = "x-amzn-bedrock-agentcore-runtime-custom"
+
+    def __init__(self, mcp: FastMCP):
+        """
+        Inicializa o utilitário BiaUtil com uma instância do FastMCP.
+
+        Args:
+            mcp (FastMCP): Instância do servidor FastMCP para obter contexto da requisição.
+        """
+        self.mcp = mcp
+
+    def __get_from_ssm(self, parameter_name: str) -> str:
+        """
+        Busca o valor de um parâmetro no AWS SSM Parameter Store.
+
+        Args:
+            parameter_name (str): Nome do parâmetro a ser buscado.
+
+        Returns:
+            str: Valor do parâmetro, ou None se não encontrado.
+        """
+        ctx = self.mcp.get_context()  # Obtém o contexto da requisição atual
+        headers = ctx.request_context.request.headers  # Acessa os headers da requisição
+        prefix = headers.get(f"{self.HEADER_PREFIX}-prefix", None)  # Prefixo customizado do header
+        client = boto3.client('ssm', region_name="sa-east-1")  # Cria cliente SSM na região especificada
+        try:
+            response = client.get_parameter(
+                Name=f"{prefix}/{parameter_name}",
+                WithDecryption=True
+            )
+        except client.exceptions.ParameterNotFound:
+            return None  # Retorna None se o parâmetro não for encontrado
+        return response.get('Parameter').get('Value')  # Retorna o valor do parâmetro
+
+    def get_header(self) -> Header:
+        """
+        Retorna os parâmetros padrão contidos no header da requisição.
+
+        Returns:
+            Header: Objeto Header preenchido com os valores dos headers customizados.
+        """
+        ctx = self.mcp.get_context()  # Obtém o contexto da requisição atual
+        headers = ctx.request_context.request.headers  # Acessa os headers da requisição
+        # Preenche o objeto Header com os valores dos headers customizados
+        return Header(
+            current_host=headers.get(f"{self.HEADER_PREFIX}-current-host", None),
+            user_email=headers.get(f"{self.HEADER_PREFIX}-user-email", None),
+            jwt_token=headers.get(f"{self.HEADER_PREFIX}-jwt-token", None),
+            jsessionid=headers.get(f"{self.HEADER_PREFIX}-jsessionid", None),
+            organization_id=int(headers.get(f"{self.HEADER_PREFIX}-organization-id", 0)),
+            codparc=int(headers.get(f"{self.HEADER_PREFIX}-codparc", 0)),
+            iam_user_id=int(headers.get(f"{self.HEADER_PREFIX}-iam-user-id", 0)),
+            gateway_token=headers.get(f"{self.HEADER_PREFIX}-gateway-token", None)
+        )
+    
+    def get_parameter(self, parameter_name: str) -> str:
+        """
+        Retorna o valor do parâmetro, buscando primeiro na variável de ambiente
+        e depois no AWS SSM Parameter Store caso não exista na variável de ambiente.
+
+        Args:
+            parameter_name (str): Nome do parâmetro a ser buscado.
+
+        Returns:
+            str: Valor do parâmetro, ou None se não encontrado.
+        """
+        # Busca o valor na variável de ambiente, se não existir busca no SSM
+        return os.getenv(
+            parameter_name,
+            self.__get_from_ssm(parameter_name)
+        )

{biatoolkit-1.0.2 → biatoolkit-1.0.3}/biatoolkit.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: biatoolkit
-Version: 1.0.2
+Version: 1.0.3
 Summary: Biblioteca para desenvolvedores que utilizam o BiaAgentBuilder
 Author: Bia Platform Team
 Author-email: data.platform@sankhya.com.br

{biatoolkit-1.0.2 → biatoolkit-1.0.3}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name='biatoolkit',
-    version='1.0.2',
+    version='1.0.3',
     packages=find_packages(),
     install_requires=[],
     author='Bia Platform Team',  

biatoolkit-1.0.2/biatoolkit/basic_client.py

@@ -1,34 +0,0 @@
-from mcp import ClientSession
-from mcp.client.streamable_http import streamablehttp_client
-
-class BasicClient:
-
-    def __init__(self, url: str = "http://0.0.0.0:8000/mcp", headers: dict = None):
-        sufix = "/mcp"
-        self.url = url if url.endswith(sufix) else f"{url}{sufix}"
-        self.headers = headers
-
-    async def list_tools(self) -> dict:
-        """Lista as ferramentas disponíveis no MCP"""
-        async with streamablehttp_client(self.url, self.headers, timeout=120, terminate_on_close=False) as (
-            read_stream,
-            write_stream,
-            _,
-        ):
-            async with ClientSession(read_stream, write_stream) as session:
-                await session.initialize()
-                tool_result = await session.list_tools()
-        return tool_result
-    
-    async def call_tool(self, tool_name: str, params: dict = None) -> dict:
-        """Executa a ferramenta disponível no MCP"""
-        async with streamablehttp_client(self.url, self.headers, timeout=120, terminate_on_close=False) as (
-            read_stream,
-            write_stream,
-            _,
-        ):
-            async with ClientSession(read_stream, write_stream) as session:
-                await session.initialize()
-                result = await session.call_tool(tool_name, params)
-        return result
-    

biatoolkit-1.0.2/biatoolkit/util.py

@@ -1,52 +0,0 @@
-from mcp.server.fastmcp import FastMCP
-from .schema.header import Header
-
-import boto3
-import os
-
-class Util():
-
-    HEADER_PREFIX = "x-amzn-bedrock-agentcore-runtime-custom"
-
-    def __init__(self, mcp: FastMCP):
-        self.mcp = mcp
-
-    def __get_from_ssm(self, parameter_name: str) -> str:
-        """Busca o parâmetro no AWS SSM Parameter Store"""
-        ctx = self.mcp.get_context()
-        headers = ctx.request_context.request.headers
-        prefix = headers.get(f"{self.HEADER_PREFIX}-prefix", None)
-        client = boto3.client('ssm', region_name="sa-east-1")
-        try:
-            response = client.get_parameter(
-                Name=f"{prefix}/{parameter_name}",
-                WithDecryption=True
-            )
-        except client.exceptions.ParameterNotFound:
-            return None
-        return response.get('Parameter').get('Value')
-
-    def get_header(self) -> Header:
-        """Retorna os parâmetros padrão contidos no header da requisição"""
-        ctx = self.mcp.get_context()
-        headers = ctx.request_context.request.headers
-        return Header(
-            current_host=headers.get(f"{self.HEADER_PREFIX}-current-host", None),
-            user_email=headers.get(f"{self.HEADER_PREFIX}-user-email", None),
-            jwt_token=headers.get(f"{self.HEADER_PREFIX}-jwt-token", None),
-            jsessionid=headers.get(f"{self.HEADER_PREFIX}-jsessionid", None),
-            organization_id=int(headers.get(f"{self.HEADER_PREFIX}-organization-id", 0)),
-            codparc=int(headers.get(f"{self.HEADER_PREFIX}-codparc", 0)),
-            iam_user_id=int(headers.get(f"{self.HEADER_PREFIX}-iam-user-id", 0)),
-            gateway_token=headers.get(f"{self.HEADER_PREFIX}-gateway-token", None)
-        )
-    
-    def get_parameter(self, parameter_name: str) -> str:
-        """
-            Retorna o valor do parâmetro, buscando primeiro na variável de
-            ambiente e depois no AWS SSM Parameter Store.
-        """
-        return os.getenv(
-            parameter_name,
-            self.__get_from_ssm(parameter_name)
-        )