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.
41 lines
1022 B
41 lines
1022 B
import os
|
|
|
|
from dotenv import load_dotenv
|
|
|
|
from swarms.structs.custom_agent import CustomAgent
|
|
|
|
load_dotenv()
|
|
|
|
# Example usage with Anthropic API
|
|
if __name__ == "__main__":
|
|
# Initialize the agent for Anthropic API
|
|
anthropic_agent = CustomAgent(
|
|
base_url="https://api.anthropic.com",
|
|
endpoint="v1/messages",
|
|
headers={
|
|
"x-api-key": os.getenv("ANTHROPIC_API_KEY"),
|
|
"anthropic-version": "2023-06-01",
|
|
},
|
|
)
|
|
|
|
# Example payload for Anthropic API
|
|
payload = {
|
|
"model": "claude-3-5-sonnet-20241022",
|
|
"max_tokens": 1000,
|
|
"messages": [
|
|
{
|
|
"role": "user",
|
|
"content": "Hello! Can you explain what artaddificial intelligence is?",
|
|
}
|
|
],
|
|
}
|
|
|
|
# Make the request
|
|
try:
|
|
response = anthropic_agent.run(payload)
|
|
print("Anthropic API Response:")
|
|
print(response)
|
|
print(type(response))
|
|
except Exception as e:
|
|
print(f"Error: {e}")
|