MeUtils 2024.3.4.13.4.45__py3-none-any.whl → 2025.1.16.17.15.52__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- {MeUtils-2024.3.4.13.4.45.dist-info → MeUtils-2025.1.16.17.15.52.dist-info}/METADATA +38 -32
- MeUtils-2025.1.16.17.15.52.dist-info/RECORD +864 -0
- {MeUtils-2024.3.4.13.4.45.dist-info → MeUtils-2025.1.16.17.15.52.dist-info}/WHEEL +1 -1
- {MeUtils-2024.3.4.13.4.45.dist-info → MeUtils-2025.1.16.17.15.52.dist-info}/entry_points.txt +1 -0
- apps/spider.py +24 -8
- examples/_openaisdk/4v.py +110 -0
- examples/_openaisdk/__init__.py +11 -0
- examples/_openaisdk/baichuan.py +38 -0
- examples/_openaisdk/bpo.py +138 -0
- examples/_openaisdk/chat_latex.py +95 -0
- examples/_openaisdk/chattts.py +85 -0
- examples/_openaisdk/copilot.py +48 -0
- examples/_openaisdk/dalle3.py +48 -0
- examples/_openaisdk/deeplx.py +31 -0
- examples/_openaisdk/demo.py +77 -0
- examples/_openaisdk/embeddings.py +34 -0
- examples/_openaisdk/gpt4all.py +69 -0
- examples/_openaisdk/gpt_fc.py +23 -0
- examples/_openaisdk/gr_vl.py +46 -0
- examples/_openaisdk/json_mode.py +12 -0
- examples/_openaisdk/kimi.py +91 -0
- examples/_openaisdk/kimi_qa.py +57 -0
- examples/_openaisdk/minimax.py +75 -0
- examples/_openaisdk/open_router.py +48 -0
- examples/_openaisdk/openai_aiplus.py +54 -0
- examples/_openaisdk/openai_audio.py +20 -0
- examples/_openaisdk/openai_baichuan.py +59 -0
- examples/_openaisdk/openai_cache.py +37 -0
- examples/_openaisdk/openai_chatfire.py +228 -0
- examples/_openaisdk/openai_chatfire_all.py +166 -0
- examples/_openaisdk/openai_deepinfra.py +55 -0
- examples/_openaisdk/openai_deepseek.py +29 -0
- examples/_openaisdk/openai_doubao.py +43 -0
- examples/_openaisdk/openai_embeddings.py +36 -0
- examples/_openaisdk/openai_files.py +57 -0
- examples/_openaisdk/openai_gitee.py +33 -0
- examples/_openaisdk/openai_god.py +45 -0
- examples/_openaisdk/openai_groq.py +240 -0
- examples/_openaisdk/openai_images.py +203 -0
- examples/_openaisdk/openai_json.py +78 -0
- examples/_openaisdk/openai_lingyi.py +59 -0
- examples/_openaisdk/openai_modelscope.py +54 -0
- examples/_openaisdk/openai_moon.py +55 -0
- examples/_openaisdk/openai_oi.py +61 -0
- examples/_openaisdk/openai_ppu.py +47 -0
- examples/_openaisdk/openai_qwen.py +58 -0
- examples/_openaisdk/openai_search.py +42 -0
- examples/_openaisdk/openai_sensenova.py +81 -0
- examples/_openaisdk/openai_siliconflow.py +52 -0
- examples/_openaisdk/openai_step.py +45 -0
- examples/_openaisdk/openai_test.py +66 -0
- examples/_openaisdk/openai_together.py +57 -0
- examples/_openaisdk/openai_tune.py +38 -0
- examples/_openaisdk/openai_zhipu.py +59 -0
- examples/_openaisdk/ppu.py +28 -0
- examples/_openaisdk/rag.py +54 -0
- examples/_openaisdk/rag_.py +26 -0
- examples/_openaisdk/test.py +52 -0
- examples/_openaisdk/x.py +32 -0
- examples/_openaisdk/xx.py +29 -0
- examples/_openaisdk/zhipu_files.py +47 -0
- examples/_openaisdk/zhipu_/346/231/272/350/203/275/344/275/223.py +45 -0
- examples/_openaisdk//345/205/234/345/272/225/346/265/213/350/257/225.py +50 -0
- examples/_openaisdk//345/244/232/346/250/241/346/200/201/346/265/213/350/257/225.py +76 -0
- examples/_openaisdk//345/244/232/346/250/241/346/200/201/346/265/213/350/257/225_.py +56 -0
- examples/_openaisdk//346/226/207/344/273/266/351/227/256/347/255/224.py +36 -0
- examples/_openaisdk//346/226/207/346/241/243/350/247/243/346/236/220.py +34 -0
- examples/_openaisdk//346/250/241/345/236/213/346/265/213/350/257/225.py +53 -0
- examples/_openaisdk//351/230/277/351/207/214.py +80 -0
- {meutils/serving/jina/__demo → examples/ann}/__init__.py +1 -1
- examples/ann/main.py +31 -0
- examples/apis/kl.py +28 -0
- examples/apis/x.py +17 -0
- examples/apis/xx.py +17 -0
- examples/arq_demo/demo.py +3 -0
- examples/backgroundtasks.py +25 -0
- examples/bserver.py +513 -21
- examples/cache_demo/HermesCache_demo.py +81 -0
- examples/cache_demo/acacge.py +26 -0
- examples/cache_demo/x.py +31 -0
- {meutils/docarray_utils → examples/caches}/__init__.py +1 -1
- examples/caches/llmcache.py +18 -0
- examples/celery_demo/product_task.py +2 -0
- examples/demo.py +17 -1
- examples/fastapi_caching.py +59 -0
- {meutils/dependencies → examples/gr}/__init__.py +1 -1
- examples/gr/d.py +22 -0
- examples/gr/demo.py +30 -0
- examples/ip2/345/234/260/345/214/272.py +16 -0
- examples/jinja2_demo/j2_demo.py +20 -1
- examples/json/346/240/207/345/207/206/345/214/226.py +54 -0
- examples/md.py +29 -0
- {meutils/serving/jina → examples/nesc}/__init__.py +1 -1
- examples/nesc/main.py +76 -0
- examples/orm/mysql_orm.py +113 -0
- examples/orm/sql_creater.py +57 -0
- examples/orm/sqlm.py +134 -0
- examples/rq_demo/fns.py +18 -0
- examples/rq_demo/redis/351/230/237/345/210/227.py +14 -7
- examples/rq_demo/redis/351/230/237/345/210/227_add_chatfire.py +30 -0
- examples/size_map.py +43 -0
- examples/test.py +59 -0
- examples/webs/__init__.py +11 -0
- examples/webs/main.py +34 -0
- examples/x.py +13 -0
- examples//345/216/273/346/260/264/345/215/260.py +20 -0
- examples//346/226/207/346/241/243/346/231/272/350/203/275/__init__.py +11 -0
- meutils/_utils.py +15 -6
- meutils/ai_audio/asr/__init__.py +3 -2
- meutils/ai_audio/asr/cf_asr.py +53 -0
- meutils/ai_audio/asr/de.py +11 -0
- meutils/ai_audio/asr/fast_asr.py +15 -7
- meutils/ai_audio/asr/openai_asr.py +83 -6
- meutils/ai_audio/fast_asr.py +8 -4
- meutils/ai_audio/tts/EdgeTTS.py +33 -7
- meutils/ai_audio/tts/openai_tts.py +24 -20
- meutils/ai_audio/tts/tts_ui.py +1 -0
- meutils/ai_audio/utils.py +9 -0
- meutils/ai_cv/__init__.py +0 -1
- meutils/ai_cv/ocr.py +3 -2
- meutils/ai_cv/utils.py +154 -0
- meutils/ai_video/avmerge.py +6 -0
- meutils/ai_video/video.py +11 -2
- meutils/{api → apis}/__init__.py +1 -1
- meutils/apis/ali_apis.py +60 -0
- meutils/apis/audio/__init__.py +10 -0
- meutils/apis/audio/deepinfra.py +59 -0
- meutils/apis/audio/fish.py +248 -0
- meutils/apis/baidu/__init__.py +9 -0
- meutils/apis/baidu/bdaitpzs.py +229 -0
- meutils/apis/baidu/test.py +78 -0
- meutils/apis/chatglm/__init__.py +11 -0
- meutils/apis/chatglm/glm_video.py +273 -0
- meutils/apis/chatglm/glm_video_api.py +116 -0
- meutils/apis/chatglm/images.py +63 -0
- meutils/apis/chatglm/temp.py +259 -0
- meutils/apis/chatglm/x.py +31 -0
- meutils/{api → apis}/common.py +10 -6
- meutils/apis/fal/__init__.py +11 -0
- meutils/apis/fal/files.py +53 -0
- meutils/apis/fal/images.py +57 -0
- meutils/apis/fal/images_.py +72 -0
- meutils/apis/fal/videos.py +77 -0
- meutils/apis/firecrawl.py +45 -0
- meutils/apis/gitee/__init__.py +11 -0
- meutils/apis/gitee/images/__init__.py +9 -0
- meutils/apis/gitee/images/kolors.py +99 -0
- meutils/apis/hailuoai/__init__.py +11 -0
- meutils/apis/hailuoai/demo.py +34 -0
- meutils/apis/hailuoai/hasha_new.py +248 -0
- meutils/apis/hailuoai/music.py +11 -0
- meutils/apis/hailuoai/upload.py +116 -0
- meutils/apis/hailuoai/videos.py +460 -0
- meutils/apis/hailuoai/yy.py +242 -0
- meutils/apis/hf/__init__.py +11 -0
- meutils/apis/hf/got_ocr.py +64 -0
- meutils/apis/hf/gradio.py +34 -0
- meutils/apis/hf/hivisionidphotos.py +80 -0
- meutils/apis/hf/kolors.py +68 -0
- meutils/apis/hf/kolors_virtual_try_on.py +107 -0
- meutils/apis/hf/r.py +53 -0
- meutils/apis/hf/x.py +26 -0
- meutils/apis/hf//350/257/201/344/273/266/347/205/247.py +41 -0
- meutils/apis/hunyuan/__init__.py +11 -0
- meutils/apis/hunyuan/image_tools.py +84 -0
- meutils/apis/images/__init__.py +11 -0
- meutils/apis/images/deepinfra.py +92 -0
- meutils/apis/images/demo.py +150 -0
- meutils/apis/images/eidt.py +36 -0
- meutils/apis/images/flux/__init__.py +11 -0
- meutils/apis/images/flux/fluxpro.py +108 -0
- meutils/apis/images/flux/mystic.py +116 -0
- meutils/apis/images/ideogram/__init__.py +10 -0
- meutils/apis/images/ideogram/ideogram_images.py +193 -0
- meutils/apis/images/prodia/__init__.py +12 -0
- meutils/apis/images/prodia/faceswap.py +76 -0
- meutils/apis/images/recraft.py +152 -0
- meutils/apis/images/virtual_try_on/__init__.py +11 -0
- meutils/apis/images/virtual_try_on/images.py +65 -0
- meutils/apis/jiema/24mail.py +96 -0
- meutils/apis/jiema/__init__.py +11 -0
- meutils/apis/jiema/yezi.py +97 -0
- meutils/apis/jimeng/__init__.py +11 -0
- meutils/apis/jimeng/common.py +328 -0
- meutils/apis/jimeng/doubao.py +68 -0
- meutils/apis/jimeng/doubao_utils.py +175 -0
- meutils/apis/jimeng/files.py +263 -0
- meutils/apis/jimeng/images.py +140 -0
- meutils/apis/jimeng/lip_sync.py +11 -0
- meutils/apis/jina.py +55 -0
- meutils/apis/kling/__init__.py +11 -0
- meutils/apis/kling/api.py +60 -0
- meutils/apis/kling/images.py +174 -0
- meutils/apis/kling/kolors_virtual_try_on.py +111 -0
- meutils/apis/kling/kolors_virtual_try_on_web.py +126 -0
- meutils/apis/kling/videos.py +67 -0
- meutils/apis/kling//351/211/264/346/235/203.py +34 -0
- meutils/apis/kuaidi.py +32 -0
- meutils/apis/kuaishou/__init__.py +10 -0
- meutils/apis/kuaishou/klingai.py +523 -0
- meutils/apis/kuaishou/klingai_video.py +197 -0
- meutils/apis/kuaishou/kolors.py +189 -0
- meutils/apis/llm_qa.py +55 -0
- meutils/apis/luma/__init__.py +11 -0
- meutils/apis/luma/luma.py +123 -0
- meutils/apis/minicpm/__init__.py +9 -0
- meutils/apis/minicpm/luca.py +137 -0
- meutils/apis/monica/__init__.py +11 -0
- meutils/apis/monica/llm.py +11 -0
- meutils/apis/napkin/__init__.py +11 -0
- meutils/apis/napkin/icons.py +42 -0
- meutils/apis/niutrans.py +73 -0
- meutils/apis/oneapi/__init__.py +11 -0
- meutils/apis/oneapi/channel.py +68 -0
- meutils/apis/oneapi/common.py +135 -0
- meutils/apis/oneapi/log.py +47 -0
- meutils/apis/oneapi/token.py +48 -0
- meutils/apis/oneapi/token_.py +112 -0
- meutils/apis/oneapi/user.py +100 -0
- meutils/apis/oneapi/utils.py +47 -0
- meutils/apis/pixverse/__init__.py +11 -0
- meutils/apis/pixverse/pixverse.py +150 -0
- meutils/apis/proxy/__init__.py +11 -0
- meutils/apis/proxy/ips.py +178 -0
- meutils/apis/remini/__init__.py +11 -0
- meutils/apis/remini/remini.py +89 -0
- meutils/apis/replicateai/__init__.py +11 -0
- meutils/apis/replicateai/images.py +79 -0
- meutils/apis/replicateai/raw.py +53 -0
- meutils/apis/runwayml/__init__.py +10 -0
- meutils/apis/runwayml/gen.py +143 -0
- meutils/apis/search/__init__.py +11 -0
- meutils/apis/search/baichuan.py +11 -0
- meutils/apis/search/metaso.py +218 -0
- meutils/apis/search/metaso_.py +77 -0
- meutils/apis/search/n.py +99 -0
- meutils/apis/search/searxng.py +42 -0
- meutils/apis/search_music.py +39 -0
- meutils/apis/siliconflow/__init__.py +9 -0
- meutils/apis/siliconflow/audio.py +63 -0
- meutils/apis/siliconflow/image_to_image.py +116 -0
- meutils/apis/siliconflow/images.py +154 -0
- meutils/apis/siliconflow/rerankers.py +40 -0
- meutils/apis/siliconflow/text_to_image.py +132 -0
- meutils/apis/siliconflow/utils.py +66 -0
- meutils/apis/siliconflow/videos.py +102 -0
- meutils/apis/sunoai/__init__.py +10 -0
- meutils/apis/sunoai/haimian.py +135 -0
- meutils/apis/sunoai/suno.py +373 -0
- meutils/apis/textcard/__init__.py +11 -0
- meutils/apis/textcard/demo.py +25 -0
- meutils/apis/textcard/hanyuxinjie.py +81 -0
- meutils/apis/textin.py +159 -0
- meutils/apis/to_image/__init__.py +11 -0
- meutils/apis/to_image/html2image.py +29 -0
- meutils/apis/to_image/md.py +29 -0
- meutils/apis/to_image/url2image.py +41 -0
- meutils/apis/together/__init__.py +11 -0
- meutils/apis/together/images.py +80 -0
- meutils/apis/translator/__init__.py +9 -0
- meutils/apis/translator/deeplx.py +55 -0
- meutils/apis/tripo3d/__init__.py +11 -0
- meutils/apis/tripo3d/images.py +106 -0
- meutils/apis/ts.py +60 -0
- meutils/apis/uptime_kuma/__init__.py +11 -0
- meutils/apis/uptime_kuma/common.py +56 -0
- meutils/apis/uptime_kuma//345/233/275/344/272/247/345/210/206/347/273/204.py +68 -0
- meutils/apis/utils.py +47 -0
- meutils/apis/videos/__init__.py +11 -0
- meutils/apis/videos/sora.py +16 -0
- meutils/apis/vidu/__init__.py +9 -0
- meutils/apis/vidu/vidu_video.py +254 -0
- meutils/apis/vidu/x.py +14 -0
- meutils/apis/voice_clone/__init__.py +10 -0
- meutils/apis/voice_clone/fish.py +236 -0
- meutils/apis/voice_clone/fish_api.py +16 -0
- meutils/apis/web_search.py +31 -0
- meutils/apis/yezi.py +97 -0
- meutils/async_task/__init__.py +13 -0
- meutils/async_task/celery_config.py +106 -0
- meutils/async_task/common.py +37 -0
- meutils/async_task/demo_create_tasks.py +73 -0
- meutils/async_task/tasks/__init__.py +11 -0
- meutils/async_task/tasks/_all.py +20 -0
- meutils/async_task/tasks/hailuo.py +24 -0
- meutils/async_task/tasks/kling.py +30 -0
- meutils/async_task/tasks/replicateai.py +24 -0
- meutils/async_task/tasks/test.py +124 -0
- meutils/async_task/tasks/vidu.py +28 -0
- meutils/async_task/utils.py +191 -0
- meutils/async_task//351/200/232/347/224/250/350/256/276/350/256/241.py +119 -0
- meutils/async_utils/asyncer_.py +37 -0
- meutils/async_utils/background.py +68 -0
- meutils/async_utils/common.py +136 -16
- meutils/async_utils/test.py +47 -0
- meutils/cache_utils.py +29 -23
- meutils/caches/__init__.py +9 -0
- meutils/caches/acache.py +45 -0
- meutils/caches/redis_cache.py +63 -0
- meutils/clis/check_api.py +66 -0
- meutils/clis/cli.py +1 -1
- meutils/common.py +56 -17
- meutils/config_utils/__init__.py +11 -0
- meutils/config_utils/lark_utils/__init__.py +11 -0
- meutils/config_utils/lark_utils/common.py +385 -0
- meutils/config_utils/lark_utils/demo.py +13 -0
- meutils/config_utils/lark_utils/x.py +50 -0
- meutils/config_utils/manager.py +108 -0
- meutils/crawlers/__init__.py +11 -0
- meutils/data/VERSION +1 -1
- meutils/data/cowboy-hat-face.webp +0 -0
- meutils/data/oneapi/FOOTER.md +7 -0
- meutils/data/oneapi/NOTICE.md +138 -0
- meutils/data/oneapi/__init__.py +15 -0
- meutils/db/orm.py +179 -0
- meutils/db/redis_db.py +87 -0
- meutils/decorators/cache.py +1 -1
- meutils/decorators/common.py +84 -5
- meutils/decorators/contextmanagers.py +17 -6
- meutils/decorators/fastapi_decorator.py +77 -3
- meutils/decorators/polling.py +46 -0
- meutils/decorators/retry.py +150 -26
- meutils/fastapi_utils/__init__.py +11 -0
- meutils/fastapi_utils/exceptions/http_error.py +72 -0
- meutils/fastapi_utils/exceptions/validation_error.py +44 -0
- meutils/hash_utils.py +9 -4
- meutils/hooks/__init__.py +11 -0
- meutils/hooks/hook_test.py +174 -0
- meutils/hooks/wechat.py +162 -0
- meutils/hooks/wechat_channel.py +303 -0
- meutils/init/evn.py +1 -1
- meutils/io/files_utils.py +232 -0
- meutils/io/image.py +148 -10
- meutils/io/x.py +75 -0
- meutils/llm/__init__.py +10 -0
- meutils/llm/check_api.py +109 -0
- meutils/llm/check_utils.py +106 -0
- meutils/llm/clients.py +38 -0
- meutils/llm/completions/__init__.py +11 -0
- meutils/llm/completions/agents/__init__.py +11 -0
- meutils/llm/completions/agents/file.py +125 -0
- meutils/llm/completions/cp.py +112 -0
- meutils/llm/completions/delilegal.py +135 -0
- meutils/llm/completions/dify.py +81 -0
- meutils/llm/completions/kimi.py +47 -0
- meutils/llm/completions/modelscope.py +11 -0
- meutils/{fileparser/filetype.py → llm/completions/oi.py} +5 -3
- meutils/llm/completions/rag/__init__.py +11 -0
- meutils/llm/completions/rag/fire.py +157 -0
- meutils/llm/completions/rag/qwen.py +11 -0
- meutils/llm/completions/rag/rag.py +41 -0
- meutils/llm/completions/rag.py +38 -0
- meutils/llm/completions/tryblend.py +201 -0
- meutils/llm/completions/tune.py +284 -0
- meutils/llm/completions/x.py +26 -0
- meutils/llm/completions/xx.py +61 -0
- meutils/llm/completions/yuanbao.py +176 -0
- meutils/llm/demo.py +114 -0
- meutils/llm/functions/__init__.py +11 -0
- meutils/llm/mappers.py +15 -0
- meutils/llm/openai_utils/__init__.py +11 -0
- meutils/llm/openai_utils/common.py +284 -0
- meutils/llm/openai_utils/tool_outputs.py +45 -0
- meutils/llm/output_parsers/__init__.py +80 -0
- meutils/llm/prompts/__init__.py +244 -0
- meutils/llm/prompts/demo.py +155 -0
- meutils/llm/prompts/html2image_test.py +19 -0
- meutils/llm/utils.py +133 -0
- meutils/llm/x.py +75 -0
- meutils/notice/feishu.py +40 -9
- meutils/notice/wechat.py +23 -21
- meutils/np_utils.py +10 -1
- meutils/office_automation/pdf.py +6 -1
- meutils/oss/__init__.py +20 -0
- meutils/oss/minio_oss.py +184 -0
- meutils/oss/minio_utils.py +48 -0
- meutils/other/__demo.py +6 -4
- meutils/pandas_utils/__init__.py +1 -0
- meutils/pandas_utils/common.py +31 -0
- meutils/pandas_utils/pd_utils.py +10 -6
- meutils/parsers/__init__.py +10 -0
- meutils/parsers/file_parsers.py +110 -0
- meutils/parsers/fileparser/demo.py +41 -0
- meutils/parsers/fileparser/filetype.py +41 -0
- meutils/pay.py +37 -0
- meutils/pipe.py +37 -4
- meutils/playwright_utils/common.py +20 -12
- meutils/plots/common.py +35 -34
- meutils/queues/demo.py +56 -0
- meutils/queues/smooth_queue.py +120 -0
- meutils/queues/uniform_queue.py +3 -1
- meutils/request_utils/__init__.py +26 -2
- meutils/request_utils/ark.py +47 -0
- meutils/request_utils/crawler.py +34 -5
- meutils/request_utils/jwt_utils/__init__.py +11 -0
- meutils/request_utils/jwt_utils/common.py +42 -0
- meutils/request_utils/volc.py +160 -0
- meutils/schemas/__init__.py +0 -1
- meutils/schemas/baidu_types.py +70 -0
- meutils/schemas/batch_types.py +450 -0
- meutils/schemas/celery_types.py +64 -0
- meutils/schemas/chatfire_types.py +15 -0
- meutils/schemas/chatglm_types.py +197 -0
- meutils/schemas/db/__init__.py +11 -0
- meutils/schemas/db/oneapi_types.py +117 -0
- meutils/schemas/dify_types.py +40 -0
- meutils/schemas/embedding.py +31 -0
- meutils/schemas/fal_types.py +13 -0
- meutils/schemas/fish_types.py +11 -0
- meutils/schemas/hailuo_types.py +208 -0
- meutils/schemas/haimian_types.py +51 -0
- meutils/schemas/idphoto_types.py +43 -0
- meutils/schemas/image_types.py +476 -0
- meutils/schemas/jimeng_types.py +28 -0
- meutils/schemas/jina_types.py +67 -0
- meutils/schemas/kimi_types.py +86 -0
- meutils/schemas/kling_types.py +235 -0
- meutils/schemas/kuaishou_types.py +328 -0
- meutils/schemas/luma_types.py +59 -0
- meutils/schemas/message_types.py +165 -0
- meutils/schemas/message_types_.py +219 -0
- meutils/schemas/metaso_types.py +64 -0
- meutils/schemas/napkin_types.py +85 -0
- meutils/schemas/ocr_types.py +37 -0
- meutils/schemas/oneapi/__init__.py +11 -0
- meutils/schemas/oneapi/_types.py +49 -0
- meutils/schemas/oneapi/common.py +883 -0
- meutils/schemas/oneapi/icons.py +30 -0
- meutils/schemas/oneapi/model_group_info.py +48 -0
- meutils/schemas/oneapi/model_info.py +34 -0
- meutils/schemas/oneapi/models.py +26 -0
- meutils/schemas/oneapi/x.py +26 -0
- meutils/schemas/oneapi//351/207/215/345/256/232/345/220/221.py +132 -0
- meutils/schemas/openai_api_protocol.py +411 -0
- meutils/schemas/openai_types.py +366 -0
- meutils/schemas/pixverse_types.py +88 -0
- meutils/schemas/playwright_types.py +57 -0
- meutils/schemas/prodia_types.py +19 -0
- meutils/schemas/replicate_types.py +112 -0
- meutils/schemas/request_types.py +20 -0
- meutils/schemas/runwayml_types.py +190 -0
- meutils/schemas/siliconflow_types.py +80 -0
- meutils/schemas/step_types.py +19 -0
- meutils/schemas/suno_types.py +319 -0
- meutils/schemas/task_types.py +192 -0
- meutils/schemas/translator_types.py +29 -0
- meutils/schemas/tripo3d_types.py +57 -0
- meutils/schemas/tryblend_types.py +51 -0
- meutils/schemas/video_types.py +62 -0
- meutils/schemas/vidu_types.py +350 -0
- meutils/schemas/wechat_types.py +316 -0
- meutils/schemas/yuanbao_types.py +260 -0
- meutils/serving/celery/__init__.py +8 -0
- meutils/serving/celery/config.py +115 -0
- meutils/serving/celery/router.py +4 -6
- meutils/serving/celery/tasks.py +6 -4
- meutils/serving/celery//351/200/232/347/224/250/350/256/276/350/256/241.py +119 -0
- meutils/serving/fastapi/common.py +27 -31
- meutils/serving/fastapi/dependencies/__init__.py +0 -1
- meutils/serving/fastapi/dependencies/auth.py +55 -2
- meutils/serving/fastapi/exceptions/http_error.py +67 -2
- meutils/serving/fastapi/exceptions/validation_error.py +18 -2
- meutils/serving/fastapi/lifespans.py +73 -0
- meutils/serving/fastapi/routers/scheduler.py +12 -0
- meutils/serving/fastapi/routers/screenshot.py +47 -0
- meutils/serving/fastapi/routers/spider.py +8 -3
- meutils/serving/fastapi/routers/task.py +48 -0
- meutils/serving/fastapi/utils.py +48 -1
- meutils/serving/streamlit/common.py +1 -1
- meutils/smooth_utils.py +3 -0
- meutils/str_utils/__init__.py +22 -3
- meutils/str_utils/json_utils.py +7 -0
- meutils/str_utils/regular_expression.py +102 -10
- meutils/templates/xx.html +21 -0
- meutils/templates/xxx.html +117 -0
- meutils/todo.py +12 -0
- meutils/tools/token_monitor.py +33 -0
- MeUtils-2024.3.4.13.4.45.dist-info/RECORD +0 -540
- meutils/ai_audio/asr/subtitle.srt +0 -40
- meutils/ai_audio/demo.ipynb +0 -1215
- meutils/ai_audio/example.srt +0 -348
- meutils/ai_audio/new.srt +0 -179
- meutils/ai_audio/subtitles.srt +0 -696
- meutils/ai_audio/tts/new.srt +0 -179
- meutils/ai_audio//350/247/206/351/242/221/345/220/210/345/271/266.sh +0 -32
- meutils/ai_cv/1.jpg +0 -0
- meutils/ai_cv/197.jpg +0 -0
- meutils/ai_cv/2.jpg +0 -0
- meutils/ai_cv/img.png +0 -0
- meutils/ai_cv/invoice.jpg +0 -0
- meutils/ai_cv/tbl.png +0 -0
- meutils/ai_cv/test.png +0 -0
- meutils/ann/README.md +0 -33
- meutils/ann/README_gensim.md +0 -47
- meutils/ann/examples/client.py +0 -59
- meutils/ann/examples/demo.py +0 -24
- meutils/api/deeplx.py +0 -29
- meutils/api/qr.png +0 -0
- meutils/clis/README.md +0 -29
- meutils/clis/__test.sh +0 -17
- meutils/clis/deepseek.txt +0 -8
- meutils/clis/deepseek_13003330042.json +0 -1
- meutils/clis/deepseek_13003872192.json +0 -1
- meutils/clis/deepseek_13852263862.json +0 -1
- meutils/clis/deepseek_13913898681.json +0 -1
- meutils/clis/deepseek_13962978617.json +0 -1
- meutils/clis/deepseek_15251801790.json +0 -1
- meutils/clis/deepseek_15720826383.json +0 -1
- meutils/clis/deepseek_18395563611.json +0 -1
- meutils/clis/deepseek_313303303@qq.com.json +0 -1
- meutils/clis/kimi_state.json +0 -1
- meutils/cmds/README.md +0 -55
- meutils/coding/__init__.py +0 -11
- meutils/coding/find132.py +0 -40
- meutils/db/README.md +0 -51
- meutils/decorators/README.md +0 -17
- meutils/docarray_utils/demo_es.py +0 -34
- meutils/docarray_utils/demo_hnsw.py +0 -55
- meutils/docarray_utils/in_memory.py +0 -38
- meutils/docarray_utils//346/224/271/351/200/240/344/270/213hnsw.py +0 -43
- meutils/io/file.py +0 -20
- meutils/io/img.png +0 -0
- meutils/io/x.yml +0 -1
- meutils/notice/img.png +0 -0
- meutils/notice/todo.md +0 -10
- meutils/office_automation//346/212/225/350/265/204/347/256/241/347/220/206/347/263/273/347/273/237O3.2_/344/272/244/346/230/223/347/273/204.pdm +0 -22469
- meutils/playwright_utils/__test.sh +0 -2
- meutils/playwright_utils/kimi1_cookies.json +0 -1
- meutils/playwright_utils/kimi2_cookies.json +0 -1
- meutils/playwright_utils/kimi_cookies.json +0 -93
- meutils/serving/README.md +0 -1
- meutils/serving/celery/_run.sh +0 -10
- meutils/serving/gui/run.sh +0 -9
- meutils/serving/jina/__demo/client.py +0 -42
- meutils/serving/jina/__demo/flow.svg +0 -1
- meutils/serving/jina/__demo/s.py +0 -34
- meutils/serving/jina/__demo/s2.py +0 -37
- meutils/serving/jina/__demo/server.py +0 -83
- meutils/serving/jina/__demo/test.py +0 -40
- meutils/serving/jina/executors/SentenceEncoder.py +0 -62
- meutils/serving/jina/executors/SentenceEncoder_.py +0 -63
- meutils/serving/jina/executors/__init__.py +0 -46
- meutils/serving/jina/executors/base.py +0 -40
- meutils/serving/jina/nlp_serving/__init__.py +0 -11
- meutils/serving/jina/nlp_serving/word_segmentation.py +0 -40
- meutils/serving/streamlit/conf.yaml +0 -5
- meutils/serving/streamlit/ocr.png +0 -0
- meutils/serving/streamlit/run.sh +0 -17
- meutils/serving/webui/.streamlit/_config.toml +0 -186
- meutils/serving/webui/.streamlit/config.toml +0 -26
- meutils/serving/webui/pages/_1_/345/210/206/350/257/215.py +0 -56
- meutils/serving/webui/pages/_2_/350/257/215/346/200/247/346/240/207/346/263/250/344/270/216/345/256/236/344/275/223/350/257/206/345/210/253.py +0 -54
- meutils/serving/webui/pages/_3_/346/226/207/346/234/254/345/214/271/351/205/215.py +0 -64
- meutils/serving/webui/run.sh +0 -9
- meutils/spark/__init__.py +0 -26
- meutils/tools/monitor.yml +0 -29
- {MeUtils-2024.3.4.13.4.45.dist-info → MeUtils-2025.1.16.17.15.52.dist-info}/LICENSE +0 -0
- {MeUtils-2024.3.4.13.4.45.dist-info → MeUtils-2025.1.16.17.15.52.dist-info}/top_level.txt +0 -0
- {meutils → examples}/comp_utils/__init__.py +0 -0
- {meutils → examples}/comp_utils/reverse_metric.py +0 -0
- /meutils/{fileparser/README.md → fastapi_utils/exceptions/__init__.py} +0 -0
- /meutils/{fileparser → parsers/fileparser}/PDF/346/212/275/345/217/226.py" +0 -0
- /meutils/{fileparser → parsers/fileparser}/__init__.py +0 -0
- /meutils/{fileparser → parsers/fileparser}/common.py +0 -0
- /meutils/{fileparser → parsers/fileparser}/filetype/__init__.py +0 -0
- /meutils/{fileparser → parsers/fileparser}/filetype/__main__.py +0 -0
- /meutils/{fileparser → parsers/fileparser}/filetype/filetype.py +0 -0
- /meutils/{fileparser → parsers/fileparser}/filetype/helpers.py +0 -0
- /meutils/{fileparser → parsers/fileparser}/filetype/match.py +0 -0
- /meutils/{fileparser → parsers/fileparser}/filetype/types/__init__.py +0 -0
- /meutils/{fileparser → parsers/fileparser}/filetype/types/application.py +0 -0
- /meutils/{fileparser → parsers/fileparser}/filetype/types/archive.py +0 -0
- /meutils/{fileparser → parsers/fileparser}/filetype/types/audio.py +0 -0
- /meutils/{fileparser → parsers/fileparser}/filetype/types/base.py +0 -0
- /meutils/{fileparser → parsers/fileparser}/filetype/types/document.py +0 -0
- /meutils/{fileparser → parsers/fileparser}/filetype/types/font.py +0 -0
- /meutils/{fileparser → parsers/fileparser}/filetype/types/image.py +0 -0
- /meutils/{fileparser → parsers/fileparser}/filetype/types/isobmff.py +0 -0
- /meutils/{fileparser → parsers/fileparser}/filetype/types/video.py +0 -0
- /meutils/{fileparser → parsers/fileparser}/filetype/utils.py +0 -0
- /meutils/{fileparser → parsers/fileparser}/pdf.py +0 -0
- /meutils/{fileparser → parsers/fileparser}//350/241/250/346/240/274/346/212/275/345/217/226.py" +0 -0
@@ -0,0 +1,284 @@
|
|
1
|
+
#!/usr/bin/env python
|
2
|
+
# -*- coding: utf-8 -*-
|
3
|
+
# @Project : AI. @by PyCharm
|
4
|
+
# @File : tune
|
5
|
+
# @Time : 2024/9/20 13:51
|
6
|
+
# @Author : betterme
|
7
|
+
# @WeChat : meutils
|
8
|
+
# @Software : PyCharm
|
9
|
+
# @Description :
|
10
|
+
|
11
|
+
from meutils.pipe import *
|
12
|
+
from meutils.apis.proxy.ips import get_one_proxy
|
13
|
+
from meutils.decorators.retry import retrying
|
14
|
+
from meutils.str_utils.regular_expression import remove_date_suffix
|
15
|
+
from meutils.notice.feishu import send_message as _send_message
|
16
|
+
from meutils.config_utils.lark_utils import aget_spreadsheet_values, get_next_token_for_polling
|
17
|
+
|
18
|
+
from meutils.llm.utils import oneturn2multiturn
|
19
|
+
from meutils.schemas.openai_types import chat_completion, chat_completion_chunk, ChatCompletionRequest, CompletionUsage
|
20
|
+
|
21
|
+
BASE_URL = "https://chat.tune.app"
|
22
|
+
FEISHU_URL_VIP = "https://xchatllm.feishu.cn/sheets/Bmjtst2f6hfMqFttbhLcdfRJnNf?sheet=gCrlN4"
|
23
|
+
FEISHU_URL_API = "https://xchatllm.feishu.cn/sheets/Bmjtst2f6hfMqFttbhLcdfRJnNf?sheet=9HwQtX"
|
24
|
+
"https://chat.tune.app/api/models"
|
25
|
+
# r = requests.get("https://chat.tune.app/tune-api/appConfig")
|
26
|
+
# r = requests.get("https://chat.tune.app/api/guestLogin")
|
27
|
+
|
28
|
+
# https://chat.tune.app/api/models?noCache=true
|
29
|
+
|
30
|
+
send_message = partial(
|
31
|
+
_send_message,
|
32
|
+
url="https://open.feishu.cn/open-apis/bot/v2/hook/e0db85db-0daf-4250-9131-a98d19b909a9",
|
33
|
+
title=__name__
|
34
|
+
)
|
35
|
+
|
36
|
+
# data = httpx.get("https://chat.tune.app/api/models?noCache=true", timeout=30).json()
|
37
|
+
# models = jsonpath.jsonpath(data, expr='$..id')
|
38
|
+
#
|
39
|
+
# DEFAULT_MODEL = "openai/gpt-4o-mini" # "kaushikaakash04/tune-blob"
|
40
|
+
# MODELS = {model.split('/')[-1]: model for model in models}
|
41
|
+
|
42
|
+
# logger.debug(bjson(MODELS))
|
43
|
+
|
44
|
+
|
45
|
+
DEFAULT_MODEL = "openai/gpt-4o-mini"
|
46
|
+
MODELS = {
|
47
|
+
"o1-mini": "openai/o1-mini",
|
48
|
+
"o1-preview": "openai/o1-preview",
|
49
|
+
"o1-mini-2024-09-12": "openai/o1-mini",
|
50
|
+
"o1-preview-2024-09-12": "openai/o1-preview",
|
51
|
+
"gpt-4o": "openai/gpt-4o",
|
52
|
+
"gpt-4o-mini": "openai/gpt-4o-mini",
|
53
|
+
|
54
|
+
"claude-3.5-sonnet": "anthropic/claude-3.5-sonnet",
|
55
|
+
"claude-3.5-haiku": "anthropic/claude-3.5-haiku",
|
56
|
+
"claude-3-haiku": "anthropic/claude-3-haiku",
|
57
|
+
|
58
|
+
"claude-3.5-sonnet-20240620": "anthropic/claude-3.5-sonnet",
|
59
|
+
"claude-3.5-sonnet-20241022": "anthropic/claude-3.5-sonnet",
|
60
|
+
"claude-3.5-haiku-20241022": "anthropic/claude-3.5-haiku",
|
61
|
+
|
62
|
+
"gemini-flash-1.5-8b": "google/gemini-flash-1.5-8b",
|
63
|
+
|
64
|
+
"tune-blob": "kaushikaakash04/tune-blob",
|
65
|
+
"llama-3.1-8b-instruct": "meta/llama-3.1-8b-instruct",
|
66
|
+
"pixtral-12B-2409": "mistral/pixtral-12B-2409",
|
67
|
+
"qwen-2.5-coder-32b": "qwen/qwen-2.5-coder-32b",
|
68
|
+
"llama-3.1-70b-instruct": "meta/llama-3.1-70b-instruct",
|
69
|
+
"llama-3.2-90b-vision": "meta/llama-3.2-90b-vision",
|
70
|
+
"llama3.1-euryale-70b": "sao10k/llama3.1-euryale-70b",
|
71
|
+
"qwen-2.5-72b": "qwen/qwen-2.5-72b",
|
72
|
+
"openrouter-goliath-120b-4k": "rohan/openrouter-goliath-120b-4k",
|
73
|
+
"mistral-large": "mistral/mistral-large",
|
74
|
+
"llama-3.1-405b-instruct": "meta/llama-3.1-405b-instruct",
|
75
|
+
"mixtral-8x7b-instruct": "mistral/mixtral-8x7b-instruct",
|
76
|
+
"hermes-3-llama-3.1-405b": "nousresearch/hermes-3-llama-3.1-405b",
|
77
|
+
"tune-wizardlm-2-8x22b": "rohan/tune-wizardlm-2-8x22b",
|
78
|
+
|
79
|
+
"tune-mythomax-l2-13b": "rohan/tune-mythomax-l2-13b",
|
80
|
+
"qwen-2-vl-72b": "qwen/qwen-2-vl-72b",
|
81
|
+
}
|
82
|
+
|
83
|
+
|
84
|
+
#
|
85
|
+
# html_content = httpx.get(url).text
|
86
|
+
#
|
87
|
+
# # 正则表达式匹配以 "/_next/static/chunks/7116-" 开头的 JS 文件
|
88
|
+
# pattern = r'(/_next/static/chunks/7116-[^"]+\.js)'
|
89
|
+
#
|
90
|
+
# # 使用 re.findall() 找到所有匹配项
|
91
|
+
# matches = re.findall(pattern, html_content)
|
92
|
+
|
93
|
+
|
94
|
+
#
|
95
|
+
# "/_next/static/chunks/7116-aed224a0caaab94c.js"
|
96
|
+
#
|
97
|
+
# # 打印结果
|
98
|
+
# for match in matches:
|
99
|
+
# print(match)
|
100
|
+
|
101
|
+
|
102
|
+
@alru_cache(ttl=1000)
|
103
|
+
@retrying(predicate=lambda r: r is None)
|
104
|
+
async def get_access_token():
|
105
|
+
async with httpx.AsyncClient(base_url=BASE_URL, timeout=30) as client:
|
106
|
+
response = await client.get("/api/guestLogin")
|
107
|
+
|
108
|
+
logger.debug(response.status_code)
|
109
|
+
logger.debug(response.text)
|
110
|
+
|
111
|
+
if response.is_success:
|
112
|
+
return response.json()["accessToken"]
|
113
|
+
|
114
|
+
|
115
|
+
@alru_cache(ttl=60)
|
116
|
+
@retrying(predicate=lambda r: r is None)
|
117
|
+
async def create_conversation_id(token: str):
|
118
|
+
headers = {
|
119
|
+
"authorization": token
|
120
|
+
}
|
121
|
+
conversation_id = str(uuid.uuid4()) # shortuuid.random()
|
122
|
+
# conversation_id = "af306c40-8f85-47a7-a027-185da084c6cc"
|
123
|
+
params = {
|
124
|
+
"conversation_id": conversation_id,
|
125
|
+
"organization_id": "undefined",
|
126
|
+
"model": "anthropic/claude-3.5-sonnet", # "kaushikaakash04/tune-blob",
|
127
|
+
"currency": "USD"
|
128
|
+
}
|
129
|
+
async with httpx.AsyncClient(base_url=BASE_URL, headers=headers, timeout=30) as client:
|
130
|
+
response = await client.post("/api/new", params=params)
|
131
|
+
logger.debug(response.status_code)
|
132
|
+
logger.debug(response.text)
|
133
|
+
return conversation_id
|
134
|
+
|
135
|
+
# response = await client.get(f"/?id={conversation_id}", headers=headers)
|
136
|
+
# html_content = response.text
|
137
|
+
# pattern = r'(/_next/static/chunks/7116-[^"]+\.js)'
|
138
|
+
# js = re.findall(pattern, html_content)[-1]
|
139
|
+
#
|
140
|
+
# logger.debug(js)
|
141
|
+
#
|
142
|
+
# response = await client.get(js)
|
143
|
+
#
|
144
|
+
# logger.debug(response.status_code)
|
145
|
+
# next_action = response.text.split(',A=(0,a.$)("')[1][:40]
|
146
|
+
# logger.debug(next_action) # 2bc738a7215e149dbd4601a440f3b6df45089338 过期时间
|
147
|
+
#
|
148
|
+
# headers = {
|
149
|
+
# 'next-action': next_action,
|
150
|
+
# 'Cookie': f"AccessToken={token}",
|
151
|
+
#
|
152
|
+
# 'content-type': 'text/plain;charset=UTF-8',
|
153
|
+
# }
|
154
|
+
# payload = "[]"
|
155
|
+
# response = await client.post(f"/?id={conversation_id}", headers=headers, content=payload)
|
156
|
+
#
|
157
|
+
# logger.debug(response.status_code)
|
158
|
+
# logger.debug(response.text)
|
159
|
+
#
|
160
|
+
# return conversation_id
|
161
|
+
|
162
|
+
|
163
|
+
@retrying(max_retries=3)
|
164
|
+
async def create(request: ChatCompletionRequest, token: Optional[str] = None, vip: Optional[bool] = False):
|
165
|
+
request.model = remove_date_suffix(request.model)
|
166
|
+
|
167
|
+
if vip:
|
168
|
+
token = await get_next_token_for_polling(feishu_url=FEISHU_URL_VIP)
|
169
|
+
|
170
|
+
token = token or await get_access_token()
|
171
|
+
conversation_id = await create_conversation_id(token)
|
172
|
+
|
173
|
+
logger.debug(conversation_id)
|
174
|
+
|
175
|
+
use_search = False
|
176
|
+
if request.messages[0].get('role') != 'system': # 还原系统信息
|
177
|
+
request.messages.insert(0, {'role': 'system', 'content': f'You are {request.model}'})
|
178
|
+
|
179
|
+
if request.model.startswith("net-") or request.model.endswith("all"):
|
180
|
+
request.model = "kaushikaakash04/tune-blob"
|
181
|
+
use_search = True
|
182
|
+
else:
|
183
|
+
request.model = MODELS.get(request.model, DEFAULT_MODEL)
|
184
|
+
logger.debug(request)
|
185
|
+
|
186
|
+
headers = {
|
187
|
+
"authorization": token,
|
188
|
+
# 'Cookie': f"AccessToken={token};",
|
189
|
+
"content-type": "text/plain;charset=UTF-8",
|
190
|
+
|
191
|
+
}
|
192
|
+
params = {
|
193
|
+
# "organization_id": "undefined",
|
194
|
+
# "organization_id": "eb0fb996-2317-467b-9847-15f6c40000b7",
|
195
|
+
"retry": 2,
|
196
|
+
}
|
197
|
+
payload = {
|
198
|
+
# "query": request.last_content,
|
199
|
+
"query": oneturn2multiturn(request.messages),
|
200
|
+
# "images": request.urls, # todo: 兼容base64
|
201
|
+
|
202
|
+
"conversation_id": conversation_id,
|
203
|
+
"model_id": request.model, # "kaushikaakash04/tune-blob"
|
204
|
+
"browseWeb": use_search,
|
205
|
+
"attachement": "",
|
206
|
+
"attachment_name": "",
|
207
|
+
# "messageId": "4a33e497-efb7-4d8f-ae45-9aa7d2c1c5af1726811555410",
|
208
|
+
# "prevMessageId": "4a33e497-efb7-4d8f-ae45-9aa7d2c1c5af1726811555410",
|
209
|
+
|
210
|
+
"check": "286600"
|
211
|
+
}
|
212
|
+
|
213
|
+
yield "\n" # 提升首字速度
|
214
|
+
async with httpx.AsyncClient(base_url=BASE_URL, headers=headers, timeout=300) as client:
|
215
|
+
async with client.stream("POST", "/api/prompt", json=payload, params=params) as response:
|
216
|
+
logger.debug(response.status_code)
|
217
|
+
# logger.debug(response.text)
|
218
|
+
|
219
|
+
async for chunk in response.aiter_lines():
|
220
|
+
# logger.debug(chunk)
|
221
|
+
|
222
|
+
# if chunk == '{"value":""}':
|
223
|
+
# async for _chunk in create(request, token, vip):
|
224
|
+
# yield _chunk
|
225
|
+
# return
|
226
|
+
|
227
|
+
if chunk and chunk.startswith("{"):
|
228
|
+
chunk = (
|
229
|
+
chunk.replace("Blob", "OpenAI")
|
230
|
+
.replace("TuneAI", "OpenAI")
|
231
|
+
.replace("TuneStudio", "OpenAI")
|
232
|
+
.replace("Tune", "OpenAI")
|
233
|
+
.replace("https://studio.tune.app", "https://openai.com")
|
234
|
+
.replace("https://tunehq.ai", "https://openai.com")
|
235
|
+
.replace("https://chat.tune.app", "https://openai.com")
|
236
|
+
)
|
237
|
+
|
238
|
+
try:
|
239
|
+
chunk = json.loads(chunk)
|
240
|
+
chunk = chunk.get('value', "")
|
241
|
+
yield chunk
|
242
|
+
# break
|
243
|
+
|
244
|
+
except Exception as e:
|
245
|
+
_ = f"{e}\n{chunk}"
|
246
|
+
logger.error(_)
|
247
|
+
send_message(_)
|
248
|
+
yield ""
|
249
|
+
elif chunk.strip():
|
250
|
+
logger.debug(chunk)
|
251
|
+
|
252
|
+
# {"value": "完成"}
|
253
|
+
|
254
|
+
|
255
|
+
if __name__ == '__main__':
|
256
|
+
# arun(get_access_token())
|
257
|
+
# arun(create_conversation_id(None))
|
258
|
+
|
259
|
+
# model = "claude-3.5-sonnet"
|
260
|
+
# model = "net-anthropic/claude-3.5-sonnet"
|
261
|
+
# model = "all"
|
262
|
+
|
263
|
+
model = "kaushikaakash04/tune-blob"
|
264
|
+
# model = "openai/o1-mini"
|
265
|
+
# model = "o1-mini-0609"
|
266
|
+
|
267
|
+
# model = "openai/gpt-4o-mini"
|
268
|
+
|
269
|
+
url = "https://d2e931syjhr5o9.cloudfront.net/bb5cc8e4-5cc2-45c5-8307-585b7ec6eada_1731634172097.png"
|
270
|
+
|
271
|
+
request = ChatCompletionRequest(model=model, messages=[
|
272
|
+
{'role': 'user', 'content': '你的知识库截止的几月份'},
|
273
|
+
# {
|
274
|
+
# 'role': 'user', 'content': [
|
275
|
+
# {"type": "text", "text": "解释下这张图片"},
|
276
|
+
# {"type": "image_url", "image_url": {"url": url}},
|
277
|
+
# ]
|
278
|
+
# }
|
279
|
+
])
|
280
|
+
|
281
|
+
# arun(create(request, token=token, vip=True))
|
282
|
+
# arun(create(request, vip=True))
|
283
|
+
# arun(create(request, vip=False))
|
284
|
+
arun(create(request))
|
@@ -0,0 +1,26 @@
|
|
1
|
+
#!/usr/bin/env python
|
2
|
+
# -*- coding: utf-8 -*-
|
3
|
+
# @Project : AI. @by PyCharm
|
4
|
+
# @File : x
|
5
|
+
# @Time : 2024/9/20 17:59
|
6
|
+
# @Author : betterme
|
7
|
+
# @WeChat : meutils
|
8
|
+
# @Software : PyCharm
|
9
|
+
# @Description :
|
10
|
+
|
11
|
+
from meutils.pipe import *
|
12
|
+
|
13
|
+
from meutils.config_utils.lark_utils import aget_spreadsheet_values, get_next_token_for_polling
|
14
|
+
|
15
|
+
|
16
|
+
|
17
|
+
if __name__ == '__main__':
|
18
|
+
feishu_url = 'https://xchatllm.feishu.cn/sheets/Bmjtst2f6hfMqFttbhLcdfRJnNf?sheet=hxVlQw'
|
19
|
+
d = arun(aget_spreadsheet_values(feishu_url=feishu_url, to_dataframe=True))
|
20
|
+
# d = arun(aget_spreadsheet_values(feishu_url=feishu_url))
|
21
|
+
|
22
|
+
|
23
|
+
|
24
|
+
|
25
|
+
|
26
|
+
|
@@ -0,0 +1,61 @@
|
|
1
|
+
#!/usr/bin/env python
|
2
|
+
# -*- coding: utf-8 -*-
|
3
|
+
# @Project : AI. @by PyCharm
|
4
|
+
# @File : xx
|
5
|
+
# @Time : 2024/11/15 09:19
|
6
|
+
# @Author : betterme
|
7
|
+
# @WeChat : meutils
|
8
|
+
# @Software : PyCharm
|
9
|
+
# @Description :
|
10
|
+
|
11
|
+
from meutils.pipe import *
|
12
|
+
|
13
|
+
# ResponseT = TypeVar(
|
14
|
+
# "ResponseT",
|
15
|
+
# bound=Union[
|
16
|
+
# object,
|
17
|
+
# str,
|
18
|
+
# None,
|
19
|
+
# "BaseModel",
|
20
|
+
# List[Any],
|
21
|
+
# Dict[str, Any],
|
22
|
+
# Response,
|
23
|
+
# ModelBuilderProtocol,
|
24
|
+
# "APIResponse[Any]",
|
25
|
+
# "AsyncAPIResponse[Any]",
|
26
|
+
# "HttpxBinaryResponseContent",
|
27
|
+
# ],
|
28
|
+
# )
|
29
|
+
|
30
|
+
payload = {
|
31
|
+
"messages": [
|
32
|
+
{
|
33
|
+
"role": "user",
|
34
|
+
"content": "hi"
|
35
|
+
}
|
36
|
+
],
|
37
|
+
"use_search": False,
|
38
|
+
"extend": {
|
39
|
+
"sidebar": True
|
40
|
+
},
|
41
|
+
"kimiplus_id": "kimi",
|
42
|
+
"use_research": False,
|
43
|
+
"use_math": False,
|
44
|
+
"refs": [],
|
45
|
+
"refs_file": []
|
46
|
+
}
|
47
|
+
|
48
|
+
from openai import OpenAI
|
49
|
+
|
50
|
+
client = OpenAI(
|
51
|
+
base_url="https://kimi.moonshot.cn",
|
52
|
+
api_key="eyJhbGciOiJIUzUxMiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJ1c2VyLWNlbnRlciIsImV4cCI6MTczOTY4NDk3NywiaWF0IjoxNzMxOTA4OTc3LCJqdGkiOiJjc3RkYXNmZDBwODBpaGtkNTY4ZyIsInR5cCI6ImFjY2VzcyIsImFwcF9pZCI6ImtpbWkiLCJzdWIiOiJja2kwOTRiM2Flc2xnbGo2Zm8zMCIsInNwYWNlX2lkIjoiY2tpMDk0YjNhZXNsZ2xqNmZvMmciLCJhYnN0cmFjdF91c2VyX2lkIjoiY2tpMDk0YjNhZXNsZ2xqNmZvMzAifQ.uhEQ3sB6SJLR_Duuu4w-WilRsvllI611flQ_uQoI5ufm_GWtLLJfHZ8rE9-RS2YtkprtYovvEf1U1E6ybcL1Jg"
|
53
|
+
)
|
54
|
+
# "https://kimi.moonshot.cn/api/chat/ctb5lpaflk1f1dda5mv0/completion/stream"
|
55
|
+
resp = client.post(
|
56
|
+
"/api/chat/ctb5lpaflk1f1dda5mv0/completion/stream",
|
57
|
+
cast_to=object,
|
58
|
+
body=payload,
|
59
|
+
stream=True
|
60
|
+
)
|
61
|
+
|
@@ -0,0 +1,176 @@
|
|
1
|
+
#!/usr/bin/env python
|
2
|
+
# -*- coding: utf-8 -*-
|
3
|
+
# @Project : AI. @by PyCharm
|
4
|
+
# @File : yuanbao
|
5
|
+
# @Time : 2024/6/11 18:56
|
6
|
+
# @Author : betterme
|
7
|
+
# @WeChat : meutils
|
8
|
+
# @Software : PyCharm
|
9
|
+
# @Description :
|
10
|
+
import asyncio
|
11
|
+
|
12
|
+
from aiostream import stream
|
13
|
+
|
14
|
+
from meutils.pipe import *
|
15
|
+
from meutils.io.image import image2nowatermark_image
|
16
|
+
|
17
|
+
from meutils.llm.utils import oneturn2multiturn
|
18
|
+
from meutils.schemas.openai_types import ChatCompletionRequest
|
19
|
+
from meutils.schemas.image_types import HunyuanImageRequest
|
20
|
+
|
21
|
+
from meutils.schemas.yuanbao_types import FEISHU_URL, SSEData, YUANBAO_BASE_URL, API_CHAT, API_GENERATE_ID, \
|
22
|
+
API_DELETE_CONV, \
|
23
|
+
GET_AGENT_CHAT
|
24
|
+
from meutils.config_utils.lark_utils import get_next_token_for_polling, aget_spreadsheet_values
|
25
|
+
|
26
|
+
|
27
|
+
# import rev_HunYuan
|
28
|
+
|
29
|
+
|
30
|
+
class Completions(object):
|
31
|
+
|
32
|
+
@classmethod
|
33
|
+
async def generate(cls, request: HunyuanImageRequest):
|
34
|
+
response = cls().create(image_request=request)
|
35
|
+
urls = await stream.list(response)
|
36
|
+
urls = await asyncio.gather(*map(image2nowatermark_image, urls))
|
37
|
+
|
38
|
+
return {
|
39
|
+
"data": [{"url": url} for url in urls]
|
40
|
+
}
|
41
|
+
|
42
|
+
async def create(
|
43
|
+
self,
|
44
|
+
request: Optional[ChatCompletionRequest] = None,
|
45
|
+
image_request: Optional[HunyuanImageRequest] = None,
|
46
|
+
token: Optional[str] = None
|
47
|
+
):
|
48
|
+
token = token or await get_next_token_for_polling(FEISHU_URL, check_token=check_token)
|
49
|
+
|
50
|
+
logger.debug(token)
|
51
|
+
|
52
|
+
prompt = request and oneturn2multiturn(request.messages) or image_request.prompt
|
53
|
+
|
54
|
+
payload = {
|
55
|
+
"model": "gpt_175B_0404",
|
56
|
+
"version": "v2",
|
57
|
+
"supportHint": 2,
|
58
|
+
|
59
|
+
"prompt": prompt,
|
60
|
+
# "displayPrompt": "画条可爱的狗狗",
|
61
|
+
# "displayPromptType": 1,
|
62
|
+
"multimedia": [],
|
63
|
+
# "agentId": "gtcnTp5C1G",
|
64
|
+
|
65
|
+
"plugin": "Adaptive",
|
66
|
+
|
67
|
+
"options": {
|
68
|
+
"imageIntention": {
|
69
|
+
"needIntentionModel": True,
|
70
|
+
"backendUpdateFlag": 2,
|
71
|
+
"intentionStatus": True,
|
72
|
+
"userIntention": {
|
73
|
+
"resolution": "1280x1280",
|
74
|
+
}
|
75
|
+
}
|
76
|
+
},
|
77
|
+
|
78
|
+
}
|
79
|
+
if image_request:
|
80
|
+
payload["displayImageIntentionLabels"] = [
|
81
|
+
{"type": "resolution", "disPlayValue": "超清", "startIndex": 0, "endIndex": 1}
|
82
|
+
]
|
83
|
+
payload["options"]["imageIntention"]["userIntention"].update(
|
84
|
+
{
|
85
|
+
"style": image_request.style,
|
86
|
+
|
87
|
+
"scale": image_request.size,
|
88
|
+
|
89
|
+
# todo: 默认四张 不生效
|
90
|
+
# "N": image_request.n,
|
91
|
+
# "num": image_request.n,
|
92
|
+
# "Count": image_request.n,
|
93
|
+
|
94
|
+
}
|
95
|
+
)
|
96
|
+
|
97
|
+
headers = {
|
98
|
+
'cookie': token
|
99
|
+
}
|
100
|
+
async with httpx.AsyncClient(base_url=YUANBAO_BASE_URL, headers=headers, timeout=300) as client:
|
101
|
+
# chatid = (await client.post(API_GENERATE_ID)).text
|
102
|
+
chatid = uuid.uuid4()
|
103
|
+
|
104
|
+
async with client.stream(method="POST", url=f"{API_CHAT}/{chatid}", json=payload) as response:
|
105
|
+
logger.debug(response.status_code)
|
106
|
+
response.raise_for_status()
|
107
|
+
|
108
|
+
async for chunk in response.aiter_lines():
|
109
|
+
sse = SSEData(chunk=chunk)
|
110
|
+
if image_request and sse.image:
|
111
|
+
logger.debug(sse.image)
|
112
|
+
yield sse.image
|
113
|
+
|
114
|
+
if request:
|
115
|
+
yield sse.content
|
116
|
+
|
117
|
+
def generate_id(self, random: bool = True):
|
118
|
+
if random:
|
119
|
+
return f'{uuid.uuid4()}'
|
120
|
+
return httpx.post(API_GENERATE_ID).text
|
121
|
+
|
122
|
+
def delete_conv(self, chatid):
|
123
|
+
response = httpx.post(f"{API_DELETE_CONV}/{chatid}")
|
124
|
+
return response.status_code == 200
|
125
|
+
|
126
|
+
|
127
|
+
async def check_token(token):
|
128
|
+
headers = {
|
129
|
+
"cookie": token
|
130
|
+
}
|
131
|
+
try:
|
132
|
+
async with httpx.AsyncClient(base_url=YUANBAO_BASE_URL, headers=headers, timeout=10) as client:
|
133
|
+
response = await client.get("/api/info/general")
|
134
|
+
response.raise_for_status()
|
135
|
+
logger.debug(response.status_code)
|
136
|
+
return True
|
137
|
+
except Exception as e:
|
138
|
+
logger.error(e)
|
139
|
+
return False
|
140
|
+
|
141
|
+
|
142
|
+
if __name__ == '__main__':
|
143
|
+
# chatid = generate_id()
|
144
|
+
# print(chatid)
|
145
|
+
# print(delete_conv(chatid))
|
146
|
+
# payload = {
|
147
|
+
# # "model": "gpt_175B_0404",
|
148
|
+
# # "prompt": "1+1",
|
149
|
+
# "prompt": "错了",
|
150
|
+
#
|
151
|
+
# # "plugin": "Adaptive",
|
152
|
+
# # "displayPrompt": "1+1",
|
153
|
+
# # "displayPromptType": 1,
|
154
|
+
# # "options": {},
|
155
|
+
# # "multimedia": [],
|
156
|
+
# # "agentId": "naQivTmsDa",
|
157
|
+
# # "version": "v2"
|
158
|
+
# }
|
159
|
+
# chat(payload)
|
160
|
+
|
161
|
+
# async2sync_generator(Completions(api_key).achat('画条狗')) | xprint
|
162
|
+
request = HunyuanImageRequest(prompt='画条狗', size='16:9')
|
163
|
+
|
164
|
+
# arun(Completions().create(
|
165
|
+
# ChatCompletionRequest(messages=[{'role': 'user', 'content': '南京天气如何'}]),
|
166
|
+
# # image_request=request,
|
167
|
+
# # token=token
|
168
|
+
# ))
|
169
|
+
arun(Completions.generate(request))
|
170
|
+
|
171
|
+
# df = arun(aget_spreadsheet_values(feishu_url=FEISHU__URL, to_dataframe=True))
|
172
|
+
#
|
173
|
+
# for i in df[0]:
|
174
|
+
# if not arun(check_token(i)):
|
175
|
+
# print(i)
|
176
|
+
#
|
meutils/llm/demo.py
ADDED
@@ -0,0 +1,114 @@
|
|
1
|
+
#!/usr/bin/env python
|
2
|
+
# -*- coding: utf-8 -*-
|
3
|
+
# @Project : AI. @by PyCharm
|
4
|
+
# @File : demo
|
5
|
+
# @Time : 2024/4/24 14:18
|
6
|
+
# @Author : betterme
|
7
|
+
# @WeChat : meutils
|
8
|
+
# @Software : PyCharm
|
9
|
+
# @Description : https://python.langchain.com/docs/modules/model_io/output_parsers/types/pydantic/
|
10
|
+
|
11
|
+
from meutils.pipe import *
|
12
|
+
from langchain.output_parsers import PydanticOutputParser
|
13
|
+
from pydantic import BaseModel, Field
|
14
|
+
from typing import List
|
15
|
+
|
16
|
+
|
17
|
+
# 使用Pydantic创建一个数据格式,表示材料
|
18
|
+
class Flower(BaseModel):
|
19
|
+
name: str = Field(description="name of a Material")
|
20
|
+
colors: List[str] = Field(description="the colors of this Material")
|
21
|
+
|
22
|
+
|
23
|
+
# 定义一个用于获取某种材料的颜色列表的查询
|
24
|
+
flower_query = "Generate the charaters for a random Material."
|
25
|
+
|
26
|
+
# 定义一个格式不正确的输出
|
27
|
+
misformatted = "{'name': '汽车漆', 'colors': ['粉红色','白色','红色','紫色','黄色']}"
|
28
|
+
|
29
|
+
# 创建一个用于解析输出的Pydantic解析器,此处希望解析为Flower格式
|
30
|
+
parser = PydanticOutputParser(pydantic_object=Flower)
|
31
|
+
|
32
|
+
# 使用Pydantic解析器解析不正确的输出
|
33
|
+
# parser.parse(misformatted) # 这行代码会出错
|
34
|
+
# OutputParserException: Failed to parse Flower from completion {'name': '汽车漆', 'colors': ['粉红色','白色','红色','紫色','黄色']}. Got: Expecting property name enclosed in double quotes: line 1 column 2 (char 1)
|
35
|
+
|
36
|
+
# 从langchain库导入所需的模块
|
37
|
+
from langchain.chat_models import ChatOpenAI
|
38
|
+
from langchain.output_parsers import OutputFixingParser
|
39
|
+
|
40
|
+
# 使用OutputFixingParser创建一个新的解析器,该解析器能够纠正格式不正确的输出
|
41
|
+
new_parser = OutputFixingParser.from_llm(parser=parser, llm=ChatOpenAI())
|
42
|
+
|
43
|
+
# print(new_parser)
|
44
|
+
# 使用新的解析器解析不正确的输出
|
45
|
+
result = new_parser.parse(misformatted) # 错误被自动修正
|
46
|
+
print(result) # 打印解析后的输出结果
|
47
|
+
# name='汽车漆' colors=['粉红色', '白色', '红色', '紫色', '黄色']
|
48
|
+
|
49
|
+
if __name__ == '__main__':
|
50
|
+
text = """
|
51
|
+
请根据我的自定义歌词和音乐风格来创作歌曲:中文标题:酒入愁肠
|
52
|
+
英文标题:Wine and Sorrows
|
53
|
+
|
54
|
+
歌词结构:
|
55
|
+
[Intro]
|
56
|
+
(独奏)
|
57
|
+
|
58
|
+
[Verse 1]
|
59
|
+
月光洒在破旧的篱笆,
|
60
|
+
李白提壶独自彷徨。
|
61
|
+
酒香飘逸入云端,
|
62
|
+
笔下是豪迈的想象。
|
63
|
+
|
64
|
+
[Chorus]
|
65
|
+
一壶浊酒喜相逢,
|
66
|
+
千载孤愁只杜甫懂。
|
67
|
+
古道西风瘦马间,
|
68
|
+
人间事,两行清泪长。
|
69
|
+
|
70
|
+
[Verse 2]
|
71
|
+
江水流淌带走忧伤,
|
72
|
+
杜甫凝眉思国家。
|
73
|
+
文章锋利剑如霜,
|
74
|
+
心中苦,却言者无罪当。
|
75
|
+
|
76
|
+
[Chorus]
|
77
|
+
一壶浊酒喜相逢,
|
78
|
+
千载孤愁只杜甫懂。
|
79
|
+
古道西风瘦马间,
|
80
|
+
人间事,两行清泪长。
|
81
|
+
|
82
|
+
[Bridge]
|
83
|
+
一千年来谁能懂,
|
84
|
+
酒与愁是最深的重。
|
85
|
+
历史长河波澜壮,
|
86
|
+
诗人眼中看尽繁华空。
|
87
|
+
|
88
|
+
[Chorus]
|
89
|
+
一壶浊酒喜相逢,
|
90
|
+
千载孤愁只杜甫懂。
|
91
|
+
古道西风瘦马间,
|
92
|
+
人间事,两行清泪长。
|
93
|
+
|
94
|
+
[Outro]
|
95
|
+
(独奏)
|
96
|
+
|
97
|
+
音乐风格:
|
98
|
+
传统民谣
|
99
|
+
主要乐器:古筝、琵琶、笛子
|
100
|
+
节奏:缓慢而深情
|
101
|
+
氛围:沉郁、古典、充满历史感
|
102
|
+
"""
|
103
|
+
|
104
|
+
|
105
|
+
class Song(BaseModel):
|
106
|
+
title: str = Field(description="song title")
|
107
|
+
lyrics: str = Field(description="Enter your own lyrics", examples=['[Intro]...[Verse]...[Chorus]...'])
|
108
|
+
music_style: str = Field(description="Style of Music, Maximum 10 words", examples=['syncopated country ...'])
|
109
|
+
|
110
|
+
|
111
|
+
song_parser = OutputFixingParser.from_llm(parser=PydanticOutputParser(pydantic_object=Song), llm=ChatOpenAI(model="kimi"))
|
112
|
+
|
113
|
+
result = song_parser.parse(text) # 错误被自动修正
|
114
|
+
print(result) # 打印解析后的输出结果
|
meutils/llm/mappers.py
ADDED
@@ -0,0 +1,15 @@
|
|
1
|
+
#!/usr/bin/env python
|
2
|
+
# -*- coding: utf-8 -*-
|
3
|
+
# @Project : AI. @by PyCharm
|
4
|
+
# @File : mappers
|
5
|
+
# @Time : 2024/12/16 14:49
|
6
|
+
# @Author : betterme
|
7
|
+
# @WeChat : meutils
|
8
|
+
# @Software : PyCharm
|
9
|
+
# @Description :
|
10
|
+
|
11
|
+
from meutils.pipe import *
|
12
|
+
|
13
|
+
|
14
|
+
def llm_mapper(model):
|
15
|
+
return
|