MeUtils 2024.3.4.13.4.45__py3-none-any.whl → 2025.1.16.17.15.52__py3-none-any.whl

examples/_openaisdk/openai_gitee.py

@@ -0,0 +1,33 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Project      : AI.  @by PyCharm
+# @File         : openai_gitee
+# @Time         : 2025/1/3 10:34
+# @Author       : betterme
+# @WeChat       : meutils
+# @Software     : PyCharm
+# @Description  : https://ai.gitee.com/hf-models/Kwai-Kolors/Kolors/api
+
+from meutils.pipe import *
+from openai import OpenAI
+from meutils.io.files_utils import to_file
+
+client = OpenAI(
+    base_url="https://ai.gitee.com/v1",
+    api_key="WPCSA3ZYD8KBQQ2ZKTAPVUA059J2Q47TLWGB2ZMQ",
+    # default_headers={"X-Failover-Enabled": "true", "X-Package": "1910"},
+)
+
+print(client.models.list())
+
+response = client.images.generate(
+    model="Kolors",
+    size="1024x1024",
+    # extra_body={
+    #     "num_inference_steps": 25,
+    #     "guidance_scale": 7.5,
+    # },
+    prompt="a white siamese cat",
+)
+
+to_file(response.data[0].b64_json, 'x.png')

examples/_openaisdk/openai_god.py

@@ -0,0 +1,45 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Project      : AI.  @by PyCharm
+# @File         : openai_siliconflow
+# @Time         : 2024/6/26 10:42
+# @Author       : betterme
+# @WeChat       : meutils
+# @Software     : PyCharm
+# @Description  :
+import os
+
+from meutils.pipe import *
+from openai import OpenAI, APIStatusError
+
+client = OpenAI(
+    api_key=os.getenv("GOD_API_KEY"),
+    base_url=os.getenv("GOD_BASE_URL"),
+)
+
+try:
+    completion = client.chat.completions.create(
+        # model="net-gpt-3.5-turbo",
+        # model="net-gpt-3.5-turbo-16k",
+        # model="net-gpt-4o-mini",
+        # model="net-gpt-4o",
+        model="net-claude-1.3-100k",
+        messages=[
+            {"role": "user", "content": "南京天气如何"}
+        ],
+        # top_p=0.7,
+        top_p=None,
+        temperature=None,
+        stream=True,
+        max_tokens=6000
+    )
+except APIStatusError as e:
+    print(e.status_code)
+
+    print(e.response)
+    print(e.message)
+    print(e.code)
+
+for chunk in completion:
+    print(chunk.choices[0].delta.content)
+

examples/_openaisdk/openai_groq.py

