Online Research

1
!pip install -U athena-intelligence

1
import json
2
import pandas as pd
3
import os
4
5
ATHENA_API_KEY = os.environ["ATHENA_API_KEY"]
6
7
from athena import Model, Tools
8
from athena.client import AsyncAthena
9
10
athena = AsyncAthena(
11
    api_key=ATHENA_API_KEY,
12
)

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)