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

examples/_openaisdk/demo.py

@@ -0,0 +1,77 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Project      : AI.  @by PyCharm
+# @File         : gpt4all
+# @Time         : 2024/1/2 13:28
+# @Author       : betterme
+# @WeChat       : meutils
+# @Software     : PyCharm
+# @Description  :
+
+from meutils.pipe import *
+from openai import OpenAI, _base_client
+
+# _base_client.BaseClient.default_headers = {}
+
+base_url = "https://api.chatllm.vip/v1"  # 更快
+base_url = "https://api.chatfire.cn/v1"
+base_url = "https://chat.aifree.best/api/openai/v1"
+
+
+# base_url="https://api.gptgod.online/v1"
+
+
+class MeOpenAI(OpenAI):
+    # @default_headers.default_headers.setter
+    # def default_headers(self, value):
+    #     self._default_headers = value
+    @property
+    def default_headers(self):
+        return {}
+
+
+client = MeOpenAI(
+    base_url=base_url,
+)
+
+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-4',
+        # 'model': 'backup-gpt',
+        # 'model': 'test',
+        # 'model': 'per',
+
+        'messages': [
+            {'role': 'system', 'content': 'You are a helpful assistant.'},
+            {'role': 'user', 'content': 'hi'}
+        ],
+        'stream': True,
+
+    }
+
+    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/embeddings.py

@@ -0,0 +1,34 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Project      : AI.  @by PyCharm
+# @File         : embeddings
+# @Time         : 2024/1/4 11:05
+# @Author       : betterme
+# @WeChat       : meutils
+# @Software     : PyCharm
+# @Description  :
+
+from meutils.pipe import *
+from openai import OpenAI
+
+client = OpenAI(
+    # base_url='https://apis.chatfire.cn/v1',
+)
+model = "text-embedding-ada-002"
+# model = "text-embedding-3-small"
+model = "text-embedding-3-large"
+
+# model = "jina-embedding"
+# model = "text-embedding-3-small"
+# model='bge-large-zh-v1.5-q4'
+# model='acge_text_embedding'
+# model = 'doubao-embedding'
+model = 'BAAI/bge-m3'
+_ = client.embeddings.create(input='1', model=model)
+print(_)
+
+#
+# curl http://111.173.117.175:11434/api/embeddings -d '{
+#   "model": "bge-large-zh-v1.5-q4",
+#   "prompt": "Here is an article about llamas..."
+# }'

examples/_openaisdk/gpt4all.py

@@ -0,0 +1,69 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Project      : AI.  @by PyCharm
+# @File         : gpt4all
+# @Time         : 2024/1/2 13:28
+# @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"
+base_url = "https://api.chatllm.vip/v1"
+
+from openai import OpenAI
+
+client = OpenAI(
+    base_url=base_url,
+)
+
+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-4',
+        # 'model': 'backup-gpt',
+        # 'model': 'test',
+        # 'model': 'per',
+        # 'model': 'glm-4-all',
+        'model': '65f41c8a9c0ebbcbe28bb9c1',
+
+        'messages': [
+            {'role': 'system', 'content': 'You are a helpful assistant.'},
+            # {'role': 'user', 'content': '你是谁'},
+
+            {'role': 'user', 'content': '你是谁'}
+
+        ],
+        'stream': False,
+    }
+
+    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/gpt_fc.py

@@ -0,0 +1,23 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Project      : AI.  @by PyCharm
+# @File         : gpt_fc
+# @Time         : 2024/2/22 14:50
+# @Author       : betterme
+# @WeChat       : meutils
+# @Software     : PyCharm
+# @Description  :
+
+from meutils.pipe import *
+
+
+{
+  "apiName": "text2image",
+  "arguments": "{\"prompts\":[\"A realistic photo of a pig's head with a pink snout and floppy ears.\"],\"quality\":\"standard\",\"size\":\"1024x1024\",\"style\":\"natural\"}",
+  "identifier": "lobe-image-designer",
+  "type": "builtin"
+}
+
+
+
+

examples/_openaisdk/gr_vl.py