@@ -0,0 +1,240 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Project      : AI.  @by PyCharm
+# @File         : openai_tts
+# @Time         : 2024/3/27 12:10
+# @Author       : betterme
+# @WeChat       : meutils
+# @Software     : PyCharm
+# @Description  :
+
+from meutils.pipe import *
+from openai import OpenAI, AsyncOpenAI
+import os
+
+#
+# chat_completion = client.chat.completions.create(
+#     messages=[
+#         {
+#             "role": "system",
+#             "content": "you are a helpful assistant."
+#         },
+#         {
+#             "role": "user",
+#             "content": "Explain the importance of fast language models",
+#         }
+#     ],
+#     model="llama3-8b-8192",
+# )
+#
+# print(chat_completion.choices[0].message.content)
+#
+# base_url = 'http//:0.0.0.0:8000/v1'
+# api_key = 'xx'
+# client = OpenAI(base_url=base_url, api_key=base_url)
+#
+#
+#
+#
+# #!/usr/bin/env python
+# # -*- coding: utf-8 -*-
+# # @Project      : AI.  @by PyCharm
+# # @File         : openai_asr
+# # @Time         : 2023/11/23 13:10
+# # @Author       : betterme
+# # @WeChat       : meutils
+# # @Software     : PyCharm
+# # @Description
+# import io
+# import os
+#
+# import requests
+#
+# from meutils.pipe import *
+from openai import OpenAI
+
+from groq import Groq
+
+client = Groq(
+    # This is the default and can be omitted
+    api_key=os.environ.get("GROQ_API_KEY", "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"),
+    # base_url="https://api.groq.com/openai/v1",
+
+)
+
+client = OpenAI(
+    api_key=os.environ.get("GROQ_API_KEY"),
+    # base_url=os.getenv("GROQ_BASE_URL"),
+    base_url="https://groq.chatfire.cc/v1",
+
+)
+#
+# # # response_format: Literal["json", "text", "srt", "verbose_json", "vtt"]
+# file = open("/Users/betterme/PycharmProjects/AI/MeUtils/meutils/ai_audio/asr/whisper-1719913495729-54f08dde5.wav.mp3.mp3", 'rb')  # 正确
+#
+# with timer():
+#     _ = client.audio.transcriptions.create(
+#         file=file,
+#         model="whisper-large-v3",
+#         # model="hailuo2whisper",
+#         response_format="text",  # ["json", "text", "srt", "verbose_json", "vtt"]
+#         # response_format="srt",  # ["json", "text", "srt", "verbose_json", "vtt"]
+#         # response_format="verbose_json",  # ["json", "text", "srt", "verbose_json", "vtt"]
+#         # response_format="vtt",  # ["json", "text", "srt", "verbose_json", "vtt"]
+#         # response_format="json",  # ["json", "text", "srt", "verbose_json", "vtt"]
+#         # language='ja'
+# # [sv no br mn pa sa ja pt jv ko gu lv sq mg mi sk la ml lb ln ha tr cs sl kn ne mr uz bo ca fi ur bg et eu lo su es nl fa gl km af tg el te ro hu bs si sn ka en fr am fo so yi ta hy sr is zh th kk be as cy bn hr sw mt ru pl uk ms az ht my haw ar he hi vi da lt mk yo de id tl tt yue sd ps tk nn ba it oc]',
+#
+#     )
+#     print(_)
+
+# _ = client.audio.translations.create(
+#     file=file,
+#     model="whisper-large-v3",
+#     # model="hailuo2whisper",
+#     # response_format="text",  # ["json", "text", "srt", "verbose_json", "vtt"]
+#     # response_format="srt",  # ["json", "text", "srt", "verbose_json", "vtt"]
+#     # response_format="verbose_json",  # ["json", "text", "srt", "verbose_json", "vtt"]
+#     # response_format="vtt",  # ["json", "text", "srt", "verbose_json", "vtt"]
+#     response_format="json",  # ["json", "text", "srt", "verbose_json", "vtt"]
+#     prompt="将其翻译成日语"
+#
+# )
+# print(_)
+
+
+# model = "llama-3.1-8b-instant"
+# messages = [
+#     {'role': 'user', 'content': '你是谁'}
+# ]
+# response = client.chat.completions.create(
+#     # model='alibaba/Qwen1.5-110B-Chat',
+#     model=model,
+#     # messages=[
+#     #     {'role': 'user', 'content': "抛砖引玉是什么意思呀"}
+#     # ],
+#     messages=messages,
+#     stream=False
+# )
+#
+# print(response)
+
+
+# print(client.models.list().model_dump_json(indent=4))
+# {
+#     "data": [
+#         {
+#             "id": "gemma2-9b-it",
+#             "created": 1693721698,
+#             "object": "model",
+#             "owned_by": "Google",
+#             "active": true,
+#             "context_window": 8192,
+#             "public_apps": null
+#         },
+#         {
+#             "id": "gemma-7b-it",
+#             "created": 1693721698,
+#             "object": "model",
+#             "owned_by": "Google",
+#             "active": true,
+#             "context_window": 8192,
+#             "public_apps": null
+#         },
+#         {
+#             "id": "llama-3.1-405b-reasoning",
+#             "created": 1693721698,
+#             "object": "model",
+#             "owned_by": "Meta",
+#             "active": true,
+#             "context_window": 131072,
+#             "public_apps": [
+#                 "chat"
+#             ]
+#         },
+#         {
+#             "id": "llama-3.1-70b-versatile",
+#             "created": 1693721698,
+#             "object": "model",
+#             "owned_by": "Meta",
+#             "active": true,
+#             "context_window": 131072,
+#             "public_apps": null
+#         },
+#         {
+#             "id": "llama-3.1-8b-instant",
+#             "created": 1693721698,
+#             "object": "model",
+#             "owned_by": "Meta",
+#             "active": true,
+#             "context_window": 131072,
+#             "public_apps": null
+#         },
+#         {
+#             "id": "llama3-70b-8192",
+#             "created": 1693721698,
+#             "object": "model",
+#             "owned_by": "Meta",
+#             "active": true,
+#             "context_window": 8192,
+#             "public_apps": null
+#         },
+#         {
+#             "id": "llama3-8b-8192",
+#             "created": 1693721698,
+#             "object": "model",
+#             "owned_by": "Meta",
+#             "active": true,
+#             "context_window": 8192,
+#             "public_apps": null
+#         },
+#         {
+#             "id": "llama3-groq-70b-8192-tool-use-preview",
+#             "created": 1693721698,
+#             "object": "model",
+#             "owned_by": "Groq",
+#             "active": true,
+#             "context_window": 8192,
+#             "public_apps": null
+#         },
+#         {
+#             "id": "llama3-groq-8b-8192-tool-use-preview",
+#             "created": 1693721698,
+#             "object": "model",
+#             "owned_by": "Groq",
+#             "active": true,
+#             "context_window": 8192,
+#             "public_apps": null
+#         },
+#         {
+#             "id": "mixtral-8x7b-32768",
+#             "created": 1693721698,
+#             "object": "model",
+#             "owned_by": "Mistral AI",
+#             "active": true,
+#             "context_window": 32768,
+#             "public_apps": null
+#         },
+#         {
+#             "id": "whisper-large-v3",
+#             "created": 1693721698,
+#             "object": "model",
+#             "owned_by": "OpenAI",
+#             "active": true,
+#             "context_window": 1500,
+#             "public_apps": null
+#         }
+#     ],
+#     "object": "list"
+# }
+# from meutils.schemas.openai_types import ChatCompletionRequest
+# from meutils.llm.openai_utils import to_openai_completion_params, token_encoder, token_encoder_with_cache
+#
+# request = ChatCompletionRequest(
+#     model="llama-3.1-8b-instant",
+#     messages=[{'role': 'user', 'content': '你是谁'}])
+#
+# data = to_openai_completion_params(request)
+#
+# data.pop('extra_body')
+# client.chat.completions.create(**data)

