add StableDiffusionImageSynthesisWrapper test case

cmgzn · cmgzn · commit 6b9d416d8c06 · 2025-01-09T17:25:40.000+08:00
diff --git a/tests/model_test.py b/tests/model_test.py
@@ -28,6 +28,7 @@
     OpenAIChatWrapper,
     PostAPIChatWrapper,
     AnthropicChatWrapper,
+    StableDiffusionImageSynthesisWrapper,
 )
 
 
@@ -79,6 +80,7 @@ def test_build_in_model_wrapper_classes(self) -> None:
                 "litellm_chat": LiteLLMChatWrapper,
                 "yi_chat": YiChatWrapper,
                 "anthropic_chat": AnthropicChatWrapper,
+                "sd_txt2img": StableDiffusionImageSynthesisWrapper,
             },
         )
 

Original file line number	Diff line number	Diff line change
`@@ -28,6 +28,7 @@`
`28`	`28`	`OpenAIChatWrapper,`
`29`	`29`	`PostAPIChatWrapper,`
`30`	`30`	`AnthropicChatWrapper,`
	`31`	`+ StableDiffusionImageSynthesisWrapper,`
`31`	`32`	`)`
`32`	`33`
`33`	`34`
`@@ -79,6 +80,7 @@ def test_build_in_model_wrapper_classes(self) -> None:`
`79`	`80`	`"litellm_chat": LiteLLMChatWrapper,`
`80`	`81`	`"yi_chat": YiChatWrapper,`
`81`	`82`	`"anthropic_chat": AnthropicChatWrapper,`
	`83`	`+ "sd_txt2img": StableDiffusionImageSynthesisWrapper,`
`82`	`84`	`},`
`83`	`85`	`)`
`84`	`86`