Skip to content

Commit

Permalink
add full provider names
Browse files Browse the repository at this point in the history
  • Loading branch information
@pkelaita
pkelaita committed May 2, 2024
1 parent c34e22f commit 8e7876f
Show file tree
Hide file tree
Showing 3 changed files with 26 additions and 19 deletions.
32 changes: 16 additions & 16 deletions README.md
View file
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
# L2M2: A Simple Python LLM Manager 💬👍

[![Tests](https://github.com/pkelaita/l2m2/actions/workflows/tests.yml/badge.svg?timestamp=1714634875)](https://github.com/pkelaita/l2m2/actions/workflows/tests.yml) [![PyPI version](https://badge.fury.io/py/l2m2.svg?timestamp=1714634875)](https://badge.fury.io/py/l2m2)
[![Tests](https://github.com/pkelaita/l2m2/actions/workflows/tests.yml/badge.svg?timestamp=1714636672)](https://github.com/pkelaita/l2m2/actions/workflows/tests.yml) [![PyPI version](https://badge.fury.io/py/l2m2.svg?timestamp=1714636672)](https://badge.fury.io/py/l2m2)

**L2M2** ("LLM Manager" → "LLMM" → "L2M2") is a very simple LLM manager for Python that exposes lots of models through a unified API. This is useful for evaluation, demos, and other apps that need to easily be model-agnostic.

Expand All @@ -14,21 +14,21 @@

L2M2 currently supports the following models:

| Model Name | Provider(s) | Model Version(s) |
| ------------------ | ---------------------------------------------------------------------- | -------------------------------------------------- |
| `gpt-4-turbo` | [`openai`](https://openai.com/product) | `gpt-4-turbo-2024-04-09` |
| `gpt-4-turbo-0125` | [`openai`](https://openai.com/product) | `gpt-4-0125-preview` |
| `gemini-1.5-pro` | [`google`](https://ai.google.dev/) | `gemini-1.5-pro-latest` |
| `gemini-1.0-pro` | [`google`](https://ai.google.dev/) | `gemini-1.0-pro-latest` |
| `claude-3-opus` | [`anthropic`](https://www.anthropic.com/api) | `claude-3-opus-20240229` |
| `claude-3-sonnet` | [`anthropic`](https://www.anthropic.com/api) | `claude-3-sonnet-20240229` |
| `claude-3-haiku` | [`anthropic`](https://www.anthropic.com/api) | `claude-3-haiku-20240307` |
| `command-r` | [`cohere`](https://docs.cohere.com/) | `command-r` |
| `command-r-plus` | [`cohere`](https://docs.cohere.com/) | `command-r-plus` |
| `mixtral-8x7b` | [`groq`](https://wow.groq.com/) | `mixtral-8x7b-32768` |
| `gemma-7b` | [`groq`](https://wow.groq.com/) | `gemma-7b-it` |
| `llama3-8b` | [`groq`](https://wow.groq.com/), [`replicate`](https://replicate.com/) | `llama3-8b-8192`, `meta/meta-llama-3-8b-instruct` |
| `llama3-70b` | [`groq`](https://wow.groq.com/), [`replicate`](https://replicate.com/) | `llama3-70b-8192`, `meta/meta-llama-3-8b-instruct` |
| Model Name | Provider(s) | Model Version(s) |
| ------------------ | ------------------------------------------------------------------ | -------------------------------------------------- |
| `gpt-4-turbo` | [OpenAI](https://openai.com/product) | `gpt-4-turbo-2024-04-09` |
| `gpt-4-turbo-0125` | [OpenAI](https://openai.com/product) | `gpt-4-0125-preview` |
| `gemini-1.5-pro` | [Google](https://ai.google.dev/) | `gemini-1.5-pro-latest` |
| `gemini-1.0-pro` | [Google](https://ai.google.dev/) | `gemini-1.0-pro-latest` |
| `claude-3-opus` | [Anthropic](https://www.anthropic.com/api) | `claude-3-opus-20240229` |
| `claude-3-sonnet` | [Anthropic](https://www.anthropic.com/api) | `claude-3-sonnet-20240229` |
| `claude-3-haiku` | [Anthropic](https://www.anthropic.com/api) | `claude-3-haiku-20240307` |
| `command-r` | [Cohere](https://docs.cohere.com/) | `command-r` |
| `command-r-plus` | [Cohere](https://docs.cohere.com/) | `command-r-plus` |
| `mixtral-8x7b` | [Groq](https://wow.groq.com/) | `mixtral-8x7b-32768` |
| `gemma-7b` | [Groq](https://wow.groq.com/) | `gemma-7b-it` |
| `llama3-8b` | [Groq](https://wow.groq.com/), [Replicate](https://replicate.com/) | `llama3-8b-8192`, `meta/meta-llama-3-8b-instruct` |
| `llama3-70b` | [Groq](https://wow.groq.com/), [Replicate](https://replicate.com/) | `llama3-70b-8192`, `meta/meta-llama-3-8b-instruct` |

You can also call any language model from the above providers that L2M2 doesn't officially support, without guarantees of well-defined behavior.

Expand Down
6 changes: 6 additions & 0 deletions l2m2/model_info.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -7,21 +7,27 @@

PROVIDER_INFO = {
"openai": {
"name": "OpenAI",
"homepage": "https://openai.com/product",
},
"google": {
"name": "Google",
"homepage": "https://ai.google.dev/",
},
"anthropic": {
"name": "Anthropic",
"homepage": "https://www.anthropic.com/api",
},
"cohere": {
"name": "Cohere",
"homepage": "https://docs.cohere.com/",
},
"groq": {
"name": "Groq",
"homepage": "https://wow.groq.com/",
},
"replicate": {
"name": "Replicate",
"homepage": "https://replicate.com/",
},
}
Expand Down
7 changes: 4 additions & 3 deletions scripts/create_model_table.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -11,9 +11,10 @@
rows = ""


def get_provider_link(provider_name):
provider_homepage = PROVIDER_INFO[provider_name]["homepage"]
return f"[`{provider_name}`]({provider_homepage})"
def get_provider_link(provider_key):
provider_name = PROVIDER_INFO[provider_key]["name"]
provider_homepage = PROVIDER_INFO[provider_key]["homepage"]
return f"[{provider_name}]({provider_homepage})"


def make_row(model_name):
Expand Down

0 comments on commit 8e7876f

Please sign in to comment.