examples/_openaisdk/openai_images.py

@@ -0,0 +1,203 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Project      : AI.  @by PyCharm
+# @File         : openai_siliconflow
+# @Time         : 2024/6/26 10:42
+# @Author       : betterme
+# @WeChat       : meutils
+# @Software     : PyCharm
+# @Description  :
+import os
+import langchain
+from meutils.pipe import *
+from openai import OpenAI
+from openai import OpenAI, APIStatusError
+
+from openai import OpenAI
+
+# response = OpenAI().images.generate(
+#     model="stable-diffusion-3",
+#     # model="kling",
+#
+#     prompt="a white siamese cat",
+#     size="1024x1024",
+#     quality="standard",
+#     style="vividx",
+#     n=1,
+#     extra_body={
+#         "guidance_scale": 4.5,
+#         "num_inference_steps": 25,
+#         "negative_prompt": ""
+#     }
+# )
+#
+
+# from openai import OpenAI
+#
+# response = OpenAI().images.generate(
+#     # model="stable-diffusion-3",
+#     model="kling",
+#
+#     prompt="漫天的星河，浪漫，神秘。超高清，最好的质量，超高细节的画质，8K分辨率",
+#     size="1024x1024",
+#     n=1
+# )
+
+
+data = {
+    'prompt': '一只可爱的边牧在坐公交车，卡通贴纸。动漫3D风格，超写实油画，超高分辨率，最好的质量，8k',
+
+    # 'model': 'step-1x-medium',
+    'model': 'seededit',
+
+    # 'n': 2, 'quality': 'hd', 'response_format': 'url', 'size': '1024x1024', 'style': 'vivid',
+    # 'extra_body': {'guidance_scale': 4.5, 'num_inference_steps': 25, 'seed': None, 'negative_prompt': None}
+}
+
+from openai import OpenAI
+
+# model = "stable-diffusion-3"
+# with timer(model):
+#     response = OpenAI(
+#         # base_url="http://0.0.0.0:8000/v1"
+#         # base_url="https://openai-dev.chatfire.cn/images/v1",
+#         # base_url="https://openai.chatfire.cn/images/v1"
+#
+#         # base_url="http://111.173.117.175:40001/images/v1"
+#         # base_url="https://api.stepfun.com/v1",
+#         # api_key="4ewbnANQogxIX8DUY3Cz2VAMlB3ij39cwDrgQfkDtuNAoexu3E7CEJbR8USXVa3pA"
+#     ).images.generate(
+#         model=model,
+#         prompt=prompt,
+#     )
+#     print(response)
+
+# with timer('image'):
+#     model = 'flux-schnell'
+#
+#     response = OpenAI(
+#     ).images.generate(
+#         model=model,
+#         prompt=prompt,
+#     )
+#     print(response)
+#
+# with timer('image'):
+#     model = 'flux-pro'
+#
+#     response = OpenAI(
+#     ).images.generate(
+#         model=model,
+#         prompt=prompt,
+#     )
+#     print(response)
+
+# model = 'step-1x-medium'
+# with timer(model):
+#     response = OpenAI().images.generate(
+#         model=model,
+#         prompt=prompt,
+#     )
+#     print(response)
+#
+# model = 'kling'
+# with timer(model):
+#     response = OpenAI().images.generate(
+#         model=model,
+#         prompt=prompt,
+#     )
+#     print(response)
+#
+
+
+with timer('image'):
+    model = 'flux-pro'
+    model = 'flux-dev'
+    model = 'flux-schnell'
+    # model = "cogview-3"
+
+    # model = "cogview-3-plus"
+    # model = "cogview-3-plus"
+
+    # model = "cogview-3-plus"
+
+    # model = 'step-1x-medium'
+    # model = 'stable-diffusion-xl-base-1.0'
+    # model = "black-forest-labs/flux.1.1-pro"
+
+    # model = "black-forest-labs/FLUX.1-schnell"
+    # model = "black-forest-labs/FLUX.1-dev"
+    # model = "flux-dev"
+
+    client = OpenAI(
+        # api_key=os.getenv("OPENAI_API_KEY"),
+        # api_key='13OafcA8KSNO4XLrBR4DpvgVV9d59KCvPAmEdrtsvOIcU4T50B0EGstjLEqPnBI2N',
+
+        # api_key=os.getenv("SILICONFLOW_API_KEY"),
+        # base_url=os.getenv("SILICONFLOW_BASE_URL"),
+
+        # base_url='https://api.stepfun.com/v1',
+        # base_url="http://0.0.0.0:8000/v1",
+        # base_url='https://api-dev.chatfire.cn/v1',
+        # base_url='https://openai-dev.chatfire.cn/api/v1',
+        # base_url='https://openai.chatfire.cn/api/v1',
+        # base_url="https://openai-dev.chatfire.cn/v1",
+
+        # base_url="http://110.42.51.201:40009/v1"
+
+    )
+
+    # print(client.models.list())
+    # model = 'flux1.0-turbo'
+    # model = 'flux1.0-schnell'
+    # model = 'flux-dev'
+
+    # model = 'flux1.0-dev'
+    # model = 'flux1.0-pro'
+    # model = 'flux1.1-pro'
+    # model = 'kling'
+    # model = 'hunyuan'
+    # model = 'cogview-3'
+    # model = 'stable-diffusion'
+    # model = 'flux.1.1-pro'
+    # model = 'flux'
+    model = 'stable-diffusion-3-5-large'
+    model = 'flux-schnell'
+    model = 'flux-schnell'
+    model = 'seededit'
+    model = 'dall-e-3'
+    model = 'kolors'
+    model = 'images'
+
+    # response = client.images.generate(
+    #     model=model,
+    #     prompt=prompt,
+    #     extra_body={
+    #         "num_inference_steps": 1,
+    #         "prompt_enhancement": True
+    #     },
+    #     # size="1700x1275"
+    # )
+    prompt = '一只可爱的边牧在坐公交车，卡通贴纸。动漫3D风格，超写实油画，超高分辨率，最好的质量，8k'
+    prompt = "一个小女孩举着横幅，上面写着“新年快乐”"
+
+    response = client.images.generate(
+        model=model,
+        prompt=prompt,
+        extra_body={
+            "Style": "True",
+            "controls": {}, ######### 这个看起来更通用些
+            "aspect_ratio": "1",
+
+            "Width": 1024,
+            "Height": 1024,
+            "prompt_enhancement": True,
+        },
+
+        # size="1700x1275"
+    )
+
+    logger.debug(bjson(response.model_dump()))
+
+# openai.AuthenticationError: Error code: 401 - {'error': {'message': 'Incorrect API key provided', 'type': 'invalid_api_key'}}
+# openai.APIStatusError: Error code: 402 - {'error': {'message': 'You exceeded your current quota, please check your plan and billing details', 'type': 'quota_exceeded'}}

