XFE Git
XFE Studio Git
Git 首页 全局搜索
XFE 主站 文档 NuGet
公开
关注 0 Fork 0 Star 1
返回提交历史

XFEstudio/gpt4free

Add variant example

151f8b8b
Heiner Lohaus <hlohaus@users.noreply.github.com>
提交于

代码差异

4 个文件 +20 -5
Modified README.md +3 -0
@@ -227,6 +227,9 @@ docker-compose down
227 227 ## 💡 Usage
228 228
229 229 ### New Client with Image Generation
230
231 #### not jet released
232
230 233 ```python
231 234 from g4f.client import Client
232 235
Added docs/cat.webp +0 -0
二进制文件已变更,无法进行逐行预览。
Modified docs/client.md +6 -0
@@ -68,4 +68,10 @@ response = client.images.create_variation(
68 68 image_url = response.data[0].url
69 69 ```
70 70
71 Orginal:
72 [![Image with cat](/docs/cat.jpeg)](/docs/client.md)
73
74 Variant:
75 [![Image with cat](/docs/cat.webp)](/docs/client.md)
76
71 77 [to Home](/docs/client.md)
Modified g4f/client.py +11 -5
@@ -241,7 +241,8 @@ class Images():
241 241 "",
242 242 [{"role": "user", "content": prompt}],
243 243 True,
244 proxy=self.client.get_proxy()
244 proxy=self.client.get_proxy(),
245 **kwargs
245 246 )
246 247 else:
247 248 response = provider.create(prompt)
@@ -253,15 +254,20 @@ class Images():
253 254
254 255 def create_variation(self, image: ImageType, model: str = None, **kwargs):
255 256 provider = self.models.get(model) if model else self.provider
256 if isinstance(provider, BaseProvider):
257 result = None
258 if isinstance(provider, type) and issubclass(provider, BaseProvider):
257 259 response = provider.create_completion(
258 260 "",
259 261 [{"role": "user", "content": "create a image like this"}],
260 262 True,
261 263 image=image,
262 proxy=self.client.get_proxy()
264 proxy=self.client.get_proxy(),
265 **kwargs
263 266 )
264 267 for chunk in response:
265 268 if isinstance(chunk, ImageProviderResponse):
266 return ImagesResponse([Image(image)for image in list(chunk.images)])
267 raise NoImageResponseError()
269 result = ([chunk.images] if isinstance(chunk.images, str) else chunk.images)
270 result = ImagesResponse([Image(image)for image in result])
271 if result is None:
272 raise NoImageResponseError()
273 return result