Skip to content

Commit 00105ae

Browse files
Merge pull request #26 from andrewyng/rename-package
New name
2 parents 1413874 + 97ec95b commit 00105ae

38 files changed

+50
-50
lines changed

CONTRIBUTING.md

Lines changed: 16 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
<!-- omit in toc -->
2-
# Contributing to aimodels
2+
# Contributing to aisuite
33

44
First off, thanks for taking the time to contribute!
55

@@ -33,15 +33,15 @@ and smooth out the experience for all involved. The community looks forward to y
3333
## I Have a Question
3434

3535
> If you want to ask a question, we assume that you have read the available
36-
> [Documentation](https://github.com/andrewyng/aimodels/blob/main/README.md).
36+
> [Documentation](https://github.com/andrewyng/aisuite/blob/main/README.md).
3737
38-
Before you ask a question, it is best to search for existing [Issues](https://github.com/andrewyng/aimodels/issues)
38+
Before you ask a question, it is best to search for existing [Issues](https://github.com/andrewyng/aisuite/issues)
3939
that might help you. In case you have found a suitable issue and still need clarification, you can write
4040
your question in this issue. It is also advisable to search the internet for answers first.
4141

4242
If you then still feel the need to ask a question and need clarification, we recommend the following:
4343

44-
- Open an [Issue](https://github.com/andrewyng/aimodels/issues/new).
44+
- Open an [Issue](https://github.com/andrewyng/aisuite/issues/new).
4545
- Provide as much context as you can about what you're running into.
4646
- Provide project and platform versions (python, OS, etc.), depending on what seems relevant.
4747

@@ -66,10 +66,10 @@ complete the following steps in advance to help us fix any potential bug as fast
6666

6767
- Make sure that you are using the latest version.
6868
- Determine if your bug is really a bug and not an error on your side e.g. using incompatible environment
69-
components/versions (Make sure that you have read the [documentation](https://github.com/andrewyng/aimodels/blob/main/README.md).
69+
components/versions (Make sure that you have read the [documentation](https://github.com/andrewyng/aisuite/blob/main/README.md).
7070
If you are looking for support, you might want to check [this section](#i-have-a-question)).
7171
- To see if other users have experienced (and potentially already solved) the same issue you are having,
72-
check if there is not already a bug report existing for your bug or error in the [bug tracker](https://github.com/andrewyng/aimodels?q=label%3Abug).
72+
check if there is not already a bug report existing for your bug or error in the [bug tracker](https://github.com/andrewyng/aisuite?q=label%3Abug).
7373
- Also make sure to search the internet (including Stack Overflow) to see if users outside of the GitHub
7474
community have discussed the issue.
7575
- Collect information about the bug:
@@ -89,7 +89,7 @@ complete the following steps in advance to help us fix any potential bug as fast
8989
9090
We use GitHub issues to track bugs and errors. If you run into an issue with the project:
9191

92-
- Open an [Issue](https://github.com/andrewyng/aimodels/issues/new). (Since we can't be sure at
92+
- Open an [Issue](https://github.com/andrewyng/aisuite/issues/new). (Since we can't be sure at
9393
this point whether it is a bug or not, we ask you not to talk about a bug yet and not to label the issue.)
9494
- Explain the behavior you would expect and the actual behavior.
9595
- Please provide as much context as possible and describe the *reproduction steps* that someone else can
@@ -112,31 +112,31 @@ Please use the issue templates provided.
112112

113113
### Suggesting Enhancements
114114

115-
This section guides you through submitting an enhancement suggestion for aimodels,
115+
This section guides you through submitting an enhancement suggestion for aisuite,
116116
**including completely new features and minor improvements to existing functionality**. Following these
117117
guidelines will help maintainers and the community to understand your suggestion and find related suggestions.
118118

119119
<!-- omit in toc -->
120120
#### Before Submitting an Enhancement
121121

122122
- Make sure that you are using the latest version.
123-
- Read the [documentation](https://github.com/andrewyng/aimodels/blob/main/README.md) carefully
123+
- Read the [documentation](https://github.com/andrewyng/aisuite/blob/main/README.md) carefully
124124
and find out if the functionality is already covered, maybe by an individual configuration.
125-
- Perform a [search](https://github.com/andrewyng/aimodels/issues) to see if the enhancement has
125+
- Perform a [search](https://github.com/andrewyng/aisuite/issues) to see if the enhancement has
126126
already been suggested. If it has, add a comment to the existing issue instead of opening a new one.
127127
- Find out whether your idea fits with the scope and aims of the project. It's up to you to make a strong
128128
case to convince the project's developers of the merits of this feature. Keep in mind that we want features that will be useful to the majority of our users and not just a small subset. If you're just targeting a minority of users, consider writing an add-on/plugin library.
129129

130130
<!-- omit in toc -->
131131
#### How Do I Submit a Good Enhancement Suggestion?
132132

133-
Enhancement suggestions are tracked as [GitHub issues](https://github.com/andrewyng/aimodels/issues).
133+
Enhancement suggestions are tracked as [GitHub issues](https://github.com/andrewyng/aisuite/issues).
134134

135135
- Use a **clear and descriptive title** for the issue to identify the suggestion.
136136
- Provide a **step-by-step description of the suggested enhancement** in as many details as possible.
137137
- **Describe the current behavior** and **explain which behavior you expected to see instead** and why.
138138
At this point you can also tell which alternatives do not work for you.
139-
- **Explain why this enhancement would be useful** to most aimodels users. You may also want to
139+
- **Explain why this enhancement would be useful** to most aisuite users. You may also want to
140140
point out the other projects that solved it better and which could serve as inspiration.
141141

142142

@@ -145,10 +145,10 @@ Enhancement suggestions are tracked as [GitHub issues](https://github.com/andrew
145145
#### Pre-requisites
146146

147147
You should first [fork](https://docs.github.com/en/pull-requests/collaborating-with-pull-requests/working-with-forks/fork-a-repo)
148-
the `aimodels` repository and then clone your forked repository:
148+
the `aisuite` repository and then clone your forked repository:
149149

150150
```bash
151-
git clone https://github.com/<YOUR_GITHUB_USER>/aimodels.git
151+
git clone https://github.com/<YOUR_GITHUB_USER>/aisuite.git
152152
```
153153

154154

@@ -164,7 +164,7 @@ Please install the development and test dependencies:
164164
poetry install --with dev,test
165165
```
166166

167-
`aimodels` uses pre-commit to ensure the formatting is consistent:
167+
`aisuite` uses pre-commit to ensure the formatting is consistent:
168168
```bash
169169
pre-commit install
170170
```
@@ -185,7 +185,7 @@ pass, so please make sure it passes locally.
185185

186186

187187
#### Testing
188-
`aimodels` tracks unit tests. Pytest is used to execute said unit tests in `tests/`:
188+
`aisuite` tracks unit tests. Pytest is used to execute said unit tests in `tests/`:
189189

190190
```bash
191191
poetry run pytest tests

README.md

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,18 +1,18 @@
1-
# aimodels
1+
# aisuite
22

33
[![Code style: black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black)
44

55
Simple, unified interface to multiple Generative AI providers.
66

7-
`aimodels` is an tool designed for researchers who need to evaluate and compare the responses of
8-
multiple LLMs through a standardized interface. Based on the OpenAI interface standard, `aimodels`
7+
`aisuite` is an tool designed for researchers who need to evaluate and compare the responses of
8+
multiple LLMs through a standardized interface. Based on the OpenAI interface standard, `aisuite`
99
makes it easy to interact with the most popular LLMs and compare the results of their chat based
1010
functionality, with support for more interfaces coming in the near future.
1111

1212
## Installation
1313

1414
```shell
15-
pip install aimodels
15+
pip install aisuite
1616
```
1717

1818
## Set up
@@ -44,7 +44,7 @@ pip install openai anthropic
4444
In your python code:
4545

4646
```python
47-
import aimodels as ai
47+
import aisuite as ai
4848
client = ai.Client()
4949

5050
models = ["openai:gpt-4o", "anthropic:claude-3-5-sonnet-20240620"]
@@ -67,12 +67,12 @@ for model in models:
6767
For more examples, check out the `examples` directory where you will find several
6868
notebooks that you can run to experiment with the interface.
6969

70-
The current list of supported providers can be found in the `aimodels.providers`
70+
The current list of supported providers can be found in the `aisuite.providers`
7171
package.
7272

7373
## License
7474

75-
aimodels is released under the MIT License. You are free to use, modify, and distribute
75+
aisuite is released under the MIT License. You are free to use, modify, and distribute
7676
the code for both commercial and non-commercial purposes.
7777

7878
## Contributing
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.

aimodels/framework/chat_completion_response.py renamed to aisuite/framework/chat_completion_response.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
from aimodels.framework.choice import Choice
1+
from aisuite.framework.choice import Choice
22

33

44
class ChatCompletionResponse:

aimodels/framework/choice.py renamed to aisuite/framework/choice.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
from aimodels.framework.message import Message
1+
from aisuite.framework.message import Message
22

33

44
class Choice:
File renamed without changes.
File renamed without changes.

aimodels/providers/anthropic_interface.py renamed to aisuite/providers/anthropic_interface.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22

33
import os
44

5-
from aimodels.framework import ProviderInterface, ChatCompletionResponse
5+
from aisuite.framework import ProviderInterface, ChatCompletionResponse
66

77

88
class AnthropicInterface(ProviderInterface):

aimodels/providers/google_interface.py renamed to aisuite/providers/google_interface.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
"""The interface to Google's Vertex AI."""
22

33
import os
4-
from aimodels.framework import ProviderInterface, ChatCompletionResponse
4+
from aisuite.framework import ProviderInterface, ChatCompletionResponse
55

66

77
class GoogleInterface(ProviderInterface):

aimodels/providers/mistral_interface.py renamed to aisuite/providers/mistral_interface.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import os
22

3-
from aimodels.framework import ProviderInterface
3+
from aisuite.framework import ProviderInterface
44

55

66
class MistralInterface(ProviderInterface):

aimodels/providers/ollama_interface.py renamed to aisuite/providers/ollama_interface.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
"""The interface to the Ollama API."""
22

3-
from aimodels.framework import ProviderInterface, ChatCompletionResponse
3+
from aisuite.framework import ProviderInterface, ChatCompletionResponse
44
from httpx import ConnectError
55
import os
66

examples/RAG.ipynb

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -87,11 +87,11 @@
8787
"name": "stderr",
8888
"output_type": "stream",
8989
"text": [
90-
"/Users/ksolo/Library/Caches/pypoetry/virtualenvs/aimodels-HUywTnIy-py3.12/lib/python3.12/site-packages/sentence_transformers/evaluation/SentenceEvaluator.py:81: SyntaxWarning: invalid escape sequence '\\g'\n",
90+
"/Users/ksolo/Library/Caches/pypoetry/virtualenvs/aisuite-HUywTnIy-py3.12/lib/python3.12/site-packages/sentence_transformers/evaluation/SentenceEvaluator.py:81: SyntaxWarning: invalid escape sequence '\\g'\n",
9191
" return re.sub(r\"([a-z])([A-Z])\", \"\\g<1> \\g<2>\", class_name)\n",
92-
"/Users/ksolo/Library/Caches/pypoetry/virtualenvs/aimodels-HUywTnIy-py3.12/lib/python3.12/site-packages/sentence_transformers/model_card.py:524: SyntaxWarning: invalid escape sequence '\\d'\n",
92+
"/Users/ksolo/Library/Caches/pypoetry/virtualenvs/aisuite-HUywTnIy-py3.12/lib/python3.12/site-packages/sentence_transformers/model_card.py:524: SyntaxWarning: invalid escape sequence '\\d'\n",
9393
" if dataset_name and re.match(\"_dataset_\\d+\", dataset_name):\n",
94-
"/Users/ksolo/Library/Caches/pypoetry/virtualenvs/aimodels-HUywTnIy-py3.12/lib/python3.12/site-packages/sentence_transformers/losses/DenoisingAutoEncoderLoss.py:16: SyntaxWarning: invalid escape sequence '\\_'\n",
94+
"/Users/ksolo/Library/Caches/pypoetry/virtualenvs/aisuite-HUywTnIy-py3.12/lib/python3.12/site-packages/sentence_transformers/losses/DenoisingAutoEncoderLoss.py:16: SyntaxWarning: invalid escape sequence '\\_'\n",
9595
" \"\"\"\n"
9696
]
9797
}
@@ -157,7 +157,7 @@
157157
"\n",
158158
"prompt = f'Given the following data, Please answer the question: \\n\\n ##question \\n {question}\\n\\n ##context \\n {context}'\n",
159159
"\n",
160-
"import aimodels as ai\n",
160+
"import aisuite as ai\n",
161161
"client = ai.Client()\n",
162162
"\n",
163163
"messages = [\n",

examples/client.ipynb

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@
4040
],
4141
"source": [
4242
"import sys\n",
43-
"sys.path.append('../../aimodels')\n",
43+
"sys.path.append('../../aisuite')\n",
4444
"\n",
4545
"from dotenv import load_dotenv, find_dotenv\n",
4646
"\n",
@@ -77,7 +77,7 @@
7777
},
7878
"outputs": [],
7979
"source": [
80-
"import aimodels as ai\n",
80+
"import aisuite as ai\n",
8181
"\n",
8282
"client = ai.Client()\n",
8383
"\n",

examples/llm_reasoning.ipynb

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -69,7 +69,7 @@
6969
],
7070
"source": [
7171
"import sys\n",
72-
"sys.path.append('../../aimodels')\n",
72+
"sys.path.append('../../aisuite')\n",
7373
"\n",
7474
"from dotenv import load_dotenv, find_dotenv\n",
7575
"\n",
@@ -102,7 +102,7 @@
102102
"metadata": {},
103103
"outputs": [],
104104
"source": [
105-
"import aimodels as ai\n",
105+
"import aisuite as ai\n",
106106
"\n",
107107
"client = ai.Client()"
108108
]

guides/google.md

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
# Google (Vertex) AI
22

3-
To use Google (Vertex) AI with the `aimodels` library, you'll first need to create a Google Cloud account and set up your environment to work with Google Cloud.
3+
To use Google (Vertex) AI with the `aisuite` library, you'll first need to create a Google Cloud account and set up your environment to work with Google Cloud.
44

55
## Create a Google Cloud Account and Project
66

@@ -26,7 +26,7 @@ Set the `GOOGLE_REGION` environment variable to the ID of your project. You can
2626

2727
## Create a Service Account For API Access
2828

29-
Because `aimodels` needs to authenticate with Google Cloud to access the Vertex AI API, you'll need to create a service account and set the `GOOGLE_APPLICATION_CREDENTIALS` environment variable to the path of a JSON file containing the service account's credentials, which you can download from the Google Cloud Console.
29+
Because `aisuite` needs to authenticate with Google Cloud to access the Vertex AI API, you'll need to create a service account and set the `GOOGLE_APPLICATION_CREDENTIALS` environment variable to the path of a JSON file containing the service account's credentials, which you can download from the Google Cloud Console.
3030

3131
This is documented [here](https://cloud.google.com/docs/authentication/provide-credentials-adc#how-to), and the basic steps are as follows:
3232

@@ -71,7 +71,7 @@ pip install vertexai
7171
In your code:
7272

7373
```python
74-
import aimodels as ai
74+
import aisuite as ai
7575
client = ai.Client()
7676

7777
model="vertex:gemini-1.5-pro-001"

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
[tool.poetry]
2-
name = "aimodels"
2+
name = "aisuite"
33
version = "0.1.0"
44
description = ""
55
authors = ["Andrew Ng"]

tests/client/test_client.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
import pytest
2-
from aimodels.client.client import Client, AnthropicInterface
2+
from aisuite.client.client import Client, AnthropicInterface
33

44

55
def test_get_provider_interface_with_new_instance():

tests/providers/test_anthropic_interface.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import pytest
22
from unittest.mock import patch, MagicMock
3-
from aimodels.providers.anthropic_interface import AnthropicInterface
3+
from aisuite.providers.anthropic_interface import AnthropicInterface
44

55

66
@pytest.fixture(autouse=True)

tests/providers/test_fireworks_interface.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import pytest
22
from unittest.mock import patch, MagicMock
3-
from aimodels.providers.fireworks_interface import FireworksInterface
3+
from aisuite.providers.fireworks_interface import FireworksInterface
44

55

66
@pytest.fixture(autouse=True)

tests/providers/test_google_interface.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import pytest
22
from unittest.mock import patch, MagicMock
3-
from aimodels.providers.google_interface import GoogleInterface
3+
from aisuite.providers.google_interface import GoogleInterface
44
from vertexai.generative_models import Content, Part
55

66

tests/providers/test_groq_interface.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import pytest
22
from unittest.mock import patch, MagicMock
3-
from aimodels.providers.groq_interface import GroqInterface
3+
from aisuite.providers.groq_interface import GroqInterface
44

55

66
@pytest.fixture(autouse=True)

tests/providers/test_mistral_interface.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33

44
from mistralai.models.chat_completion import ChatMessage
55

6-
from aimodels.providers.mistral_interface import MistralInterface
6+
from aisuite.providers.mistral_interface import MistralInterface
77

88

99
@pytest.fixture(autouse=True)

tests/providers/test_ollama_interface.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import pytest
22
from unittest.mock import patch, MagicMock
3-
from aimodels.providers.ollama_interface import OllamaInterface
3+
from aisuite.providers.ollama_interface import OllamaInterface
44
from httpx import ConnectError
55

66

tests/providers/test_openai_interface.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import pytest
22
from unittest.mock import patch, MagicMock
3-
from aimodels.providers.openai_interface import OpenAIInterface
3+
from aisuite.providers.openai_interface import OpenAIInterface
44

55

66
@pytest.fixture(autouse=True)

tests/providers/test_replicate_interface.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import pytest
22
from unittest.mock import patch, MagicMock
3-
from aimodels.providers.replicate_interface import ReplicateInterface
3+
from aisuite.providers.replicate_interface import ReplicateInterface
44

55

66
@pytest.fixture(autouse=True)

0 commit comments

Comments
 (0)