Online Research

1
!pip install -U athena-intelligence

1
import json
2
import pandas as pd
3
4
ATHENA_API_KEY = "<YOUR API KEY HERE>"
5
6
from athena import Model, Tools
7
from athena.client import AsyncAthena
8
9
athena = AsyncAthena(
10
    api_key=ATHENA_API_KEY,
11
)

1
team = "KC Chiefs"
2
USER_MESSAGE_INPUT = f"""
3
Who plays quarterback in {team}?
4
"""
5
6
message = await athena.message.submit_and_poll(
7
    content=USER_MESSAGE_INPUT,
8
    model=Model.GPT_4_TURBO_PREVIEW,
9
    tools=[Tools.SEARCH, Tools.BROWSE, Tools.WIKIPEDIA],
10
)
11
12
print(message.content)

1
person = "Patrick Mahomes"
2
USER_MESSAGE_INPUT = f"""
3
Answer the question below using available tools. 
4
5
Who is {person}?
6
"""
7
8
message = await athena.message.submit_and_poll(
9
    content=USER_MESSAGE_INPUT,
10
    model=Model.GPT_4_TURBO_PREVIEW,
11
    tools=[ Tools.ENRICH_PERSON, Tools. ENRICH_COMPANY, Tools.SEARCH, Tools.BROWSE],
12
)
13
14
print(message.content)