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

fixbug: azure call function

Browse source
This commit is contained in:
莘权 马 2023-12-07 15:51:12 +08:00
parent deda409ab1
commit 5394da6d37
6 changed files with 36 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

27
tests/metagpt/test_gpt.py
View file

@ -5,9 +5,10 @@
@Author : alexanderwu
@File : test_gpt.py
"""
import openai
import pytest
from metagpt.config import CONFIG
from metagpt.logs import logger
@ -18,14 +19,17 @@ class TestGPT:
logger.info(answer)
assert len(answer) > 0
# def test_gptapi_ask_batch(self, llm_api):
# answer = llm_api.ask_batch(['请扮演一个Google Python专家工程师，如果理解，回复明白', '写一个hello world'])
# assert len(answer) > 0
def test_gptapi_ask_batch(self, llm_api):
answer = llm_api.ask_batch(["请扮演一个Google Python专家工程师，如果理解，回复明白", "写一个hello world"])
assert len(answer) > 0
def test_llm_api_ask_code(self, llm_api):
answer = llm_api.ask_code(["请扮演一个Google Python专家工程师，如果理解，回复明白", "写一个hello world"])
logger.info(answer)
assert len(answer) > 0
try:
answer = llm_api.ask_code(["请扮演一个Google Python专家工程师，如果理解，回复明白", "写一个hello world"])
logger.info(answer)
assert len(answer) > 0
except openai.NotFoundError:
assert CONFIG.openai_api_type == "azure"
@pytest.mark.asyncio
async def test_llm_api_aask(self, llm_api):
@ -35,9 +39,12 @@ class TestGPT:
@pytest.mark.asyncio
async def test_llm_api_aask_code(self, llm_api):
answer = await llm_api.aask_code(["请扮演一个Google Python专家工程师，如果理解，回复明白", "写一个hello world"])
logger.info(answer)
assert len(answer) > 0
try:
answer = await llm_api.aask_code(["请扮演一个Google Python专家工程师，如果理解，回复明白", "写一个hello world"])
logger.info(answer)
assert len(answer) > 0
except openai.NotFoundError:
assert CONFIG.openai_api_type == "azure"
@pytest.mark.asyncio
async def test_llm_api_costs(self, llm_api):