file-netai 0.1.0__tar.gz

file_netai-0.1.0/PKG-INFO

@@ -0,0 +1,18 @@
+Metadata-Version: 2.4
+Name: file_netai
+Version: 0.1.0
+Summary: AI 生成文件工具
+Author: shiroko
+Author-email: 3207774253@qq.com
+Requires-Python: >=3.6
+Description-Content-Type: text/markdown
+Requires-Dist: shiroko_netai
+Dynamic: author
+Dynamic: author-email
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
+
+AI 生成文件工具，根据用户需求自动决定文件名、路径和内容。

file_netai-0.1.0/README.md

@@ -0,0 +1,37 @@
+# file_netai
+
+##  免责声明
+
+1. **本工具生成的所有代码均由 AI 自动生成，未经人工逐行审查。**
+2. **作者不对生成代码的正确性、安全性、可靠性作任何明示或暗示的保证。**
+3. **使用者必须自行审查、测试、修改生成代码，确认无误后方可使用。**
+4. **因使用生成代码导致的任何直接或间接损失（包括但不限于数据丢失、系统损坏、法律纠纷），作者不承担任何责任。**
+5. **禁止使用本工具生成恶意代码、病毒、木马等危害性内容。**
+6. **继续使用本工具即代表您已阅读、理解并同意上述条款。**
+
+**⚠️ 安装、下载、使用本工具即视为已阅读并同意以上免责声明。**
+**如不同意以上条款，请立即卸载并停止使用本工具。**
+
+
+AI 文件生成工具 —— 根据自然语言需求直接生成代码文件，带短时记忆
+
+## 安装
+
+```bash
+pip install file_netai
+```
+
+依赖
+
+· Python >= 3.6
+· shiroko_netai
+
+使用
+
+```bash
+# 生成文件
+file_netai "生成一个 Flask 登录页面"
+
+# 查询上一次生成的内容
+file_netai "刚才生成的代码是什么"
+```

file_netai-0.1.0/file_netai/__init__.py

@@ -0,0 +1,5 @@
+from .cli import main
+from .core import generate_from_prompt
+
+__version__ = "0.2.0"
+__all__ = ["main", "generate_from_prompt"]

file_netai-0.1.0/file_netai/cli.py

@@ -0,0 +1,31 @@
+#!/usr/bin/env python3
+import sys
+import argparse
+import json
+from .core import generate_from_prompt
+
+def main():
+    parser = argparse.ArgumentParser(description="AI 文件生成工具")
+    parser.add_argument("prompts", nargs="*", help="需求描述（可多个）")
+    parser.add_argument("--batch", help="批量生成配置文件 (JSON 数组)")
+    parser.add_argument("--url", default="http://127.0.0.1:11434/api/chat", help="AI API 地址")
+    parser.add_argument("--model", default="deepseek-r1:1.5b", help="模型名称")
+    parser.add_argument("--no-memory", action="store_true", help="不保存记忆")
+    
+    args = parser.parse_args()
+    
+    if args.batch:
+        with open(args.batch, "r", encoding="utf-8") as f:
+            prompts = json.load(f)
+        for prompt in prompts:
+            print(f"\n--- 处理: {prompt} ---")
+            generate_from_prompt(prompt, url=args.url, model=args.model, memory=not args.no_memory)
+    elif args.prompts:
+        for prompt in args.prompts:
+            print(f"\n--- 处理: {prompt} ---")
+            generate_from_prompt(prompt, url=args.url, model=args.model, memory=not args.no_memory)
+    else:
+        parser.print_help()
+
+if __name__ == "__main__":
+    main()

file_netai-0.1.0/file_netai/core.py

