mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-04-13 06:23:03 +00:00
* fix: add path and URL validation to RAG tools Add validation utilities to prevent unauthorized file reads and SSRF when RAG tools accept LLM-controlled paths/URLs at runtime. Changes: - New crewai_tools.utilities.safe_path module with validate_file_path(), validate_directory_path(), and validate_url() - File paths validated against base directory (defaults to cwd). Resolves symlinks and ../ traversal. Rejects escape attempts. - URLs validated: file:// blocked entirely. HTTP/HTTPS resolves DNS and blocks private/reserved IPs (10.x, 172.16-31.x, 192.168.x, 127.x, 169.254.x, 0.0.0.0, ::1, fc00::/7). - Validation applied in RagTool.add() — catches all RAG search tools (JSON, CSV, PDF, TXT, DOCX, MDX, Directory, etc.) - Removed file:// scheme support from DataTypes.from_content() - CREWAI_TOOLS_ALLOW_UNSAFE_PATHS=true env var for backward compat - 27 tests covering traversal, symlinks, private IPs, cloud metadata, IPv6, escape hatch, and valid paths/URLs * fix: validate path/URL keyword args in RagTool.add() The original patch validated positional *args but left all keyword arguments (path=, file_path=, directory_path=, url=, website=, github_url=, youtube_url=) unvalidated, providing a trivial bypass for both path-traversal and SSRF checks. Applies validate_file_path() to path/file_path/directory_path kwargs and validate_url() to url/website/github_url/youtube_url kwargs before they reach the adapter. Adds a regression-test file covering all eight kwarg vectors plus the two existing positional-arg checks. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com> * fix: address CodeQL and review comments on RAG path/URL validation - Replace insecure tempfile.mktemp() with inline symlink target in test - Remove unused 'target' variable and unused tempfile import - Narrow broad except Exception: pass to only catch urlparse errors; validate_url ValueError now propagates instead of being silently swallowed - Fix ruff B904 (raise-without-from-inside-except) in safe_path.py - Fix ruff B007 (unused loop variable 'family') in safe_path.py - Use validate_directory_path in DirectorySearchTool.add() so the public utility is exercised in production code Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com> * style: fix ruff format + remaining lint issues * fix: resolve mypy type errors in RAG path/URL validation - Cast sockaddr[0] to str() to satisfy mypy (socket.getaddrinfo returns sockaddr where [0] is str but typed as str | int) - Remove now-unnecessary `type: ignore[assignment]` and `type: ignore[literal-required]` comments in rag_tool.py Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com> * fix: unroll dynamic TypedDict key loops to satisfy mypy literal-required Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com> * test: allow tmp paths in RAG data-type tests via CREWAI_TOOLS_ALLOW_UNSAFE_PATHS TemporaryDirectory creates files under /tmp/ which is outside CWD and is correctly blocked by the new path validation. These tests exercise data-type handling, not security, so add an autouse fixture that sets CREWAI_TOOLS_ALLOW_UNSAFE_PATHS=true for the whole file. Path/URL security is covered by test_rag_tool_path_validation.py. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com> * test: allow tmp paths in search-tool and rag_tool tests via CREWAI_TOOLS_ALLOW_UNSAFE_PATHS test_search_tools.py has tests for TXTSearchTool, CSVSearchTool, MDXSearchTool, JSONSearchTool, and DirectorySearchTool that create files under /tmp/ via tempfile, which is outside CWD and correctly blocked by the new path validation. rag_tool_test.py has one test that calls tool.add() with a TemporaryDirectory path. Add the same autouse allow_tmp_paths fixture used in test_rag_tool_add_data_type.py. Security is covered separately by test_rag_tool_path_validation.py. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com> * chore: update tool specifications * docs: document CodeInterpreterTool removal and RAG path/URL validation Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com> * fix: address three review comments on path/URL validation - safe_path._is_private_or_reserved: after unwrapping IPv4-mapped IPv6 to IPv4, only check against IPv4 networks to avoid TypeError when comparing an IPv4Address against IPv6Network objects. - safe_path.validate_file_path: handle filesystem-root base_dir ('/') by not appending os.sep when the base already ends with a separator, preventing the '//'-prefix bug. - rag_tool.add: path-detection heuristic now checks for both '/' and os.sep so forward-slash paths are caught on Windows as well as Unix. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com> * fix: remove unused _BLOCKED_NETWORKS variable after IPv4/IPv6 split * chore: update tool specifications --------- Co-authored-by: Claude Sonnet 4.6 <noreply@anthropic.com> Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
215 lines
8.6 KiB
Plaintext
215 lines
8.6 KiB
Plaintext
---
|
|
title: Interpretador de Código
|
|
description: O `CodeInterpreterTool` é uma poderosa ferramenta projetada para executar código Python 3 em um ambiente seguro e isolado.
|
|
icon: code-simple
|
|
mode: "wide"
|
|
---
|
|
|
|
# `CodeInterpreterTool`
|
|
|
|
<Warning>
|
|
**Depreciado:** O `CodeInterpreterTool` foi removido do `crewai-tools`. Os parâmetros `allow_code_execution` e `code_execution_mode` do `Agent` também estão depreciados. Use um serviço de sandbox dedicado — [E2B](https://e2b.dev) ou [Modal](https://modal.com) — para execução de código segura e isolada.
|
|
</Warning>
|
|
|
|
## Descrição
|
|
|
|
O `CodeInterpreterTool` permite que agentes CrewAI executem códigos Python 3 gerados autonomamente. Essa funcionalidade é particularmente valiosa, pois permite que os agentes criem códigos, os executem, obtenham os resultados e usem essas informações para orientar decisões e ações subsequentes.
|
|
|
|
Há diversas formas de usar esta ferramenta:
|
|
|
|
### Container Docker (Recomendado)
|
|
|
|
Esta é a opção principal. O código é executado em um container Docker seguro e isolado, garantindo a segurança independentemente de seu conteúdo.
|
|
Certifique-se de que o Docker esteja instalado e em funcionamento em seu sistema. Se ainda não tiver, você pode instalá-lo a partir [deste link](https://docs.docker.com/get-docker/).
|
|
|
|
### Ambiente Sandbox
|
|
|
|
Se o Docker não estiver disponível — seja por não estar instalado ou inacessível por qualquer motivo — o código será executado em um ambiente Python restrito, chamado de sandbox.
|
|
Esse ambiente é bastante limitado, com restrições severas a vários módulos e funções embutidas.
|
|
|
|
### Execução Não Segura
|
|
|
|
**NÃO RECOMENDADO PARA PRODUÇÃO**
|
|
Este modo permite a execução de qualquer código Python, inclusive chamadas perigosas para os módulos `sys, os..` e semelhantes. [Veja aqui](/pt-BR/tools/ai-ml/codeinterpretertool#enabling-unsafe-mode) como habilitar este modo.
|
|
|
|
## Registro de Logs
|
|
|
|
O `CodeInterpreterTool` registra a estratégia de execução selecionada no STDOUT.
|
|
|
|
|
|
## Instalação
|
|
|
|
Para utilizar esta ferramenta, você precisa instalar o pacote de ferramentas CrewAI:
|
|
|
|
```shell
|
|
pip install 'crewai[tools]'
|
|
```
|
|
|
|
## Exemplo
|
|
|
|
O exemplo a seguir demonstra como usar o `CodeInterpreterTool` com um agente CrewAI:
|
|
|
|
```python Code
|
|
from crewai import Agent, Task, Crew, Process
|
|
from crewai_tools import CodeInterpreterTool
|
|
|
|
# Initialize the tool
|
|
code_interpreter = CodeInterpreterTool()
|
|
|
|
# Define an agent that uses the tool
|
|
programmer_agent = Agent(
|
|
role="Python Programmer",
|
|
goal="Write and execute Python code to solve problems",
|
|
backstory="An expert Python programmer who can write efficient code to solve complex problems.",
|
|
tools=[code_interpreter],
|
|
verbose=True,
|
|
)
|
|
|
|
# Example task to generate and execute code
|
|
coding_task = Task(
|
|
description="Write a Python function to calculate the Fibonacci sequence up to the 10th number and print the result.",
|
|
expected_output="The Fibonacci sequence up to the 10th number.",
|
|
agent=programmer_agent,
|
|
)
|
|
|
|
# Create and run the crew
|
|
crew = Crew(
|
|
agents=[programmer_agent],
|
|
tasks=[coding_task],
|
|
verbose=True,
|
|
process=Process.sequential,
|
|
)
|
|
result = crew.kickoff()
|
|
```
|
|
|
|
Você também pode habilitar a execução de código diretamente ao criar um agente:
|
|
|
|
```python Code
|
|
from crewai import Agent
|
|
|
|
# Create an agent with code execution enabled
|
|
programmer_agent = Agent(
|
|
role="Python Programmer",
|
|
goal="Write and execute Python code to solve problems",
|
|
backstory="An expert Python programmer who can write efficient code to solve complex problems.",
|
|
allow_code_execution=True, # This automatically adds the CodeInterpreterTool
|
|
verbose=True,
|
|
)
|
|
```
|
|
|
|
### Habilitando o `unsafe_mode`
|
|
|
|
```python Code
|
|
from crewai_tools import CodeInterpreterTool
|
|
|
|
code = """
|
|
import os
|
|
os.system("ls -la")
|
|
"""
|
|
|
|
CodeInterpreterTool(unsafe_mode=True).run(code=code)
|
|
```
|
|
|
|
## Parâmetros
|
|
|
|
O `CodeInterpreterTool` aceita os seguintes parâmetros durante a inicialização:
|
|
|
|
- **user_dockerfile_path**: Opcional. Caminho para um Dockerfile personalizado a ser utilizado pelo container do interpretador de código.
|
|
- **user_docker_base_url**: Opcional. URL do daemon Docker que será usado para rodar o container.
|
|
- **unsafe_mode**: Opcional. Indica se o código será executado diretamente na máquina hospedeira ao invés de um container Docker ou sandbox. O padrão é `False`. Use com cautela!
|
|
- **default_image_tag**: Opcional. Tag padrão da imagem Docker. O padrão é `code-interpreter:latest`
|
|
|
|
Ao utilizar a ferramenta com um agente, o agente precisará fornecer:
|
|
|
|
- **code**: Obrigatório. O código Python 3 a ser executado.
|
|
- **libraries_used**: Opcional. Uma lista de bibliotecas usadas no código que precisam ser instaladas. O padrão é `[]`
|
|
|
|
## Exemplo de Integração com Agente
|
|
|
|
Aqui está um exemplo mais detalhado de como integrar o `CodeInterpreterTool` com um agente CrewAI:
|
|
|
|
```python Code
|
|
from crewai import Agent, Task, Crew
|
|
from crewai_tools import CodeInterpreterTool
|
|
|
|
# Initialize the tool
|
|
code_interpreter = CodeInterpreterTool()
|
|
|
|
# Define an agent that uses the tool
|
|
data_analyst = Agent(
|
|
role="Data Analyst",
|
|
goal="Analyze data using Python code",
|
|
backstory="""You are an expert data analyst who specializes in using Python
|
|
to analyze and visualize data. You can write efficient code to process
|
|
large datasets and extract meaningful insights.""",
|
|
tools=[code_interpreter],
|
|
verbose=True,
|
|
)
|
|
|
|
# Create a task for the agent
|
|
analysis_task = Task(
|
|
description="""
|
|
Write Python code to:
|
|
1. Generate a random dataset of 100 points with x and y coordinates
|
|
2. Calculate the correlation coefficient between x and y
|
|
3. Create a scatter plot of the data
|
|
4. Print the correlation coefficient and save the plot as 'scatter.png'
|
|
|
|
Make sure to handle any necessary imports and print the results.
|
|
""",
|
|
expected_output="The correlation coefficient and confirmation that the scatter plot has been saved.",
|
|
agent=data_analyst,
|
|
)
|
|
|
|
# Run the task
|
|
crew = Crew(
|
|
agents=[data_analyst],
|
|
tasks=[analysis_task],
|
|
verbose=True,
|
|
process=Process.sequential,
|
|
)
|
|
result = crew.kickoff()
|
|
```
|
|
|
|
## Detalhes de Implementação
|
|
|
|
O `CodeInterpreterTool` utiliza Docker para criar um ambiente seguro para execução de código:
|
|
|
|
```python Code
|
|
class CodeInterpreterTool(BaseTool):
|
|
name: str = "Code Interpreter"
|
|
description: str = "Interprets Python3 code strings with a final print statement."
|
|
args_schema: Type[BaseModel] = CodeInterpreterSchema
|
|
default_image_tag: str = "code-interpreter:latest"
|
|
|
|
def _run(self, **kwargs) -> str:
|
|
code = kwargs.get("code", self.code)
|
|
libraries_used = kwargs.get("libraries_used", [])
|
|
|
|
if self.unsafe_mode:
|
|
return self.run_code_unsafe(code, libraries_used)
|
|
else:
|
|
return self.run_code_safety(code, libraries_used)
|
|
```
|
|
|
|
A ferramenta executa os seguintes passos:
|
|
1. Verifica se a imagem Docker existe ou a constrói, caso necessário
|
|
2. Cria um container Docker com o diretório de trabalho atual montado
|
|
3. Instala quaisquer bibliotecas necessárias especificadas pelo agente
|
|
4. Executa o código Python dentro do container
|
|
5. Retorna a saída da execução do código
|
|
6. Limpa o ambiente, parando e removendo o container
|
|
|
|
## Considerações de Segurança
|
|
|
|
Por padrão, o `CodeInterpreterTool` executa o código em um container Docker isolado, fornecendo uma camada de segurança. No entanto, ainda há algumas considerações importantes:
|
|
|
|
1. O container Docker tem acesso ao diretório de trabalho atual, então arquivos sensíveis podem ser potencialmente acessados.
|
|
2. Caso o container Docker não esteja disponível e o código precise ser executado de forma segura, ele será executado em um ambiente sandbox. Por motivos de segurança, a instalação de bibliotecas arbitrárias não é permitida
|
|
3. O parâmetro `unsafe_mode` permite que códigos sejam executados diretamente na máquina hospedeira, o que deve ser usado apenas em ambientes confiáveis.
|
|
4. Tenha cautela ao permitir que agentes instalem bibliotecas arbitrárias, pois estas podem incluir códigos maliciosos.
|
|
|
|
## Conclusão
|
|
|
|
O `CodeInterpreterTool` oferece uma maneira poderosa para que agentes CrewAI executem código Python em um ambiente relativamente seguro. Permitindo que agentes escrevam e executem códigos, ele amplia significativamente sua capacidade de resolução de problemas, especialmente para tarefas que envolvem análise de dados, cálculos ou outros trabalhos computacionais. Esta ferramenta é especialmente útil para agentes que precisam realizar operações complexas que são mais eficientemente expressas em código do que em linguagem natural.
|