You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
38 lines
702 B
38 lines
702 B
# Importing necessary modules
|
|
import os
|
|
|
|
from dotenv import load_dotenv
|
|
|
|
from swarms import OpenAIChat, Worker, tool
|
|
|
|
# Loading environment variables from .env file
|
|
load_dotenv()
|
|
|
|
# Retrieving the OpenAI API key from environment variables
|
|
api_key = os.getenv("OPENAI_API_KEY")
|
|
|
|
|
|
# Create a tool
|
|
@tool
|
|
def search_api(query: str):
|
|
pass
|
|
|
|
|
|
# Creating a Worker instance
|
|
worker = Worker(
|
|
name="My Worker",
|
|
role="Worker",
|
|
human_in_the_loop=False,
|
|
tools=[search_api],
|
|
temperature=0.5,
|
|
llm=OpenAIChat(openai_api_key=api_key),
|
|
)
|
|
|
|
# Running the worker with a prompt
|
|
out = worker.run(
|
|
"Hello, how are you? Create an image of how your are doing!"
|
|
)
|
|
|
|
# Printing the output
|
|
print(out)
|