examples/_openaisdk/openai_json.py

@@ -0,0 +1,78 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Project      : AI.  @by PyCharm
+# @File         : openai_json
+# @Time         : 2024/4/11 13:10
+# @Author       : betterme
+# @WeChat       : meutils
+# @Software     : PyCharm
+# @Description  :
+
+from meutils.pipe import *
+
+from openai import OpenAI
+
+# base_url = "https://api.chatllm.vip/v1"
+# base_url = "http://0.0.0.0:8000/v1"
+# base_url="https://api.gptgod.online/v1"
+
+
+from openai import OpenAI
+
+
+client = OpenAI(
+    # api_key=os.getenv('OPENAI_API_KEY_OPENAI'),
+)
+
+with timer('聊天模型'):
+    data = {
+        # 'model': 'gpt-3.5-turbo',
+        # 'model': 'gpt-3.5-turbo-0125',
+        # "model": "rag-gpt-3.5-turbo",
+        # "model": "rag-gpt-4",
+        # 'model': 'gpt-4-0125-preview',
+        # 'model': 'gpt-4-turbo-preview',
+        # 'model': 'gemini-pro',
+        'model': 'gpt-4o',
+        # 'model': 'gpt-4o-mini',
+        # 'model': "gpt-4o-2024-08-06",
+
+        # 'model': 'backup-gpt',
+        # 'model': 'test',
+        # 'model': 'per',
+        # 'model': 'glm-4-all',
+        # 'model': '65f41c8a9c0ebbcbe28bb9c1',
+
+        'messages': [
+            {'role': 'system', 'content': 'You are a helpful assistant designed to output JSON.'},
+
+            {'role': 'user', 'content': "请将1改为2 {'a': 1}"}
+
+        ],
+        'stream': False,
+
+        'response_format': {"type": "json_object"}
+
+    }
+
+    try:
+        # r = client.chat.completions.create(**data, extra_body={"file_ids": ["cn2a0s83r07am0knkeag"]})
+        r = client.chat.completions.create(**data)
+
+        print(isinstance(r, Iterator))
+
+        print(type(r))
+
+        for i in r:
+            print(i)
+
+    except Exception as e:
+        print(e)
+
+# 向量模型
+# model = 'text-embedding-3-small'
+# model = 'text-embedding-3-large'
+# r = client.embeddings.create(input='hi', model=model)
+# print(r)
+
+

