<div id="readability-page-1" class="page"><div>We are E2B. We provide sandboxed cloud environments for AI-powered apps and agentic workflows. Check out our sandbox runtime for LLMs.
In this guide, we will create an example of a LangChain agent that uses E2B cloud sandbox and GPT-4 to analyze your uploaded data.
See the final guide and code in the official LangChain documentation here.
<h2 id="why-build-your-own-agent-with-e2b">Why build your own agent with E2B?</h2>
E2B's cloud environments are runtime sandboxes for LLMs. They are an ideal fit for building AI assistants like code interpreters or advanced data-analyzing tools. We can use E2B's Data Analysis Sandbox for our use case.
Compared to assistants running their code locally, e.g. via Docker, the Data Analysis Sandbox allows for safe code execution in a remote environment.&nbsp;
That is a secure way to run the unpredictable LLM-generated code on your computer without the potential harm that such code can cause to your machine, e.g., unauthorized access to vulnerable data.
<figure></figure>
We will create an assistant that will use OpenAI’s GPT-4 and E2B's Data Analysis sandbox to perform analysis on uploaded files using Python.
Let's get to hacking!
<h3 id="1-get-api-keys-import-packages">1. Get API keys, import packages</h3>
First, we have to ensure that we have the latest version of E2B.
<figure><div><pre tabindex="0"><code>pip install -U e2b langchain</code></pre></div></figure>
We import <code>E2BDataAnalysisTool</code> and other necessary modules from LangChain. We get our OpenAI API key <a href="https://platform.openai.com/account/api-keys" target="_blank" rel="noopener noreferrer" data-external="true">here</a>, and our E2B API key <a href="https://docs.e2b.dev/api-key" target="_blank" rel="noopener noreferrer" data-external="true">here</a> and set them as environment variables.
<figure></figure>
🔎 Find the full OpenAI API documentation <a href="https://openai.com/blog/openai-api" target="_blank" rel="noopener noreferrer" data-external="true">here</a>.
<figure><div><pre tabindex="0"><code>import os
from langchain.chat_models import ChatOpenAI
from langchain.tools import E2BDataAnalysisTool
from langchain.agents import initialize_agent, AgentType
 
os.environ["E2B_API_KEY"] = "&lt;E2B_API_KEY&gt;"
os.environ["OPENAI_API_KEY"] = "&lt;OPENAI_API_KEY&gt;"</code></pre></div></figure>
<h3 id="2-initialize-the-e2b-tool-for-the-langchain-agent">2. Initialize the E2B tool for the LangChain agent</h3>
💡When creating an instance of the <code>E2BDataAnalysisTool</code>, you can pass callbacks to listen to the output of the sandbox. This is useful, for example, when creating a more responsive UI. Especially with the combination of streaming output from LLMs. We define a Python function save_artifact, which is used to handle and save charts created by Matplotlib. &nbsp;When a chart is generated with <code>plt.show()</code>, this function is called to print a message about the newly generated Matplotlib chart, downloads it as bytes, and then saves it to a directory named "charts."
<figure><div><pre tabindex="0"><code># Artifacts are charts created by matplotlib when `plt.show()` is called
def save_artifact(artifact):
&nbsp;&nbsp;print("New matplotlib chart generated:", artifact.name)
&nbsp;&nbsp;# Download the artifact as `bytes` and leave it up to the user to display them (on frontend, for example)
&nbsp;&nbsp;file = artifact.download()
&nbsp;&nbsp;basename = os.path.basename(artifact.name)
&nbsp;&nbsp;# Save the chart to the `charts` directory
&nbsp;&nbsp;# Make sure the "charts" directory exists
&nbsp;&nbsp;with open(f"./charts/{basename}", "wb") as f:
&nbsp;&nbsp;&nbsp;&nbsp;f.write(file)
e2b_data_analysis_tool = E2BDataAnalysisTool(
&nbsp;&nbsp;&nbsp;&nbsp;on_stdout=lambda stdout: print("stdout:", stdout),
&nbsp;&nbsp;&nbsp;&nbsp;on_stderr=lambda stderr: print("stderr:", stderr),
&nbsp;&nbsp;&nbsp;&nbsp;on_artifact=save_artifact,
)</code></pre></div></figure>
<h3 id="3-upload-your-data-file">3. Upload your data file</h3>
You can choose your own CSV data file to upload to the E2B sandbox. In our example, we chose a file about Netflix TV shows. You can download the file <a href="https://storage.googleapis.com/e2b-examples/netflix.csv" target="_blank" rel="noopener noreferrer" data-external="true">here</a>.
The following code reads a file named "netflix.csv" from the local file system. It then uses the <code>e2b_data_analysis_tool.upload_file</code> method to upload the contents of this file to the sandbox and print the path where the file is saved in the sandbox.
<figure><div><pre tabindex="0"><code>with open("./netflix.csv") as f:
&nbsp;&nbsp;&nbsp;&nbsp;remote_path = e2b_data_analysis_tool.upload_file(
&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;file=f,
&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;description="Data about Netflix tv shows including their title, category, director, release date, casting, age rating, etc.",
&nbsp;&nbsp;&nbsp;&nbsp;)
&nbsp;&nbsp;&nbsp;&nbsp;print(remote_path)</code></pre></div></figure>
<h5 id="code-output">▶️ Code output</h5>
<figure><div><pre tabindex="0"><code>name='netflix.csv' 
remote_path='/home/user/netflix.csv' 
description='Data about Netflix tv shows including their title, category, director, release date, casting, age rating, etc.'</code></pre></div></figure>
<h3 id="4-create-tools-and-initialize-the-agent">4. Create tools and initialize the agent</h3>
Now we get to set up the LangChain agent. It will be using GPT-4 and the <code>e2b_data_analysis_tool</code> we created earlier.
<figure><div><pre tabindex="0"><code>tools = [e2b_data_analysis_tool.as_tool()]

