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.
69 lines
1.9 KiB
69 lines
1.9 KiB
8 months ago
|
from pydantic import BaseModel, Field
|
||
|
|
|
||
|
|
from swarms import Agent
|
||
|
|
from swarms.models.popular_llms import Anthropic
|
||
|
|
from swarms.tools.openai_tool_creator_decorator import tool
|
||
|
|
|
||
|
|
|
||
|
|
# Importing the search API tool
|
||
|
|
@tool
|
||
|
|
def search_api(query: str) -> str:
|
||
|
|
"""
|
||
|
|
This tool searches the web for information about COVID-19 symptoms.
|
||
|
|
"""
|
||
|
|
return f"Search API tool called with query: {query}"
|
||
|
|
|
||
|
|
|
||
|
|
print(search_api("COVID-19 symptoms"))
|
||
|
|
|
||
|
|
|
||
|
|
# Initialize the schema for the person's information
|
||
|
|
class Schema(BaseModel):
|
||
|
|
name: str = Field(..., title="Name of the person")
|
||
|
|
agent: int = Field(..., title="Age of the person")
|
||
|
|
is_student: bool = Field(..., title="Whether the person is a student")
|
||
|
|
courses: list[str] = Field(
|
||
|
|
..., title="List of courses the person is taking"
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
# Convert the schema to a JSON string
|
||
|
|
tool_schema = Schema(
|
||
|
|
name="Tool Name",
|
||
|
|
agent=1,
|
||
|
|
is_student=True,
|
||
|
|
courses=["Course1", "Course2"],
|
||
|
|
)
|
||
|
|
|
||
|
|
# Define the task to generate a person's information
|
||
|
|
task = "Generate a person's information based on the following schema:"
|
||
|
|
|
||
|
|
# Initialize the agent
|
||
|
|
agent = Agent(
|
||
|
|
agent_name="WeatherMan Agent",
|
||
|
|
# Set the tool schema to the JSON string -- this is the key difference
|
||
|
|
tool_schema=tool_schema,
|
||
|
|
llm=Anthropic(),
|
||
|
|
max_loops=3,
|
||
|
|
autosave=True,
|
||
|
|
dashboard=False,
|
||
|
|
streaming_on=True,
|
||
|
|
tools=[], # or list of tools
|
||
|
|
verbose=True,
|
||
|
|
interactive=True,
|
||
|
|
# Set the output type to the tool schema which is a BaseModel
|
||
|
|
output_type=tool_schema, # or dict, or str
|
||
|
|
metadata_output_type="json",
|
||
|
|
# List of schemas that the agent can handle
|
||
|
|
list_tool_schemas=[tool_schema],
|
||
|
|
function_calling_format_type="OpenAI",
|
||
|
|
function_calling_type="json", # or soon yaml
|
||
|
|
execute_tool=True,
|
||
|
|
)
|
||
|
|
|
||
|
|
# Run the agent to generate the person's information
|
||
|
|
generated_data = agent.run(task)
|
||
|
|
|
||
|
|
# Print the generated data
|
||
|
|
print(f"Generated data: {generated_data}")
|