MeUtils 2025.2.20.17.30.7__py3-none-any.whl → 2025.2.25.19.14.31__py3-none-any.whl

meutils/caches/acache.py

@@ -10,12 +10,29 @@
 import os
 
 from meutils.pipe import *
+from meutils.db.redis_db import async_pool
 
 from aiocache import cached
 
 from aiocache import cached, Cache, RedisCache, caches
 from aiocache import multi_cached
 
+logger.debug(async_pool.connection_kwargs)
+def rcache(**kwargs): # todo
+    # if REDIS_URL := os.getenv("REDIS_URL"):
+    # endpoint = "127.0.0.1",
+    # port = 6379,
+    # db = 0,
+    # password = None,
+
+    return cached(
+        cache=RedisCache,
+        # endpoint=REDIS_URL,
+        # connection_pool_kwargs=1,
+        **async_pool.connection_kwargs,
+        **kwargs
+    )
+
 
 
 @cached(ttl=60)
@@ -23,11 +40,12 @@ from aiocache import multi_cached
 async def cached_fc(user_id, **kwargs):
     logger.debug(user_id)
     return False
-#
-rcache = RedisCache(
-    endpoint="127.0.0.1", port=6379, namespace="me"  # 缓存键前缀
-)
-@cached(cache=rcache)
+
+
+
+
+
+@rcache(ttl=3)
 async def redis_fc(user_id, **kwargs):
     logger.debug(user_id)
     return False
@@ -59,8 +77,34 @@ async def redis_fc(user_id, **kwargs):
 #
 
 class A(BaseModel):
-    a: Any = 1
+    a: Any = 111
+
+
+import asyncio
+from aiocache import cached
+from aiocache.backends.redis import RedisCache
+from aiocache.serializers import PickleSerializer
+
+
+# 使用 @cached 装饰器缓存函数结果
+@cached(cache=RedisCache, endpoint="127.0.0.1", port=6379, namespace="main",
+        # serializer=PickleSerializer(),
+        key="my_key", ttl=60)
+async def expensive_operation():
+    print("Performing expensive operation...")
+    await asyncio.sleep(2)  # 模拟耗时操作
+    return {"result": "data"}
+
 
 
 if __name__ == '__main__':
-    arun(cached_fc(A()))
+
+    async def main():
+        for i in range(10):
+            logger.debug(i)
+
+            await redis_fc(A())
+            await asyncio.sleep(1)
+
+
+    arun(main())

meutils/data/VERSION

@@ -1 +1 @@
-2025.02.20.17.30.07
+2025.02.25.19.14.31

meutils/data/oneapi/FOOTER.md

@@ -1,7 +1,7 @@
-<div class="custom-footer">@2024  <a href="https://api.chatfire.cn">🔥ChatfireAPI</a>  由火宝AI出品 
+<div class="custom-footer">@2025  <a href="https://api.chatfire.cn">🔥ChatfireAPI</a>  由火宝AI出品 
 | <a href="https://api.chatfire.cn" target="_blank">支持对公&发票</a> 
 | <a href="https://api.chatfire.cn" target="_blank">提供模型定制化服务</a> 
 | <a href="https://api.chatfire.cn" target="_blank">个性化智能体</a> 
-|  🛎<a href="https://api.chatfire.cn/about" target="_blank">企微客服，咨询更多</a>
+|  🛎<a href="https://cdn.apifox.com/app/apidoc-image/custom/20241110/f36e2152-b4fb-43be-a33d-a380b4d97ef6.jpeg" target="_blank">企微客服，咨询更多</a>
 
 </div>

meutils/data/oneapi/NOTICE.md