@@ -0,0 +1,46 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Project      : AI.  @by PyCharm
+# @File         : gr_vl
+# @Time         : 2024/6/3 20:33
+# @Author       : betterme
+# @WeChat       : meutils
+# @Software     : PyCharm
+# @Description  : https://huggingface.co/spaces/Dzkaka/ChatTTS
+
+from meutils.pipe import *
+from gradio_client import Client, file
+
+# client = Client("Dzkaka/ChatTTS", hf_token="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")
+#
+#
+# result = client.predict(
+#     text="四川美食确实以辣闻名，但也有不辣的选择。比如甜水面、赖汤圆、蛋烘糕、叶儿粑等，这些小吃口味温和，甜而不腻，也很受欢迎。",
+#     temperature=0.3,
+#     top_P=0.7,
+#     top_K=20,
+#     audio_seed_input=42,
+#     text_seed_input=42,
+#     refine_text_flag=True,
+#     api_name="/generate_audio"
+# )
+# print(result)
+
+
+from gradio_client import Client
+
+client = Client("https://playgroundai-playground-v2-5.hf.space/--replicas/yd7l4/",
+                hf_token="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")
+
+result = client.predict(
+    "Hello!!",  # str  in 'Prompt' Textbox component
+    "Hello!!",  # str  in 'Negative prompt' Textbox component
+    True,  # bool  in 'Use negative prompt' Checkbox component
+    0,  # float (numeric value between 0 and 2147483647) in 'Seed' Slider component
+    256,  # float (numeric value between 256 and 1536) in 'Width' Slider component
+    256,  # float (numeric value between 256 and 1536) in 'Height' Slider component
+    0.1,  # float (numeric value between 0.1 and 20) in 'Guidance Scale' Slider component
+    True,  # bool  in 'Randomize seed' Checkbox component
+    api_name="/run"
+)
+print(result)

examples/_openaisdk/json_mode.py

@@ -0,0 +1,12 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Project      : AI.  @by PyCharm
+# @File         : json_mode
+# @Time         : 2024/10/31 17:33
+# @Author       : betterme
+# @WeChat       : meutils
+# @Software     : PyCharm
+# @Description  : 
+
+from meutils.pipe import *
+

examples/_openaisdk/kimi.py

