Introduction
In this tutorial, you'll learn how to create and interact with simple AI agents using Python and the OpenAI API. This hands-on guide will teach you the fundamentals of building AI assistants that can help with tasks like answering questions, summarizing text, and even creating content. We'll start with basic concepts and build up to a working AI agent that you can interact with.
Prerequisites
- A computer with internet access
- Python 3.7 or higher installed
- An OpenAI API key (free to get at platform.openai.com)
- Basic understanding of Python programming concepts
Step-by-Step Instructions
Step 1: Setting Up Your Environment
Install Required Python Packages
First, we need to install the OpenAI Python library that will allow us to communicate with the AI models. Open your terminal or command prompt and run:
pip install openai
This command installs the official OpenAI Python library, which provides an easy way to interact with OpenAI's API services.
Step 2: Getting Your API Key
Creating an Account and Getting Your Key
Visit https://platform.openai.com and create a free account. After logging in, navigate to the "API Keys" section and click "Create new secret key". Copy this key - you'll need it in the next step.
Why this step is important: The API key authenticates your requests to OpenAI's servers, allowing you to use their powerful AI models for free (with limits) or paid services.
Step 3: Creating Your First AI Agent Script
Basic Python Setup
Create a new Python file called ai_agent.py and start with this basic structure:
import openai
# Set up your API key
openai.api_key = "your-api-key-here"
# Create a simple AI agent function
def ai_agent(prompt):
response = openai.ChatCompletion.create(
model="gpt-3.5-turbo",
messages=[
{"role": "system", "content": "You are a helpful assistant."},
{"role": "user", "content": prompt}
]
)
return response.choices[0].message.content
Why this code works: This sets up a basic conversation with the AI model where we define a system message (telling the AI what role it should play) and a user message (the question you want to ask).
Step 4: Testing Your AI Agent
Running Your First Test
Replace "your-api-key-here" with your actual API key from Step 2. Then run your script:
python ai_agent.py
You should see a response about artificial intelligence. This demonstrates how your AI agent can understand and respond to questions.
Step 5: Making Your Agent More Useful
Adding More Features
Let's enhance our agent to handle multiple types of tasks. Update your script with this improved version:
import openai
openai.api_key = "your-api-key-here"
def ai_agent(task, user_input):
if task == "answer":
system_prompt = "You are a helpful assistant that answers questions accurately."
user_prompt = user_input
elif task == "summarize":
system_prompt = "You are a helpful assistant that summarizes text concisely."
user_prompt = f"Please summarize this text: {user_input}"
elif task == "write":
system_prompt = "You are a helpful assistant that writes creative content."
user_prompt = f"Write a short story about {user_input}"
else:
system_prompt = "You are a helpful assistant."
user_prompt = user_input
response = openai.ChatCompletion.create(
model="gpt-3.5-turbo",
messages=[
{"role": "system", "content": system_prompt},
{"role": "user", "content": user_prompt}
]
)
return response.choices[0].message.content
# Test different tasks
print("Answering a question:")
print(ai_agent("answer", "What is machine learning?"))
print("\nSummarizing text:")
print(ai_agent("summarize", "Artificial intelligence is intelligence demonstrated by machines, in contrast to the natural intelligence displayed by humans and animals."))
print("\nWriting content:")
print(ai_agent("write", "a robot learning to dance"))
Why this approach works: By using different system prompts for different tasks, we're training our AI agent to behave appropriately for each specific use case - whether it's answering questions, summarizing, or creating content.
Step 6: Building an Interactive Agent
Creating a Chat Interface
Let's make our agent interactive by creating a simple chat interface:
import openai
openai.api_key = "your-api-key-here"
def interactive_agent():
print("AI Agent: Hello! I'm your AI assistant. Type 'quit' to exit.")
while True:
user_input = input("You: ")
if user_input.lower() in ['quit', 'exit', 'bye']:
print("AI Agent: Goodbye!")
break
response = openai.ChatCompletion.create(
model="gpt-3.5-turbo",
messages=[
{"role": "system", "content": "You are a helpful assistant."},
{"role": "user", "content": user_input}
]
)
print(f"AI Agent: {response.choices[0].message.content}")
# Start the interactive agent
interactive_agent()
Why this is useful: This creates a real conversation experience where you can ask questions and get responses in real-time, just like interacting with an AI assistant.
Step 7: Running Your Complete Agent
Testing Your Full Agent
Save your updated script and run it:
python ai_agent.py
Try asking questions like "What is the weather like today?" or "Tell me a joke". You'll see how your AI agent responds to various prompts.
Summary
In this tutorial, you've learned how to create and interact with AI agents using Python and the OpenAI API. You started with basic setup, created a simple question-answering agent, enhanced it with multiple capabilities, and finally built an interactive chat interface. This foundation gives you the tools to build more complex AI assistants that can help with research, content creation, and various other tasks. As Google and other companies continue developing AI agent ecosystems, understanding these fundamental concepts will help you make the most of these emerging technologies.



