Agent automatically figures out how to make API requests given API docs and user query in plain text
Project description
Ad-Hoc API
An Archytas tool that uses LLMs to interact with APIs given documentation. User explains what they want in plain english, and then the agent (using the APIs docs for context) writes python code to complete the task.
Installation
pip install adhoc-api
Usage
This is designed to be paired with an Archytas agent. You may omit the python tool, and the agent should instead return the source code to you rather than running it.
Here is a complete example of grabbing the HTML content of an API documentation page, converting it to markdown, and then having the adhoc-api tool interact with the API using the generated markdown documentation (see adhoc_api/example.py for reference):
from archytas.react import ReActAgent, FailedTaskError
from archytas.tools import PythonTool
from easyrepl import REPL
from adhoc_api.tool import AdhocApi, APISpec
from bs4 import BeautifulSoup
import requests
from markdownify import markdownify
def main():
# set up the API spec for the JokeAPI
gdc_api: APISpec = {
'name': "JokesAPI",
'description': 'JokeAPI is a REST API that serves uniformly and well formatted jokes.',
'documentation': get_joke_api_documentation(),
# set a key if the docs are large and you want to cache them in the LLM context
# 'cache_key': 'adhoc_joke_api',
# set proofread instructions if there are specific features you want the output code to conform to (beyond general code correctness)
# 'proofread_instructions': 'The request code should be related to fetching a joke from the Jokes API. Please make sure any output is to the console'
}
# set up the tools and agent
adhoc_api = AdhocApi(
apis=[gdc_api],
drafter_config={'model': 'gemini-1.5-flash-001'},
finalizer_config={'model': 'gpt-4o'}
)
python = PythonTool()
agent = ReActAgent(model='gpt-4o', tools=[adhoc_api, python], verbose=True)
# REPL to interact with agent
for query in REPL(history_file='.chat'):
try:
answer = agent.react(query)
print(answer)
except FailedTaskError as e:
print(f"Error: {e}")
def get_joke_api_documentation() -> str:
"""Download the HTML of the joke API documentation page with soup and convert it to markdown."""
url = 'https://sv443.net/jokeapi/v2/'
response = requests.get(url)
soup = BeautifulSoup(response.text, 'html.parser')
markdown = markdownify(str(soup))
return markdown
if __name__ == "__main__":
main()
Then you can run the script and interact with the agent in the REPL:
$ python example.py
>>> Can you tell me what apis are available?
The available API is JokesAPI, which is a REST API that serves uniformly and well formatted jokes.
>>> Can you fetch a safe joke?
Here is a safe joke from the JokesAPI:
Category: Pun
Type: Two-part
Setup: What kind of doctor is Dr. Pepper?
Delivery: He's a fizzician.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
File details
Details for the file adhoc_api-0.2.5.tar.gz
.
File metadata
- Download URL: adhoc_api-0.2.5.tar.gz
- Upload date:
- Size: 13.2 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: poetry/1.8.3 CPython/3.10.15 Linux/5.18.19-051819-generic
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | c37c20f0eed3f17e58178cf2fd5c226417beb8986fcb8a4327067e53ea5d9f43 |
|
MD5 | f647d2759f39508234dcbc2e4a4fea76 |
|
BLAKE2b-256 | eee4f386cd78f8e0244b6f944417ebf057d462c6a9ad72972216beb857f5d864 |
File details
Details for the file adhoc_api-0.2.5-py3-none-any.whl
.
File metadata
- Download URL: adhoc_api-0.2.5-py3-none-any.whl
- Upload date:
- Size: 16.0 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: poetry/1.8.3 CPython/3.10.15 Linux/5.18.19-051819-generic
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | 4c93e0033759cb4623a771bccd76e85d0e93b2f1b5b0a5ff55f703fd1a5686dd |
|
MD5 | fdababdbe49fd443a0c599b31079c1bb |
|
BLAKE2b-256 | bde25d3c2fc8e527a81b62133521e2a90d3a96904c70db1b5f4f6229e2d97878 |