@@ -0,0 +1,91 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Project      : AI.  @by PyCharm
+# @File         : kimi
+# @Time         : 2024/2/5 18:24
+# @Author       : betterme
+# @WeChat       : meutils
+# @Software     : PyCharm
+# @Description  :
+
+from meutils.pipe import *
+from openai import OpenAI
+
+api_key = os.getenv('MOONSHOT_API_KEY')
+client = OpenAI(
+    api_key=api_key,
+    base_url="https://api.moonshot.cn/v1",
+)
+
+completion = client.chat.completions.create(
+  model="moonshot-v1-8k",
+  messages=[
+    {"role": "user", "content": "总结下这篇文章的主要观点。"},
+  ],
+  # temperature=0.3, extra_body={"refs": ["cn0bmk198onv4v01aafg"]}
+)
+#
+#
+#
+# print(completion.choices[0].message)
+file_object = client.files.create(
+    file=Path("/Users/betterme/PycharmProjects/AI/ChatLLM/examples/openaisdk/xx.pdf"),
+    purpose="file-extract"
+)
+print(file_object)
+
+# url = "https://ichat.chatfire.cn/uploads/20240313/1710300916647.pdf"
+# url = "https://ichat.chatfire.cn/#/chat/1710309600703"
+# r = requests.get(url)
+# file_object = client.files.create(
+#     file=r.content,
+#     purpose="file-extract"
+# )
+print(file_object)
+# print(client.files.retrieve(file_id=file_object.id))
+
+print(client.files.retrieve_content(file_id=file_object.id))
+# {"content":"品牌情况\n一、阿迪达斯三叶草（FDD）\n1、品牌位置：B 区 1F（102-105 铺）\n2、主要经营内容：福州市区最高级次三叶草门店，销售涵盖经典贝壳头鞋款，夏季透气清\n风，monkey kingdom 系列联名款，迪士尼联名系列潮流鞋服、配件等\n3、新品两件 8.8 折，夏季部分款式 6-8 折，更有联名款式和限量发售。\n二、NIKE 750\n1、品牌位置：苏宁广场 B 区一楼进门右转（近星巴克）\n2、主要经营内容：\n苏宁广场 NIKEBEACON750 店是福州 NIKE 全系列演绎体验旗舰店，涵括 Ain Jordan、\nBasketball、Running、Training，Sportswear、Football 等系列，完美演绎全品类产品。\n3、BEACON 寓意灯塔，是 NIKE 品牌的“旗舰店”等级！\nNIKE 全系列产品 热门爆款 发售款 当季最新款都会第一时间到店 目前春夏装两件 75 折\n起 品类基本涵盖全场 主要鞋款 8 折起 可使用滔博积分等 不定期有升值劵可享","file_type":"application/pdf","filename":"福州商场.pdf","title":"","type":"file"}
+
+# for i in range(10):
+#     file_object = client.files.create(file=file, purpose="file-extract")
+#     filename = file_object.filename
+#     #
+#     print(file_object)  # cn0bmk198onv4v01aafg
+#     break
+#
+# # print(client.files.retrieve(file_id="cn0bmk198onv4v01aafg"))
+# try:
+#     client.files.retrieve_content(file_id=file_object.id)  # 单文档，多路召回
+# except Exception as e:
+#     print(e)  # Error code: 500 - {'error': {'message': 'extract task failed: failed', 'type': ''}}
+
+# openai.InternalServerError: Error code: 500 - {'error': {'message': 'extract task failed: failed', 'type': ''}}
+# file_content = client.files.content(file_id=file_object.id).text
+#
+# file_content = client.files.retrieve_content(file_id=file_object.id)
+# print(file_content)
+
+# # 把它放进请求中
+# messages=[
+#     {
+#         "role": "system",
+#         "content": "你是 Kimi，由 Moonshot AI 提供的人工智能助手，你更擅长中文和英文的对话。你会为用户提供安全，有帮助，准确的回答。同时，你会拒绝一些涉及恐怖主义，种族歧视，黄色暴力等问题的回答。Moonshot AI 为专有名词，不可翻译成其他语言。",
+#     },
+#     {
+#         "role": "system",
+#         "content": file_content,
+#     },
+#     {"role": "user", "content": "请简单介绍 xlnet.pdf 讲了啥"},
+# ]
+#
+#
+# file_list = client.files.list()
+#
+# print(file_list.data)
+#
+# for file in tqdm(file_list.data):
+#     print(file)  # 查看每个文件的信息
+#
+#     print(client.files.delete(file_id=file.id))
+# FileObject(id='cnhgqnucp7f5h7jfb3s0', bytes=505444, created_at=1709378911, filename='福州商场.pdf', object='file', purpose='file-extract', status='ok', status_details='')

examples/_openaisdk/kimi_qa.py

@@ -0,0 +1,57 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Project      : AI.  @by PyCharm
+# @File         : kimi_qa
+# @Time         : 2024/2/8 13:45
+# @Author       : betterme
+# @WeChat       : meutils
+# @Software     : PyCharm
+# @Description  :
+
+from meutils.pipe import *
+from openai import OpenAI
+
+client = OpenAI(
+    api_key=api_key or os.getenv('MOONSHOT_API_KEY'),
+    base_url=os.getenv('MOONSHOT_BASE_URL'),
+)
+
+
+
+file_list = client.files.list()
+
+print(file_list.data)
+
+# for file in file_list.data:
+#     print(file)  # 查看每个文件的信息
+#
+#     print(client.files.delete(file_id=file.id))
+
+from openai import OpenAI
+
+client = OpenAI(
+    base_url=os.getenv('OPENAI_BASE_URL'),
+)
+
+file_object = client.files.create(file=Path("xlnet.pdf"), purpose="file-extract")
+file_content = client.files.content(file_id=file_object.id).text
+
+messages = [
+    {
+        "role": "system",
+        "content": "你是 ChatfireBot，由 Chatfire AI 提供的人工智能助手，你更擅长中文和英文的对话。你会为用户提供安全，有帮助，准确的回答。同时，你会拒绝一些涉及恐怖主义，种族歧视，黄色暴力等问题的回答。Chatfire AI 为专有名词，不可翻译成其他语言。",
+    },
+    {
+        "role": "system",
+        "content": file_content,
+    },
+    {"role": "user", "content": "请简单介绍文件讲了啥"},
+]
+
+completion = client.chat.completions.create(
+    model="gpt-4",
+    messages=messages,
+    temperature=0.3,
+)
+
+print(completion.choices[0].message)