@@ -0,0 +1,77 @@
+import json
+import os
+from shiroko_netai import TCC
+
+_last_generated = None
+
+def parse_null(val):
+    return None if val == "<[null]>" else val
+
+def generate_from_prompt(user_prompt, url=None, model=None, memory=True):
+    global _last_generated
+    
+    if url is None:
+        url = "http://127.0.0.1:11434/api/chat"
+    if model is None:
+        model = "deepseek-r1:1.5b"
+    
+    ai = TCC(url=url, model=model)
+
+    if memory and "刚才" in user_prompt and ("代码" in user_prompt or "生成" in user_prompt):
+        if _last_generated:
+            print("你刚才生成的代码是：")
+            print(_last_generated.get("file_content", "无内容"))
+            return _last_generated.get("file_content")
+        else:
+            print("还没有生成过任何代码。")
+            return None
+    
+    system_prompt = (
+        "你是一个文件生成助手。根据用户的需求，返回一个 JSON 对象，不要输出任何解释。\n"
+        "JSON 格式如下：\n"
+        "{\n"
+        "    \"file_name\": \"文件名\",\n"
+        "    \"file_path\": \"目录路径或<[null]>\",\n"
+        "    \"file_content\": \"文件内容\",\n"
+        "    \"FOLDER_path\": \"附加目录或<[null]>\"\n"
+        "}\n"
+        "如果不需要目录，请使用 <[null]> 表示。"
+    )
+    full_prompt = f"{system_prompt}\n用户需求：{user_prompt}"
+    response = ai.send(full_prompt)
+    
+    try:
+        data = json.loads(response)
+        file_name = data.get("file_name")
+        file_path = parse_null(data.get("file_path"))
+        file_content = data.get("file_content", "")
+        folder_path = parse_null(data.get("FOLDER_path"))
+        
+        if not file_name:
+            print("AI 未返回文件名")
+            return None
+        
+        if memory:
+            _last_generated = data
+        
+        if file_path:
+            os.makedirs(file_path, exist_ok=True)
+            save_path = os.path.join(file_path, file_name)
+        else:
+            save_path = file_name
+        
+        if folder_path:
+            os.makedirs(folder_path, exist_ok=True)
+            print(f"已创建附加目录: {folder_path}")
+        
+        with open(save_path, "w", encoding="utf-8") as f:
+            f.write(file_content)
+        print(f"已生成文件: {save_path}")
+        return save_path
+    except json.JSONDecodeError as e:
+        print(f"AI 返回的不是有效 JSON: {e}")
+        print(f"原始响应: {response[:200]}...")
+        return None
+    except Exception as e:
+        print(f"生成失败: {e}")
+        return None

file_netai-0.1.0/file_netai.egg-info/PKG-INFO

@@ -0,0 +1,18 @@
+Metadata-Version: 2.4
+Name: file_netai
+Version: 0.1.0
+Summary: AI 生成文件工具
+Author: shiroko
+Author-email: 3207774253@qq.com
+Requires-Python: >=3.6
+Description-Content-Type: text/markdown
+Requires-Dist: shiroko_netai
+Dynamic: author
+Dynamic: author-email
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
+
+AI 生成文件工具，根据用户需求自动决定文件名、路径和内容。

file_netai-0.1.0/file_netai.egg-info/SOURCES.txt

@@ -0,0 +1,11 @@
+README.md
+setup.py
+file_netai/__init__.py
+file_netai/cli.py
+file_netai/core.py
+file_netai.egg-info/PKG-INFO
+file_netai.egg-info/SOURCES.txt
+file_netai.egg-info/dependency_links.txt
+file_netai.egg-info/entry_points.txt
+file_netai.egg-info/requires.txt
+file_netai.egg-info/top_level.txt

file_netai-0.1.0/file_netai.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

file_netai-0.1.0/file_netai.egg-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+file_netai = file_netai.cli:main

file_netai-0.1.0/file_netai.egg-info/requires.txt

@@ -0,0 +1 @@
+shiroko_netai

file_netai-0.1.0/file_netai.egg-info/top_level.txt

@@ -0,0 +1 @@
+file_netai

file_netai-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

file_netai-0.1.0/setup.py

@@ -0,0 +1,19 @@
+from setuptools import setup, find_packages
+
+setup(
+    name="file_netai",
+    version="0.1.0",
+    author="shiroko",
+    author_email="3207774253@qq.com",
+    description="AI 生成文件工具",
+    long_description="AI 生成文件工具，根据用户需求自动决定文件名、路径和内容。",
+    long_description_content_type="text/markdown",
+    packages=find_packages(),
+    install_requires=["shiroko_netai"],
+    entry_points={
+        "console_scripts": [
+            "file_netai = file_netai.cli:main",
+        ]
+    },
+    python_requires=">=3.6",
+)