examples/_openaisdk/openai_lingyi.py

@@ -0,0 +1,59 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Project      : AI.  @by PyCharm
+# @File         : openai_siliconflow
+# @Time         : 2024/6/26 10:42
+# @Author       : betterme
+# @WeChat       : meutils
+# @Software     : PyCharm
+# @Description  :
+import os
+
+from meutils.pipe import *
+from openai import OpenAI
+from openai import OpenAI, APIStatusError
+
+f = lambda m: 'yi-vision-v2' if 'v' in m else 'yi-lightning'
+
+"https://all.chatfire.cn/redirect/{model}/to/lambda m: 'yi-vision-v2' if 'v' in m else 'yi-lightning'?redirect_base_url=https://api.lingyiwanwu.com/v1"
+
+client = OpenAI(
+    # base_url="https://free.chatfire.cn/v1",
+    api_key=os.getenv("LINGYIWANWU_API_KEY"),
+    base_url=os.getenv("LINGYIWANWU_BASE_URL")
+    # api_key="eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJzdWIiOiI3YmFmYWQzYTRmZDU0OTk3YjNmYmNmYjExMjY5NThmZiIsImV4cCI6MTczODAyNDg4MiwibmJmIjoxNzIyNDcyODgyLCJpYXQiOjE3MjI0NzI4ODIsImp0aSI6IjY5Y2ZiNzgzNjRjODQxYjA5Mjg1OTgxYmY4ODMzZDllIiwidWlkIjoiNjVmMDc1Y2E4NWM3NDFiOGU2ZmRjYjEyIiwidHlwZSI6InJlZnJlc2gifQ.u9pIfuQZ7Y00DB6x3rbWYomwQGEyYDSE-814k67SH74",
+    # base_url="https://any2chat.chatfire.cn/glm/v1"
+)
+
+print(client.models.list().to_json())
+try:
+    completion = client.chat.completions.create(
+        model="yi-spark",
+        # model="xxxxxxxxxxxxx",
+        messages=[
+            {"role": "user", "content": "hi"}
+        ],
+        # top_p=0.7,
+        top_p=None,
+        temperature=None,
+        stream=True,
+        max_tokens=11
+    )
+except APIStatusError as e:
+    print(e.status_code)
+
+    print(e.response)
+    print(e.message)
+    print(e.code)
+
+for chunk in completion:
+    print(bjson(chunk))
+    print(chunk.choices[0].delta.content)
+
+# r = client.images.generate(
+#     model="cogview-3-plus",
+#     prompt="a white siamese cat",
+#     size="1024x1024",
+#     quality="hd",
+#     n=1,
+# )