XMWAI 0.1.6__py3-none-any.whl → 0.1.8__py3-none-any.whl

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/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/idiom_core.py

@@ -1,6 +1,12 @@
 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):
@@ -14,51 +20,35 @@ def idiom(word, mode=0):
                  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"]:
+    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
-        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"]:
+                elif mode == 1:
                     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"]:
+                elif mode == 2:
                     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"]:
+                elif mode == 3:
                     return i["derivation"]
-        return None
+        return False if mode == 0 else None
 
 
 def searchIdiom(text, num=1):
     """
     模糊查询成语
     :param text: 要查询的字
-    :param mode: 第几个字
-    :return: 查询结果 或 None
+    :param num: 第几个字（1~4）
+    :return: 匹配的成语 或 False
     """
     wordList = []
-    with open("idiom.json", "r", encoding="utf-8") as f:
+    with open(get_json_path(), "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
+            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.dist-info → xmwai-0.1.8.dist-info}/METADATA

@@ -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.dist-info → xmwai-0.1.8.dist-info}/RECORD

@@ -1,7 +1,7 @@
-XMWAI/__init__.py,sha256=22xWIf68Fub64LGW_EB7WBvhuc5DWyzusEm3I2EjBFA,360
+XMWAI/__init__.py,sha256=GuNnfM345TTSZW45YiwEt1dM6Ay0biP1bQBrGpqqmEg,464
 XMWAI/bomb_core.py,sha256=dRORqAyhItkFuMUHAN0vwQcvrvb3C5X18hmaUMJ0hhE,1708
-XMWAI/core.py,sha256=C4pntTFPOpMuFhJCEVoAlrJEizNSHOWZcB1dwCdkl6w,7396
-XMWAI/idiom_core.py,sha256=PPBBXYj_jJV1SnU5NROHiFkukmZDpZJgjZyuTiZondA,1774
+XMWAI/core.py,sha256=KrtdKXvSGKAHlOYF1-v2iSxpfjpHoGeGVOBIxH1Ez4E,7759
+XMWAI/idiom_core.py,sha256=0Uu9GxVwEvFAhjj0M2Orw-E9FjKT05jKQDbUvg0wAXE,1577
 XMWAI/magic_core.py,sha256=XU7K4Ta7EAvwBLk20_Hlu18E6Kz8B0HxUBq0YsawikE,3441
 XMWAI/gif/0.gif,sha256=LGpAReVTyZEb1J-bWYTpxxHbIxmLJ-3wA0lw4cBqdsY,303
 XMWAI/gif/1.gif,sha256=nsysRjMkNBQJy91SOY284gnnD_DlhLrgyIOgcJ_ZOHI,3554
@@ -90,8 +90,8 @@ XMWAI/gif/84.gif,sha256=6Ip_uQmvrr2fagLXu1YqWyI_DL2PVVtKCPtmNtzt3P4,38767
 XMWAI/gif/85.gif,sha256=6Ip_uQmvrr2fagLXu1YqWyI_DL2PVVtKCPtmNtzt3P4,38767
 XMWAI/gif/9.gif,sha256=cPouth-xwc3QPcg2m6HMP2OD1ZCeRBD_-RPImvvKAA0,9485
 XMWAI/gif/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-xmwai-0.1.6.dist-info/licenses/LICENSE.txt,sha256=bcaIQMrIhdQ3O-PoZlexjmW6h-wLGvHxh5Oksl4ohtc,1066
-xmwai-0.1.6.dist-info/METADATA,sha256=El-wSx7NCt6i27P4brabPLUWkOHeyMk4-4nYk__8y9k,1033
-xmwai-0.1.6.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
-xmwai-0.1.6.dist-info/top_level.txt,sha256=yvGcDI-sggK5jqd9wz0saipZvk3oIE3hNGHlqUjxf0c,6
-xmwai-0.1.6.dist-info/RECORD,,
+xmwai-0.1.8.dist-info/licenses/LICENSE.txt,sha256=bcaIQMrIhdQ3O-PoZlexjmW6h-wLGvHxh5Oksl4ohtc,1066
+xmwai-0.1.8.dist-info/METADATA,sha256=CJgI9oxFUvO-a7Ax4XNIQ5AOEz-t_O-p2doxPSDIIIk,1033
+xmwai-0.1.8.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
+xmwai-0.1.8.dist-info/top_level.txt,sha256=yvGcDI-sggK5jqd9wz0saipZvk3oIE3hNGHlqUjxf0c,6
+xmwai-0.1.8.dist-info/RECORD,,