Gorilla: Large Language Model Connected with Massive APIs [Project Website]
๐ข Gorilla is Apache 2.0 With Gorilla being fine-tuned on MPT, and Falcon, you can use Gorilla commercially with no obligations! โณ
Gorilla
enables LLMs to use tools by invoking APIs. Given a natural language query, Gorilla comes up with the semantically- and syntactically- correct API to invoke. With Gorilla, we are the first to demonstrate how to use LLMs to invoke 1,600+ (and growing) API calls accurately while reducing hallucination. We also release APIBench, the largest collection of APIs, curated and easy to be trained on! Join us, as we try to expand the largest API store and teach LLMs how to write them! Hop on our Discord, or open a PR, or email us if you would like to have your API incorporated as well.
- ๐ข [06/06] Released Commercially usable, Apache 2.0 licensed Gorilla models
- ๐ [05/30] Provided the CLI interface to chat with Gorilla!
- ๐ [05/28] Released Torch Hub and TensorFlow Hub Models!
- ๐ [05/27] Released the first Gorilla model!
and ๐ค!
- ๐ฅ [05/27] We released the APIZoo contribution guide for community API contributions!
- ๐ฅ [05/25] We release the APIBench dataset and the evaluation code of Gorilla!
Inference: Run Gorilla locally inference/README.md
Evaluation: We have included prompts and responces for the APIBench with and without retrievers along with the Abstract Syntax Tree (AST) matching evaluation script at evaluation.
Our repository organization is shown below.
- The
data
folder contains all the evaluation APIs(APIBench)
and the community contributed APIs. - The
eval
folder contains all our evaluation code as well as the Gorilla outputs. - The
inference
folder contains all the inference code for running Gorilla locally. - [Coming Soon!] The
train
folder contains all the training code associated with Gorilla finetuning.
For our dataset collections, all the 1640 API documentation is in data/api
. We also include the APIBench
dataset created by self-instruct in data/apibench
. For evaluation, we convert this into a LLM-friendly chat format, and the questions are in eval/eval-data/questions
, and the corresponding responces are in eval/eval-data/responses
. We have also included the evaluation scripts are in eval/eval-scripts
. This would be entirely sufficient to train Gorilla yourself, and reproduce our results. Please see evaluation for the details on how to use our evaluation pipeline.
Additionally, we have released all the model weights. gorilla-7b-hf-v0
lets you invoke over 925 Hugging Face APIs. Similarly, gorilla-7b-tf-v0
and gorilla-7b-th-v0
have 626 (exhaustive) Tensorflow v2, and 94 (exhaustive) Torch Hub APIs. gorilla-mpt-7b-hf-v0
and gorilla-falcon-7b-hf-v0
are Apache 2.0 licensed models (commercially usable) fine-tuned on MPT-7B and Falcon-7B respectively. We will release a model with all three combined with generic chat capability and community contributed APIs as soon as we can scale our serving infrastructure. You can run Gorilla locally from instructions in the inference/
sub-directory, or we also provide a hosted Gorilla chat completion API (see Colab)! If you have any suggestions, or if you run into any issues please feel free to reach out to us either through Discord or email or raise a Github issue.
gorilla
โโโ data
โ โโโ api (TF/HF/TH APIs used in generating apibench)
โ โ โโโ {api_name}_api.jsonl
โ โโโ apibench (Evaluating LLM models) v-1.0
โ โ โโโ {api_name}_train.jsonl, {api_name}_eval.jsonl
| |โโ apizoo (Contributed by the community - evolving)
โ | โโโ username1.json
โ โ โโโ username2.json
โ โ โโโ ...
โโโ eval
โ โโโ README.md
โ โโโ get_llm_responses.py
โ โโโ eval-scripts
โ โ โโโ ast_eval_{api_name}.py
โ โโโ eval-data
โ โ โโโ questions
โ โ โ โโโ API name
โ โ โ โ โโโ questions_{api_name}_{eval_metric}.jsonl
โ โ โโโ responses
โ โ โ โโโ API name
โ โ โ โ โโโ responses_{api_name}_Gorilla_FT_{eval_metric}.jsonl
โ โ โ โ โโโ responses_{api_name}_Gorilla_RT_{eval_metric}.jsonl
โโโ inference
โ โโโ README.md
โ โโโ serve
โ โ โโโ gorilla_cli.py
โ โ โโโ conv_template.py
โโโ train (Coming Soon!)
We aim to build an open-source, one-stop-shop for all APIs, LLMs can interact with! Any suggestions and contributions are welcome! Please see the details on how to contribute. THIS WILL ALWAYS REMAIN OPEN SOURCE.
- I would like to use Gorilla commercially. Is there going to be a Apache 2.0 licensed version?
Yes! We are actively working on it. We will release a Gorilla model with Apache 2.0 license by Jun 5. Please stay tuned, and let us know if you are interested.
- Can we use Gorilla with Langchain, Toolformer, AutoGPT etc?
Absolutely! You've highlighted a great aspect of our tools. Gorilla is an end-to-end model, specifically tailored to serve correct API calls without requiring any additional coding. It's designed to work as part of a wider ecosystem and can be flexibly integrated with other tools.
Langchain, is a versatile developer tool. Its "agents" can efficiently swap in any LLM, Gorilla included, making it a highly adaptable solution for various needs.
AutoGPT, on the other hand, concentrates on the art of prompting GPT series models. It's worth noting that Gorilla, as a fully fine-tuned model, consistently shows remarkable accuracy, and lowers hallucination, outperforming GPT-4 in making specific API calls.
Now, when it comes to ToolFormer, Toolformer zeroes in on a select set of tools, providing specialized functionalities. Gorilla, in contrast, has the capacity to manage thousands of API calls, offering a broader coverage over a more extensive range of tools.
The beauty of these tools truly shines when they collaborate, complementing each other's strengths and capabilities to create an even more powerful and comprehensive solution. This is where your contribution can make a difference. We enthusiastically welcome any inputs to further refine and enhance these tools.
In the immediate future, we plan to release the following:
- Dataset and Eval Code
- Opening up the APIZoo for contributions from community
- Hosted Gorilla LLM chat for HF model APIs [May 27, 2023]
- Release weights for HF model APIs [May 27, 2023]
- Run Gorilla LLM locally [May 28, 2023]
- Release weights for all APIs from APIBench [May 28, 2023]
- Release a commercially usable, Apache 2.0 licensed Gorilla model [Jun 5, 2023]
- Train a model with first batch of community contributed APIs from APIZoo [Jun 7, 2023]
- Release training code [Jun 7, 2023]
- Train SOTA Gorilla LLM with expanded APIBench and APIZoo ๐
Propose a new task you would like to work on ๐คฉ
If you use Gorilla or APIBench, please cite our paper:
@article{patil2023gorilla,
title={Gorilla: Large Language Model Connected with Massive APIs},
author={Shishir G. Patil and Tianjun Zhang and Xin Wang and Joseph E. Gonzalez},
year={2023},
journal={arXiv preprint arXiv:2305.15334},
}