remove attribute openai_vision_model and vision_max_tokens and method test_generate_webpages_with_style_and_script

This commit is contained in:
mannaandpoem 2024-02-06 09:07:29 +08:00
parent 5abde78767
commit 675b96b0f5
2 changed files with 1 additions and 13 deletions

View file

@ -75,8 +75,6 @@ class Config(CLIParams, YamlModel):
iflytek_api_key: str = ""
azure_tts_subscription_key: str = ""
azure_tts_region: str = ""
openai_vision_model: str = "gpt-4-vision-preview"
vision_max_tokens: int = 4096
@classmethod
def from_home(cls, path):

View file

@ -47,17 +47,7 @@ def image_path():
@pytest.mark.asyncio
async def test_generate_webpages_with_styles_and_scripts(mock_webpage_filename_with_styles_and_scripts, image_path):
generator = GPTvGenerator()
rsp = await generator.generate_webpages(image_path=image_path)
logs.logger.info(rsp)
assert "html" in rsp
assert "css" in rsp
assert "javascript" in rsp
@pytest.mark.asyncio
async def test_generate_webpages_with_style_and_script(mock_webpage_filename_with_style_and_script, image_path):
async def test_generate_webpages(mock_webpage_filename_with_styles_and_scripts, image_path):
generator = GPTvGenerator()
rsp = await generator.generate_webpages(image_path=image_path)
logs.logger.info(rsp)