llm = ChatOpenAI(model="gpt-4", temperature=0)
agent = initialize_agent(
&nbsp;&nbsp;&nbsp;&nbsp;tools, llm, agent=AgentType.OPENAI_FUNCTIONS, verbose=True, handle_parsing_errors=True
)</code></pre></div></figure>
<h3 id="5-execute-a-query">5. Execute a query</h3>
We initiate the execution of the agent with a specific query or task.
<figure><div><pre tabindex="0"><code>agent.run(
 "What are the 5 longest movies on netflix released between 2000 and 2010? Create a chart with their lengths."
)</code></pre></div></figure>
<h5 id="code-output-1">▶️ Code output</h5>
<figure></figure>
<h3 id="6-sandbox-advanced-features">6. Sandbox advanced features</h3>
E2B also allows you to install both Python and system (via <code>apt</code>) packages dynamically during runtime like this:
<figure><div><pre tabindex="0"><code># Install Python package
e2b_data_analysis_tool.install_python_packages('pandas')</code></pre></div></figure>
<h5 id="code-output-2">▶️ Code output</h5>
<figure></figure>
Additionally, you can download any file from the sandbox like this:
<figure><div><pre tabindex="0"><code># The path is a remote path in the sandbox
files_in_bytes = e2b_data_analysis_tool.download_file('/home/user/netflix.csv')</code></pre></div></figure>
Lastly, you can run any shell command inside the sandbox via <code>run_command</code>.
<figure><div><pre tabindex="0"><code># Install SQLite
e2b_data_analysis_tool.run_command("sudo apt update")
e2b_data_analysis_tool.install_system_packages("sqlite3")
# Check the SQLite version
output = e2b_data_analysis_tool.run_command("sqlite3 --version")
print("version: ", output["stdout"])
print("error: ", output["stderr"])
print("exit code: ", output["exit_code"])</code></pre></div></figure>
<h5 id="code-output-3">▶️ Code output</h5>
<figure></figure>
<h3 id="7-close-the-sandbox">7. Close the sandbox</h3>
When your agent is finished, don't forget to close the sandbox.
<figure><div><pre tabindex="0"><code>e2b_data_analysis_tool.close()</code></pre></div></figure>
<h2 id="output">Output</h2>
If you try this example with our <a href="https://storage.googleapis.com/e2b-examples/netflix.csv" target="_blank" rel="noopener noreferrer" data-external="true">Netflix file</a>, the output should be automatically saved into your local directory like this:
<figure></figure>
See the final guide and code in the official LangChain documentation <a href="https://python.langchain.com/docs/integrations/tools/e2b_data_analysis?ref=build-ai-data-analyst-with-langchain-and-e2b" target="_blank" rel="noopener noreferrer" data-external="true">here</a>. See <a href="https://docs.e2b.dev/" target="_blank" rel="noopener noreferrer" data-external="true">E2B docs</a> here.
<ul>
<li>Need help or want to share feedback? Join our <a href="https://discord.com/invite/U7KEcGErtQ" target="_blank" rel="noopener noreferrer" data-external="true">Discord</a>.</li>
<li>If you like the guide, please support us with a star on <a href="https://github.com/e2b-dev/e2b" target="_blank" rel="noopener noreferrer" data-external="true">GitHub</a>.</li>
<li>Follow us on <a href="https://twitter.com/e2b_dev" target="_blank" rel="noopener noreferrer" data-external="true">X (Twitter)</a>.</li>
<li>You can also reach us at <a href="mailto:hello@e2b.dev">hello@e2b.dev</a>.</li>
</ul>
‍</div></div>

E2B Blog & Customer Stories

Vaclav Mlejnsky

Create a LangChain agent with E2B cloud sandbox and GPT-4 to analyze your uploaded data.

AI Data Analyst in Cloud Sandbox with LangChain & E2B

댓글