examples/_openaisdk/minimax.py

@@ -0,0 +1,75 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Project      : AI.  @by PyCharm
+# @File         : minimax
+# @Time         : 2024/5/23 14:01
+# @Author       : betterme
+# @WeChat       : meutils
+# @Software     : PyCharm
+# @Description  :
+from asgiref.sync import async_to_sync
+
+from meutils.pipe import *
+
+from openai import OpenAI, AsyncOpenAI
+
+api_key = "eyJhbGciOiJSUzI1NiIsInR5cCI6IkpXVCJ9.eyJHcm91cE5hbWUiOiJjaGF0ZmlyZSIsIlVzZXJOYW1lIjoiY2hhdGZpcmUiLCJBY2NvdW50IjoiIiwiU3ViamVjdElEIjoiMTc2ODUzODA1NjAwMTg2NDQ2NCIsIlBob25lIjoiMTg1NTAyODgyMzMiLCJHcm91cElEIjoiMTc2ODUzODA1NTk5MzQ3NTg1NiIsIlBhZ2VOYW1lIjoiIiwiTWFpbCI6IiIsIkNyZWF0ZVRpbWUiOiIyMDI0LTA1LTIzIDEzOjMxOjU0IiwiaXNzIjoibWluaW1heCJ9.Gd4gBPZiI9VxAN-iRsNM4J4c0vNENtaV2TE4_OYPDSB5oYslZ_wx49b4lcm-rsfzJY65GWH-ATaqHQj-a28q4IfntbWFbVwvpd16n8hTnhoY2adkSKNSShtCuSdxMQgpCWI7UXVc-dwJ2LAwAwpe7t24Y_uuDy41KME07cm4_lFELgOQo7TcNI4sPeXKKmwrdv2uRkCNG1jzeI3sZF53ck-bdjTQl6k3qwE6XIbFOFpEgWWQXUK6-jP5MLIHpdJZ6_uax_OCQxx52apm0YRJqULTpiTqEB2dojINbLfexQpMtc8MPKjmx3d3Md_LANvYuQv5lPejuG1tDAuajzZ2Ww"
+
+data = {
+    "model": "abab6.5-chat",
+    "messages": [
+        {
+            "role": "user",
+            "content": "你是谁”"
+        }
+    ],
+    "stream": True,
+}
+
+# _ = OpenAI(
+#     api_key=api_key,
+#     base_url='https://any2chat.chatfire.cn/minimax/v1'
+# ).chat.completions.create(**data)
+# for i in _:
+#     print(i)
+
+_ = AsyncOpenAI(
+    api_key=api_key,
+    base_url='https://any2chat.chatfire.cn/minimax/v1'
+).chat.completions.create(**data)
+
+
+@async_to_sync()
+async def gen():
+    x = await _
+    logger.debug(type(x))
+    async for chunk in x:
+        logger.debug(chunk)
+    #
+    #     yield chunk
+    # yield "[DONE]"  # 兼容标准格式
+
+
+# async def gen():
+#     x = await _
+#     logger.debug(type(x))
+#     async for chunk in x:
+#         logger.debug(chunk)
+#     #
+#     #     yield chunk
+#     # yield "[DONE]"  # 兼容标准格式
+#
+#
+# gen()
+#
+from meutils.llm.openai_utils import create_chat_completion_chunk
+
+
+async def main():
+
+    async for i in create_chat_completion_chunk(_):
+        print(i)
+
+
+if __name__ == '__main__':
+    arun(main())

examples/_openaisdk/open_router.py

