apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-06-20 15:38:09 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

upgrade tiktoken to support azure

Browse source
This commit is contained in:
seehi 2023-12-06 16:06:17 +08:00
parent a617aab65b
commit ad347e0717
6 changed files with 50 additions and 45 deletions
Download patch file Download diff file Expand all files Collapse all files

12
tests/metagpt/provider/test_openai.py
View file

@ -118,7 +118,8 @@ class TestOpenAI:
def test_make_client_kwargs_without_proxy(self, config):
instance = OpenAIGPTAPI()
kwargs, async_kwargs = instance._make_client_kwargs(config)
instance.config = config
kwargs, async_kwargs = instance._make_client_kwargs()
assert kwargs == {"api_key": "test_key", "base_url": "test_url"}
assert async_kwargs == {"api_key": "test_key", "base_url": "test_url"}
assert "http_client" not in kwargs
@ -126,7 +127,8 @@ class TestOpenAI:
def test_make_client_kwargs_without_proxy_azure(self, config_azure):
instance = OpenAIGPTAPI()
kwargs, async_kwargs = instance._make_client_kwargs(config_azure)
instance.config = config_azure
kwargs, async_kwargs = instance._make_client_kwargs()
assert kwargs == {"api_key": "test_key", "api_version": "test_version", "azure_endpoint": "test_url"}
assert async_kwargs == {"api_key": "test_key", "api_version": "test_version", "azure_endpoint": "test_url"}
assert "http_client" not in kwargs
@ -134,12 +136,14 @@ class TestOpenAI:
def test_make_client_kwargs_with_proxy(self, config_proxy):
instance = OpenAIGPTAPI()
kwargs, async_kwargs = instance._make_client_kwargs(config_proxy)
instance.config = config_proxy
kwargs, async_kwargs = instance._make_client_kwargs()
assert "http_client" in kwargs
assert "http_client" in async_kwargs
def test_make_client_kwargs_with_proxy_azure(self, config_azure_proxy):
instance = OpenAIGPTAPI()
kwargs, async_kwargs = instance._make_client_kwargs(config_azure_proxy)
instance.config = config_azure_proxy
kwargs, async_kwargs = instance._make_client_kwargs()
assert "http_client" in kwargs
assert "http_client" in async_kwargs