XMWAI 0.1.6__tar.gz → 0.1.8__tar.gz

xmwai-0.1.8/MANIFEST.in

@@ -0,0 +1,2 @@
+recursive-include XMWAI/gif *
+include XMAI/idiom.json

{xmwai-0.1.6/XMWAI.egg-info → xmwai-0.1.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: XMWAI
-Version: 0.1.6
+Version: 0.1.8
 Summary: Small code King AI related library
 Home-page: https://github.com/Tonykai88/XMWAI.git
 Author: pydevelopment

{xmwai-0.1.6 → xmwai-0.1.8}/XMWAI/__init__.py

@@ -1,5 +1,6 @@
 from .core import story, photo, reply, poem, get_access_token  # 从子模块导入函数到顶层
 from .magic_core import birthday  # 从子模块导入函数到顶层
 from .bomb_core import bomb  # 从子模块导入函数到顶层
+from .idiom_core import idiom,searchIdiom  # 从子模块导入函数到顶层
 
-__all__ = ["story", "photo", "reply", "poem", "get_access_token", 'birthday', 'bomb']  # 可选：明确导出的内容
+__all__ = ["story", "photo", "reply", "poem", "get_access_token", 'birthday', 'bomb', "idiom", "searchIdiom"]  # 可选：明确导出的内容

{xmwai-0.1.6 → xmwai-0.1.8}/XMWAI/core.py

@@ -184,24 +184,34 @@ def poem(title, key=""):
     elif key != "CaJQ":
         print("秘钥错误！请重新输入！")
         return "秘钥错误！请重新输入！"
-    key = "ZZC"
-    access_token = get_access_token(key)
-    if access_token:
-        url = f"https://aip.baidubce.com/rpc/2.0/nlp/v1/poem?access_token={access_token}"
-        payload = {
-            "text": title,  # 直接使用普通字符串
-        }
-        headers = {
-            'Content-Type': 'application/json',
-            'Accept': 'application/json'
-        }
+    messagesList = [
+            {
+                "role": "system",
+                "content": "唐代诗人"
+            },
+            {
+                "role": "user",
+                "content": f"请以《{title}》为题，创作一首七言绝句，每句7个字，一共4句，符合古诗韵律规范，内容积极乐观向上，适合中小学生阅读，不要解析，不要题目，不要标点符号，所有文字放在一行"
+            }
+            ]
         
-        # 直接发送JSON对象，不转换为字符串
-        response = requests.post(url, headers=headers, json=payload)
-        if response.status_code == 200:
-            return response.json()['poem'][0]['content']  # 使用response.json()直接解析JSON响应
-        else:
-            print("Failed with status code:", response.status_code)
-            print("Response:", response.text)
-    else:
-        print("Failed to get access token.")
+    url = "https://qianfan.baidubce.com/v2/chat/completions"
+    
+    payload = json.dumps({
+        "model": "ernie-4.5-turbo-32k",
+        "messages": messagesList
+    }, ensure_ascii=False)
+    headers = {
+        'Content-Type': 'application/json',
+        'appid': '',
+        'Authorization': 'Bearer bce-v3/ALTAK-cGbxpVA5AbSz6h8nbLaFh/b539762075d55c76d93dc78bcf0a91beeaf0490a'
+    }
+    
+    response = requests.request("POST", url, headers=headers, data=payload.encode("utf-8"))
+    
+    response_data = json.loads(response.text)
+    content = response_data["choices"][0]["message"]["content"]
+    content.replace(" ", "")
+    content.replace("\n", "")
+    content.replace("  ", "")
+    return content

xmwai-0.1.8/XMWAI/idiom_core.py

@@ -0,0 +1,54 @@
+import random
+import json
+from importlib.resources import files
+from pathlib import Path
+
+
+def get_json_path():
+    """获取 idiom.json 的正确路径（兼容 PyPI 安装后的包内资源）"""
+    return Path(files("XMWAI").joinpath("idiom.json"))
+
+
+def idiom(word, mode=0):
+    """
+    查询成语信息
+    :param word: 要查询的成语
+    :param mode: 查询模式
+                 0 - 判断是否是成语，返回 True/False
+                 1 - 返回拼音
+                 2 - 返回解释
+                 3 - 返回出处
+    :return: 查询结果或 None
+    """
+    with open(get_json_path(), "r", encoding="utf-8") as f:
+        data = json.load(f)
+        for i in data:
+            if word == i["word"]:
+                if mode == 0:
+                    return True
+                elif mode == 1:
+                    return i["pinyin"]
+                elif mode == 2:
+                    return i["explanation"]
+                elif mode == 3:
+                    return i["derivation"]
+        return False if mode == 0 else None
+
+
+def searchIdiom(text, num=1):
+    """
+    模糊查询成语
+    :param text: 要查询的字
+    :param num: 第几个字（1~4）
+    :return: 匹配的成语 或 False
+    """
+    wordList = []
+    with open(get_json_path(), "r", encoding="utf-8") as f:
+        data = json.load(f)
+        for i in data:
+            try:
+                if text == i["word"][num - 1]:
+                    wordList.append(i["word"])
+            except:
+                pass
+    return random.choice(wordList) if wordList else False

{xmwai-0.1.6 → xmwai-0.1.8/XMWAI.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: XMWAI
-Version: 0.1.6
+Version: 0.1.8
 Summary: Small code King AI related library
 Home-page: https://github.com/Tonykai88/XMWAI.git
 Author: pydevelopment

{xmwai-0.1.6 → xmwai-0.1.8}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="XMWAI",  # 包名（pip install XMWAI）
-    version="0.1.6",  # 初始版本号
+    version="0.1.8",  # 初始版本号
     author="pydevelopment",  # 作者
     author_email="hekai@xiaoma.cn",  # 邮箱
     description="Small code King AI related library",  # 简短描述

xmwai-0.1.6/MANIFEST.in

@@ -1 +0,0 @@
-recursive-include XMWAI/gif *

xmwai-0.1.6/XMWAI/idiom_core.py

@@ -1,64 +0,0 @@
-import random
-
-import json
-
-
-def idiom(word, mode=0):
-    """
-    查询成语信息
-    :param word: 要查询的成语
-    :param mode: 查询模式
-                 0 - 判断是否是成语，返回 True/False
-                 1 - 返回拼音
-                 2 - 返回解释
-                 3 - 返回出处
-    :return: 查询结果或 None
-    """
-    if mode == 0:
-        with open("idiom.json", "r", encoding="utf-8") as f:
-            data = json.load(f)
-            for i in data:
-                if word == i["word"]:
-                    return True
-        return False
-    elif mode == 1:
-        with open("idiom.json", "r", encoding="utf-8") as f:
-            data = json.load(f)
-            for i in data:
-                if word == i["word"]:
-                    return i["pinyin"]
-        return None
-
-    elif mode == 2:
-        with open("idiom.json", "r", encoding="utf-8") as f:
-            data = json.load(f)
-            for i in data:
-                if word == i["word"]:
-                    return i["explanation"]
-        return None
-    elif mode == 3:
-        with open("idiom.json", "r", encoding="utf-8") as f:
-            data = json.load(f)
-            for i in data:
-                if word == i["word"]:
-                    return i["derivation"]
-        return None
-
-
-def searchIdiom(text, num=1):
-    """
-    模糊查询成语
-    :param text: 要查询的字
-    :param mode: 第几个字
-    :return: 查询结果 或 None
-    """
-    wordList = []
-    with open("idiom.json", "r", encoding="utf-8") as f:
-        data = json.load(f)
-        for i in data:
-            if text == i["word"][num-1]:
-                wordList.append(i["word"])
-    if wordList:
-        return random.choice(wordList)
-    else:
-        return False