aljaz/Auto-GPT
1
0
Fork 0
mirror of https://github.com/aljazceru/Auto-GPT.git synced 2025-12-21 16:04:21 +01:00
Code Issues Packages Projects Releases Wiki Activity

Organize all the llm stuff into a subpackage (#3436)

Browse Source
This commit is contained in:
James Collins
2023-04-28 12:00:54 -07:00
committed by GitHub
parent ee4043ae19
commit 3b74d2150e
31 changed files with 59 additions and 51 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/test_llm_utils.py
View File

@@ -1,8 +1,8 @@
import pytest
from openai.error import APIError, RateLimitError
from autogpt.llm_utils import get_ada_embedding, retry_openai_api
from autogpt.modelsinfo import COSTS
from autogpt.llm import COSTS, get_ada_embedding
from autogpt.llm.llm_utils import retry_openai_api
@pytest.fixture(params=[RateLimitError, APIError])
@@ -19,7 +19,7 @@ def mock_create_embedding(mocker):
mock_response.usage.prompt_tokens = 5
mock_response.__getitem__.side_effect = lambda key: [{"embedding": [0.1, 0.2, 0.3]}]
return mocker.patch(
"autogpt.llm_utils.create_embedding", return_value=mock_response
"autogpt.llm.llm_utils.create_embedding", return_value=mock_response
)