@@ -0,0 +1,48 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Project      : AI.  @by PyCharm
+# @File         : open_router
+# @Time         : 2024/10/14 19:04
+# @Author       : betterme
+# @WeChat       : meutils
+# @Software     : PyCharm
+# @Description  : 
+
+from meutils.pipe import *
+from openai import OpenAI
+from os import getenv
+
+# gets API Key from environment variable OPENAI_API_KEY
+client = OpenAI(
+    base_url="https://openrouter.ai/api/v1",
+    api_key=os.getenv("OPENROUTER_API_KEY"),
+)
+
+completion = client.chat.completions.create(
+    # extra_headers={
+    #   "HTTP-Referer": $YOUR_SITE_URL, # Optional, for including your app on openrouter.ai rankings.
+    #   "X-Title": $YOUR_APP_NAME, # Optional. Shows in rankings on openrouter.ai.
+    # },
+    # model="meta-llama/llama-3.2-11b-vision-instruct:free",
+    model="openai/o1",
+    max_tokens=1000,
+
+    messages=[
+        {
+            "role": "user",
+            "content": [
+                {
+                    "type": "text",
+                    "text": "What's in this image?"
+                },
+                {
+                    "type": "image_url",
+                    "image_url": {
+                        "url": "https://upload.wikimedia.org/wikipedia/commons/thumb/d/dd/Gfp-wisconsin-madison-the-nature-boardwalk.jpg/2560px-Gfp-wisconsin-madison-the-nature-boardwalk.jpg"
+                    }
+                }
+            ]
+        }
+    ]
+)
+print(completion.choices[0].message.content)

examples/_openaisdk/openai_aiplus.py

@@ -0,0 +1,54 @@
+#!/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
+
+
+client = OpenAI(
+    base_url="https://api.aiplus.ai/v1",
+    api_key="sk-RiTYytBkTFqnfNnDE7B6A7DaC7Dc4cBeAb8d0b021e855c34",
+
+)
+print(client.models.list())
+
+
+try:
+    completion = client.chat.completions.create(
+        model="claude-3-5-sonnet-1022",
+        # model="xxxxxxxxxxxxx",
+        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)
+
+# r = client.images.generate(
+#     model="cogview-3-plus",
+#     prompt="a white siamese cat",
+#     size="1024x1024",
+#     quality="hd",
+#     n=1,
+# )

examples/_openaisdk/openai_audio.py

@@ -0,0 +1,20 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Project      : AI.  @by PyCharm
+# @File         : chat_dev
+# @Time         : 2024/7/8 21:18
+# @Author       : betterme
+# @WeChat       : meutils
+# @Software     : PyCharm
+# @Description  :
+from openai import OpenAI
+from meutils.pipe import *
+
+# base_url = "https://openai-dev.chatfire.cn/audio/v1"
+
+
+r = OpenAI(
+    # base_url=base_url
+).audio.speech.create(input="你好呀", model='tts-1', voice="alloy")
+
+r.stream_to_file('xx.mp3')

examples/_openaisdk/openai_baichuan.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  : todo: assistants
+import os
+
+from meutils.pipe import *
+from openai import OpenAI
+from openai import OpenAI, APIStatusError
+from meutils.schemas.oneapi.models import BAICHUAN
+
+client = OpenAI(
+    api_key=os.getenv("BAICHUAN_API_KEY"),
+    base_url=os.getenv("BAICHUAN_BASE_URL")
+)
+
+# tool:search
+
+for model in BAICHUAN:
+    try:
+        completion = client.chat.completions.create(
+            model=model,
+            messages=[
+                # {"role": "system", "content": "你是个画图工具"},
+                # {"role": "user", "content": "你是谁"},
+
+                {"role": "user", "content": "总结下 https://top.baidu.com/board?platform=pc&sa=pcindex_a_right"}
+
+            ],
+            # top_p=0.7,
+            stream=True,
+
+            tools=[
+                {
+                    "type": "web_search",
+                    "web_search": {
+                        "enable": True,
+                        "search_mode": "performance_first"
+                    }
+                }
+            ]
+
+        )
+    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.replace('百川', '火哥'))
+
+    break