Build with Agents

1
from athena import GeneralAgentConfig, GeneralAgentRequest, Tool
2
from athena.client import Athena
3
from langchain_core.messages import HumanMessage
4
from langchain_core.load import load
5
6
athena = Athena(api_key="<YOUR_API_KEY>")

1
# Enable tools
2
config = GeneralAgentConfig(
3
    enabled_tools=[Tool.SEARCH, Tool.BROWSE]
4
)
5
6
# Create message
7
messages = [{
8
    "type": "human", 
9
    "content": "Use the search tool to search for information about Athena Intelligence and summarize the results."
10
}]
11
12
# Get response
13
response = athena.agents.general.invoke(
14
    request=GeneralAgentRequest(
15
        config=config,
16
        messages=messages
17
    )
18
).messages
19
20
# Access the response dict directly
21
latest_response = response[-1]["kwargs"]["content"]

1
# Add follow-up as LangChain message object
2
new_messages = load(response) + [HumanMessage(content="Tell me more about what you found")]
3
4
# Get response
5
continued_response = athena.agents.general.invoke(
6
    request=GeneralAgentRequest(
7
        config=config,
8
        messages=new_messages
9
    )
10
).messages