@@ -1,151 +1 @@
-<h1 align = "center">🔥公告🚀</h1>
-
----
-<details markdown="1">
-  <summary><b>🔥业务经营范围</b></summary>
-
-- api服务（没有的找企微客服增加）
-    - 提供主流大模型服务，gpt/claude/gemini/llama/国产大模型等等
-    - 提供多模态模型服务，文件解析/图片解析/语音解析/视频解析等等
-    - 提供垂类智能体服务，文件问答/联网问答/学术搜索等等
-    - 提供语音克隆&语音合成服务，hailuo/fish/chattts等等
-    - 提供embedding服务，bge/jina/openai等等
-    - 提供图片生成服务，kling/flux/ideogram/recraft/虚拟换衣/换头等等
-    - 提供视频生成服务，kling/cogviewx/hailuo/hunyuan/vidu/sora等等
-    - 提供图片编辑服务，变清晰、去水印、抠图等等
-    - 提供文档智能服务，ocr/pdf-to-markdown/url-to-markdown等等
-    - 提供对象存储服务
-
-- 账号服务（市面上有的都可以）
-    - gpt-plus/claude-pro
-    - api-key
-
-- 个性化服务
-    - 定制同款api聚合站点，一键对接货源
-    - 定制企业智能体，类似gpt-4-all/kimi
-    - 定制知识库智能问答（RAG）
-    - 定制AI类网站/小程序等等
-    - 承接数据标注/数据跑批任务
-    - 承接大模型微调，定制化大模型（可端到端）
-    - 承接其他项目，算法模型等等
-
-</details>
-
-<details markdown="1">
-  <summary><b>大额对公，请联系客服</b></summary>
-</details>
-
-## 2025-02-10
-### 上线新模型
-- qwen2.5-max： 比deepseek-v3效果更好，更智能（权威排行榜更靠前）
-
-## 2025-02-06
-### 上线新模型
-- deepseek-r1-think： 兼容任意客户端，再也不用担心看不到`深度思考`了
-- deepseek-search： 搜索+深度思考
-- deepseek-r1-search： 搜索+深度思考
-- deepseek-r1-metasearch： 秘塔搜索+深度思考
-
-- deepseek-ai/deepseek-v3 
-  - 即日起至北京时间 2025-02-09 00:00 同步官网倍率
-
-
----
-
-<details markdown="1">
-  <summary><b>历史更新</b></summary>
-
-## 2025-01-04
-
-- 增加模型配额 gemini-2.0-flash-exp、gemini-2.0-flash-thinking-exp-1219
-
-## 2024-12-31
-
-- 上线新模型
-    - `glm-zero/glm-zero-preview`：GLM-Zero-Preview 专注于增强模型推理能力，擅长处理数理逻辑、代码和需要深度推理的复杂问题。同基座模型相比，GLM-Zero-Preview
-      在不显著降低通用任务能力的情况下，在专家任务能力方面表现大幅提升。其在 AIME 2024、MATH500 和 LiveCodeBench 评测中，效果与
-      OpenAI-o1-Preview 相当。
-    - 兼容SparkAI客户端，文件问答&图片问答：baseurl改为`https://api.chatfire.cn/sparkai/v1`
-
-## 2024-12-27
-
-- 上线新模型
-    - deepseek-v3
-    - deepseek-r1：deepseek-v3的思考模型
-    - deepseek-search：deepseek-v3的联网模型
-
-## 2024-12-24
-
-- 上线新模型
-    - doubao-pro-256k：相比Doubao-pro-128k/240628，长文任务效果显著提升10%以上，要点提取、字数遵循、多轮对话上文记忆等能力大幅提升
-    - [qvq-72b-preview](https://mp.weixin.qq.com/s/WzL7tbFUZOgE2IFMeHT-sQ)：Qwen开源视觉推理模型QVQ，更睿智地看世界！
-
-- 增加gemini-2.0配额，支持多模型，默认分组可用
-    - "gemini-2.0-flash"
-    - "gemini-2.0-flash-exp"
-
-## 2024-12-20
-
-- 修复SunoV4无水印版本
-    - [异步任务接口文档](https://api.chatfire.cn/docs/api-246593467)
-- [增加视频解析模型](https://api.chatfire.cn/docs/api-246688638)
-- 增加高并发mj-fast
-
-## 2024-12-19
-
-- 新增生图模型 SeedEdit（文生图&图生图/图片编辑）: 一句话编辑你的世界：字节跳动推出革命性图片编辑工具SeedEdit
-    - [Chat模式接口文档](https://api.chatfire.cn/docs/api-214415540)
-    - [Images接口文档](https://api.chatfire.cn/docs/api-246137616)
-    - [异步任务接口文档](https://api.chatfire.cn/docs/api-246120232)
-- 新增视觉模型
-    - deepseek-ai/deepseek-vl2
-    - doubao-vision-pro-32k
-    - doubao-vision-lite-32k
-- 新增视频模型 Sora
-    - Chat模式：`sora-1:1-480p-5s`
-    - 异步任务接口在路上
-
-## 2024-12-13
-
-- 新增模型 混元视频（支持高并发，非逆向可商用，限时特价1毛）[接口文档](https://api.chatfire.cn/docs/api-244309840)
-  HunyuanVideo 是腾讯推出的开源视频生成基础模型，拥有超过 130
-  亿参数，是目前最大的开源视频生成模型。该模型采用统一的图像和视频生成架构，集成了数据整理、图像-视频联合模型训练和高效基础设施等关键技术。模型使用多模态大语言模型作为文本编码器，通过
-  3D VAE 进行空间-时间压缩，并提供提示词重写功能。根据专业人工评估结果，HunyuanVideo 在文本对齐、运动质量和视觉质量等方面的表现优于现有最先进的模型
-
-## 2024-12-09
-
-- 新增模型
-    - meta-llama/Llama-3.3-70B-Instruct: Llama 3.3 是 Llama 系列最先进的多语言开源大型语言模型，以极低成本体验媲美 405B
-      模型的性能。基于 Transformer
-      结构，并通过监督微调（SFT）和人类反馈强化学习（RLHF）提升有用性和安全性。其指令调优版本专为多语言对话优化，在多项行业基准上表现优于众多开源和封闭聊天模型。知识截止日期为
-      2023 年 12 月。
-    - jimeng-v2.1：豆包画图，支持即梦超强图像生成能力，兼容chat/dalle-image调用方式。
-    - 海螺最新的I2V-01-live图生视频模型：特别针对二次元图生视频效果，进行了优化，动作流畅又生动，让2D二次元角色像复活一样。
-
-## 2024-12-06
-
-- 新增模型
-    - o1-plus: （官网 plus 版本 `逆向工程`，有思考过程显示）o1 是OpenAI针对复杂任务的新推理模型，该任务需要广泛的常识。该模型具有
-      200k 上下文，目前全球最强模型，支持图片识别
-    - o1-pro: （官网 200刀 plus 版本 `逆向工程`，有思考过程显示）o1-pro 是OpenAI针对复杂任务的新推理模型，该任务需要广泛的常识。该模型具有
-      200k 上下文，目前全球最强模型，支持图片识别
-
-## 2024-12-05
-
-- 新增模型gpt-4-plus/gpt-4o-plus按倍率计算
-  > OpenAI-plus会员 逆向工程
-
-## 2024-11-29
-
-- 新增推理模型
-    - Qwen/QwQ-32B-Preview
-      > 强大的数学问题解决能力，在AIME、MATH-500数学评测上，超过了OpenAI o1-preview优秀的编码能力，LiveCodeBench接近OpenAI
-      o1-preview
-
-## 2024-11-25
-
-- 新增虚拟换衣接口
-    - [可灵官方api格式](https://api.chatfire.cn/docs/api-237182295) 0.8/次
-    - [老接口格式](https://api.chatfire.cn/docs/api-226983436) 0.1/次
-
-</details>
+![🔥最新功能](https://readpo.com/p/%23%20%E6%9C%80%E6%96%B0%E9%80%9A%E7%9F%A5%0A%3E%20%E6%94%AF%E6%8C%81%E8%B6%85%E5%A4%9A%E6%A8%A1%E5%9E%8B%EF%BC%9A%60%E5%AF%B9%E8%AF%9D%60%20%60%E5%9B%BE%E7%89%87%60%20%60%E8%A7%86%E9%A2%91%60%20%60%E8%AF%AD%E9%9F%B3%60%20%60%E9%9F%B3%E4%B9%90%60%20%60%E5%8F%98%E6%B8%85%E6%99%B0%60%20%60%E5%8E%BB%E6%B0%B4%E5%8D%B0%60%20%60%E6%96%87%E6%A1%A3%E8%A7%A3%E6%9E%90%60%20%60%E8%81%94%E7%BD%91API%60%0A---%0A%0A%23%23%202025.02.21%0A%3E%20jina-deepsearch%20%E6%BB%A1%E8%A1%80r1%20%60%E6%8E%A8%E7%90%86%60%20%60%E6%90%9C%E7%B4%A2%60%0A%0A%0A%3E%20Grok-3%20%E6%98%AF%E9%A9%AC%E6%96%AF%E5%85%8B%E6%97%97%E4%B8%8BxAI%E5%85%AC%E5%8F%B8%E5%BC%80%E5%8F%91%E7%9A%84%E4%BA%BA%E5%B7%A5%E6%99%BA%E8%83%BD%E6%A8%A1%E5%9E%8B%EF%BC%8C%E5%85%B7%E5%A4%87128%2C000%E4%B8%AAToken%E7%9A%84%E4%B8%8A%E4%B8%8B%E6%96%87%E5%A4%84%E7%90%86%E8%83%BD%E5%8A%9B%EF%BC%8C%E6%94%AF%E6%8C%81%E5%87%BD%E6%95%B0%E8%B0%83%E7%94%A8%E5%92%8C%E7%B3%BB%E7%BB%9F%E6%8F%90%E7%A4%BA%EF%BC%8C%E5%B9%B6%E8%AE%A1%E5%88%92%E6%8E%A8%E5%87%BA%E5%A4%9A%E6%A8%A1%E6%80%81%E7%89%88%E6%9C%AC%E4%BB%A5%E5%A4%84%E7%90%86%E5%9B%BE%E5%83%8F%E3%80%82%0A%20%20-%20grok-3%0A%20%20-%20grok-3-reasoner%20%60%E6%8E%A8%E7%90%86%60%0A%20%20-%20grok-3-deepsearch%20%60%E6%90%9C%E7%B4%A2%60)

meutils/data/oneapi/_NOTICE.md

@@ -0,0 +1,140 @@
+<h1 align = "center">🔥公告🚀</h1>
+
+---
+<details markdown="1">
+  <summary><b>🔥业务经营范围</b></summary>
+
+- api服务（没有的找企微客服增加）
+    - 提供主流大模型服务，gpt/claude/gemini/llama/国产大模型等等
+    - 提供多模态模型服务，文件解析/图片解析/语音解析/视频解析等等
+    - 提供垂类智能体服务，文件问答/联网问答/学术搜索等等
+    - 提供语音克隆&语音合成服务，hailuo/fish/chattts等等
+    - 提供embedding服务，bge/jina/openai等等
+    - 提供图片生成服务，kling/flux/ideogram/recraft/虚拟换衣/换头等等
+    - 提供视频生成服务，kling/cogviewx/hailuo/hunyuan/vidu/sora等等
+    - 提供图片编辑服务，变清晰、去水印、抠图等等
+    - 提供文档智能服务，ocr/pdf-to-markdown/url-to-markdown等等
+    - 提供对象存储服务
+
+- 账号服务（市面上有的都可以）
+    - gpt-plus/claude-pro
+    - api-key
+
+- 个性化服务
+    - 定制同款api聚合站点，一键对接货源
+    - 定制企业智能体，类似gpt-4-all/kimi
+    - 定制知识库智能问答（RAG）
+    - 定制AI类网站/小程序等等
+    - 承接数据标注/数据跑批任务
+    - 承接大模型微调，定制化大模型（可端到端）
+    - 承接其他项目，算法模型等等
+
+</details>
+
+<details markdown="1">
+  <summary><b>大额对公，请联系客服</b></summary>
+</details>
+
+## 2025-02-20 新上模型
+- jina-deepsearch 满血r1+搜索
+
+
+---
+
+<details markdown="1">
+  <summary><b>历史更新</b></summary>
+
+## 2025-01-04
+
+- 增加模型配额 gemini-2.0-flash-exp、gemini-2.0-flash-thinking-exp-1219
+
+## 2024-12-31
+
+- 上线新模型
+    - `glm-zero/glm-zero-preview`：GLM-Zero-Preview 专注于增强模型推理能力，擅长处理数理逻辑、代码和需要深度推理的复杂问题。同基座模型相比，GLM-Zero-Preview
+      在不显著降低通用任务能力的情况下，在专家任务能力方面表现大幅提升。其在 AIME 2024、MATH500 和 LiveCodeBench 评测中，效果与
+      OpenAI-o1-Preview 相当。
+    - 兼容SparkAI客户端，文件问答&图片问答：baseurl改为`https://api.chatfire.cn/sparkai/v1`
+
+## 2024-12-27
+
+- 上线新模型
+    - deepseek-v3
+    - deepseek-r1：deepseek-v3的思考模型
+    - deepseek-search：deepseek-v3的联网模型
+
+## 2024-12-24
+
+- 上线新模型
+    - doubao-pro-256k：相比Doubao-pro-128k/240628，长文任务效果显著提升10%以上，要点提取、字数遵循、多轮对话上文记忆等能力大幅提升
+    - [qvq-72b-preview](https://mp.weixin.qq.com/s/WzL7tbFUZOgE2IFMeHT-sQ)：Qwen开源视觉推理模型QVQ，更睿智地看世界！
+
+- 增加gemini-2.0配额，支持多模型，默认分组可用
+    - "gemini-2.0-flash"
+    - "gemini-2.0-flash-exp"
+
+## 2024-12-20
+
+- 修复SunoV4无水印版本
+    - [异步任务接口文档](https://api.chatfire.cn/docs/api-246593467)
+- [增加视频解析模型](https://api.chatfire.cn/docs/api-246688638)
+- 增加高并发mj-fast
+
+## 2024-12-19
+
+- 新增生图模型 SeedEdit（文生图&图生图/图片编辑）: 一句话编辑你的世界：字节跳动推出革命性图片编辑工具SeedEdit
+    - [Chat模式接口文档](https://api.chatfire.cn/docs/api-214415540)
+    - [Images接口文档](https://api.chatfire.cn/docs/api-246137616)
+    - [异步任务接口文档](https://api.chatfire.cn/docs/api-246120232)
+- 新增视觉模型
+    - deepseek-ai/deepseek-vl2
+    - doubao-vision-pro-32k
+    - doubao-vision-lite-32k
+- 新增视频模型 Sora
+    - Chat模式：`sora-1:1-480p-5s`
+    - 异步任务接口在路上
+
+## 2024-12-13
+
+- 新增模型 混元视频（支持高并发，非逆向可商用，限时特价1毛）[接口文档](https://api.chatfire.cn/docs/api-244309840)
+  HunyuanVideo 是腾讯推出的开源视频生成基础模型，拥有超过 130
+  亿参数，是目前最大的开源视频生成模型。该模型采用统一的图像和视频生成架构，集成了数据整理、图像-视频联合模型训练和高效基础设施等关键技术。模型使用多模态大语言模型作为文本编码器，通过
+  3D VAE 进行空间-时间压缩，并提供提示词重写功能。根据专业人工评估结果，HunyuanVideo 在文本对齐、运动质量和视觉质量等方面的表现优于现有最先进的模型
+
+## 2024-12-09
+
+- 新增模型
+    - meta-llama/Llama-3.3-70B-Instruct: Llama 3.3 是 Llama 系列最先进的多语言开源大型语言模型，以极低成本体验媲美 405B
+      模型的性能。基于 Transformer
+      结构，并通过监督微调（SFT）和人类反馈强化学习（RLHF）提升有用性和安全性。其指令调优版本专为多语言对话优化，在多项行业基准上表现优于众多开源和封闭聊天模型。知识截止日期为
+      2023 年 12 月。
+    - jimeng-v2.1：豆包画图，支持即梦超强图像生成能力，兼容chat/dalle-image调用方式。
+    - 海螺最新的I2V-01-live图生视频模型：特别针对二次元图生视频效果，进行了优化，动作流畅又生动，让2D二次元角色像复活一样。
+
+## 2024-12-06
+
+- 新增模型
+    - o1-plus: （官网 plus 版本 `逆向工程`，有思考过程显示）o1 是OpenAI针对复杂任务的新推理模型，该任务需要广泛的常识。该模型具有
+      200k 上下文，目前全球最强模型，支持图片识别
+    - o1-pro: （官网 200刀 plus 版本 `逆向工程`，有思考过程显示）o1-pro 是OpenAI针对复杂任务的新推理模型，该任务需要广泛的常识。该模型具有
+      200k 上下文，目前全球最强模型，支持图片识别
+
+## 2024-12-05
+
+- 新增模型gpt-4-plus/gpt-4o-plus按倍率计算
+  > OpenAI-plus会员 逆向工程
+
+## 2024-11-29
+
+- 新增推理模型
+    - Qwen/QwQ-32B-Preview
+      > 强大的数学问题解决能力，在AIME、MATH-500数学评测上，超过了OpenAI o1-preview优秀的编码能力，LiveCodeBench接近OpenAI
+      o1-preview
+
+## 2024-11-25
+
+- 新增虚拟换衣接口
+    - [可灵官方api格式](https://api.chatfire.cn/docs/api-237182295) 0.8/次
+    - [老接口格式](https://api.chatfire.cn/docs/api-226983436) 0.1/次
+
+</details>

meutils/files/__init__.py

@@ -0,0 +1,11 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Project      : AI.  @by PyCharm
+# @File         : __init__.py
+# @Time         : 2025/2/25 12:19
+# @Author       : betterme
+# @WeChat       : meutils
+# @Software     : PyCharm
+# @Description  : 
+
+from meutils.pipe import *

meutils/files/qwen_files.py

@@ -0,0 +1,30 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Project      : AI.  @by PyCharm
+# @File         : openai_files.py
+# @Time         : 2025/2/25 12:16
+# @Author       : betterme
+# @WeChat       : meutils
+# @Software     : PyCharm
+# @Description  : 
+
+from meutils.pipe import *
+
+from openai import OpenAI, AsyncOpenAI
+
+client = AsyncOpenAI(
+    base_url="https://all.chatfire.cn/qwen/v1",
+    api_key="eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpZCI6IjMxMGNiZGFmLTM3NTQtNDYxYy1hM2ZmLTllYzgwMDUzMjljOSIsImV4cCI6MTc0MzAzNTk4OH0.GVAoSFtK94a9CgxqHCEnxzAnRi7gafIvYyH9mIJUh4s"
+)
+
+
+async def upload():
+    file_object = await client.files.create(
+        file=Path("招标文件备案表（第二次）.pdf"),
+        purpose="file-extract"
+    )
+    return file_object
+
+
+if __name__ == '__main__':
+    pass

meutils/io/files_utils.py

@@ -224,9 +224,9 @@ if __name__ == '__main__':
 
     file = Path("/Users/betterme/PycharmProjects/AI/MeUtils/meutils/io/x.py").read_bytes()
 
-    # arun(to_url_fal([file] * 10))
+    arun(to_url_fal([file] * 1))
 
-    arun(to_bytes(None))
+    # arun(to_bytes(None))
 
     # print(mimetypes.guess_type("x.jpg"))
     # print(mimetypes.guess_type("x.png"))

meutils/llm/check_utils.py

@@ -40,6 +40,7 @@ async def check_token_for_siliconflow(api_key, threshold: float = 0):
             logger.debug(response.status_code)
 
             if response.is_success:
+                logger.debug(api_key)
                 total_balance = response.json()['data']['totalBalance']
                 return float(total_balance) >= threshold
 
@@ -97,7 +98,7 @@ if __name__ == '__main__':
     check_valid_token = partial(check_token_for_siliconflow, threshold=-1)
 
     pass
-    arun(check_valid_token("sk-geevdizajhngehlkrjlgwcugcunrircpezhqijrscmtiafxj", threshold=0.1))
+    arun(check_valid_token("sk-qnhsvewyfcswsppofrooaeqpwhejfgbyqfjfmozmmhnwvzap", threshold=0.1))
 
     FEISHU_URL = "https://xchatllm.feishu.cn/sheets/Bmjtst2f6hfMqFttbhLcdfRJnNf?sheet=KVClcs"
 

meutils/llm/clients.py

@@ -14,7 +14,6 @@ from meutils.pipe import *
 OpenAI = lru_cache(Client)
 AsyncOpenAI = lru_cache(AsyncClient)
 
-
 chatfire_client = AsyncOpenAI()
 
 moonshot_client = AsyncOpenAI(
@@ -26,6 +25,11 @@ zhipuai_client = AsyncOpenAI(
     base_url=os.getenv("ZHIPUAI_BASE_URL")
 )
 
+qwen_client = AsyncOpenAI(
+    base_url="https://all.chatfire.cn/qwen/v1",
+    api_key="eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpZCI6IjMxMGNiZGFmLTM3NTQtNDYxYy1hM2ZmLTllYzgwMDUzMjljOSIsImV4cCI6MTc0MzAzNTk4OH0.GVAoSFtK94a9CgxqHCEnxzAnRi7gafIvYyH9mIJUh4s"
+)
+
 # zhipuai_client = OpenAI(
 #     api_key=os.getenv("ZHIPUAI_API_KEY"),
 #     base_url=os.getenv("ZHIPUAI_BASE_URL")
@@ -53,4 +57,3 @@ if __name__ == '__main__':
     # )
 
     # r.model_dump_json()
-

meutils/llm/completions/qwenllm.py

@@ -7,10 +7,18 @@
 # @WeChat       : meutils
 # @Software     : PyCharm
 # @Description  :
+"""
+ File "/usr/local/lib/python3.10/site-packages/meutils/llm/completions/qwenllm.py", line 47, in create
+    yield response.choices[0].message.content
+AttributeError: 'str' object has no attribute 'choices'
+
+"""
 
 from openai import AsyncOpenAI
 
 from meutils.pipe import *
+from meutils.llm.clients import qwen_client
+
 from meutils.llm.openai_utils import to_openai_params
 
 from meutils.config_utils.lark_utils import get_next_token_for_polling
@@ -25,13 +33,27 @@ from fake_useragent import UserAgent
 ua = UserAgent()
 
 
-async def create(request: ChatCompletionRequest):
-    token = await get_next_token_for_polling(feishu_url=FEISHU_URL)
+async def create(request: ChatCompletionRequest, token: Optional[str] = None):  # ChatCompletionRequest 重构
+    token = token or await get_next_token_for_polling(feishu_url=FEISHU_URL)
 
     client = AsyncOpenAI(
-        base_url=base_url, api_key=token,
+        base_url=base_url,
+        api_key=token,
         default_headers={'User-Agent': ua.random}
     )
+
+    # qwen结构
+    if any(i in request.model.lower() for i in ("search",)):
+        request.model = "qwen-plus-latest"
+        request.messages[-1]['chat_type'] = "search"
+
+    if any(i in request.model.lower() for i in ("qwq", "think")):
+        request.model = "qwen-plus-latest"
+        request.messages[-1]['feature_config'] = {"thinking_enabled": True}
+
+    # 多模态: todo
+    # file_object = await qwen_client.files.create(file=(file.filename, file.file), purpose="file-extract")
+
     data = to_openai_params(request)
 
     if request.stream:
@@ -43,8 +65,11 @@ async def create(request: ChatCompletionRequest):
 
     else:
         response = await client.chat.completions.create(**data)
-        # logger.info(response)
-        yield response.choices[0].message.content
+        for i in range(3):
+            if not isinstance(response, str):  # 报错
+                yield response.choices[0].message.content
+            else:
+                logger.warning(f"重试 {i}\n{response}")
 
 
 if __name__ == '__main__':
@@ -60,16 +85,38 @@ if __name__ == '__main__':
     # ]
     request = ChatCompletionRequest(
         # model="qwen-turbo-2024-11-01",
-        model="qwen-max-latest",
+        # model="qwen-max-latest",
+        model="qwen-max-latest-search",
+
         # model="qwen-plus-latest",
 
         messages=[
             {
                 'role': 'user',
-                'content': 'hi'
+                # 'content': '今天南京天气',
+                # 'content': '总结下',
+
+                # "chat_type": "search",
+
+                "content": [
+                    {
+                        "type": "text",
+                        "text": "总结下",
+                        "chat_type": "t2t",
+                        "feature_config": {
+                            "thinking_enabled": False
+                        }
+                    },
+                    {
+                        "type": "file",
+                        "file": "2d677df1-45b2-4f30-829f-0d42b2b07136"
+                    }
+                ]
+
             },
 
         ],
-        stream=False,
+        stream=True,
+
     )
     arun(create(request))

meutils/llm/utils.py

@@ -113,6 +113,7 @@ def oneturn2multiturn(messages, template: Optional[str] = None):
     return context
 
 
+
 if __name__ == '__main__':
     async def main():
         with timer():