doorcore 0.1.0__tar.gz

doorcore-0.1.0/PKG-INFO

@@ -0,0 +1,66 @@
+Metadata-Version: 2.5
+Name: doorcore
+Version: 0.1.0
+Summary: door-Core v2: a lightweight offline Chinese AI engine with Markov generation and cosine retrieval
+Project-URL: Homepage, https://github.com/door-studio/doorcore
+Author-email: door 工作室 <doornb666@outlook.com>
+License: MIT
+Keywords: ai,chatbot,chinese,markov,offline
+Classifier: Development Status :: 4 - Beta
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+
+# doorcore
+
+door-Core v2 本地 AI 引擎，纯 Python 标准库、零第三方依赖。
+
+特性：
+- 中文关键词知识库检索（余弦相似）
+- 字符级马尔可夫生成
+- 中文/口语算式安全计算：支持「二乘四」「2的10次方」「根号2」「5的平方」等
+- 报时、讲笑话、简单角色扮演
+- OpenAI 兼容 API（`/v1/chat/completions`）
+
+## 安装
+
+```bash
+pip install doorcore
+```
+
+## 编程使用
+
+```python
+from doorcore import DoorCore
+
+ai = DoorCore()
+ai.load_or_train()
+
+print(ai.ask("1+1等于几"))   # 计算
+print(ai.ask("雷军是谁"))    # 知识库
+print(ai.ask("讲个笑话"))    # 笑话
+```
+
+## 命令行
+
+```bash
+doorcore          # 启动交互式聊天
+doorcore-api      # 启动 OpenAI 兼容 API 服务
+```
+
+API 默认监听 `http://localhost:8765/v1`，模型名 `door-core`。
+可在任何支持自定义 OpenAI 兼容模型的客户端里填入：
+- Base URL: `http://localhost:8765/v1`
+- Model: `door-core`
+- API Key: 任意填（不校验）
+
+## 许可证
+
+MIT

doorcore-0.1.0/README.md

@@ -0,0 +1,46 @@
+# doorcore
+
+door-Core v2 本地 AI 引擎，纯 Python 标准库、零第三方依赖。
+
+特性：
+- 中文关键词知识库检索（余弦相似）
+- 字符级马尔可夫生成
+- 中文/口语算式安全计算：支持「二乘四」「2的10次方」「根号2」「5的平方」等
+- 报时、讲笑话、简单角色扮演
+- OpenAI 兼容 API（`/v1/chat/completions`）
+
+## 安装
+
+```bash
+pip install doorcore
+```
+
+## 编程使用
+
+```python
+from doorcore import DoorCore
+
+ai = DoorCore()
+ai.load_or_train()
+
+print(ai.ask("1+1等于几"))   # 计算
+print(ai.ask("雷军是谁"))    # 知识库
+print(ai.ask("讲个笑话"))    # 笑话
+```
+
+## 命令行
+
+```bash
+doorcore          # 启动交互式聊天
+doorcore-api      # 启动 OpenAI 兼容 API 服务
+```
+
+API 默认监听 `http://localhost:8765/v1`，模型名 `door-core`。
+可在任何支持自定义 OpenAI 兼容模型的客户端里填入：
+- Base URL: `http://localhost:8765/v1`
+- Model: `door-core`
+- API Key: 任意填（不校验）
+
+## 许可证
+
+MIT

doorcore-0.1.0/pyproject.toml

@@ -0,0 +1,42 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "doorcore"
+version = "0.1.0"
+description = "door-Core v2: a lightweight offline Chinese AI engine with Markov generation and cosine retrieval"
+readme = "README.md"
+requires-python = ">=3.9"
+license = {text = "MIT"}
+authors = [
+    {name = "door 工作室", email = "doornb666@outlook.com"},
+]
+keywords = ["ai", "chatbot", "offline", "chinese", "markov"]
+classifiers = [
+    "Development Status :: 4 - Beta",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.9",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+    "Programming Language :: Python :: 3.13",
+    "License :: OSI Approved :: MIT License",
+    "Operating System :: OS Independent",
+]
+
+[project.urls]
+Homepage = "https://github.com/door-studio/doorcore"
+
+[project.scripts]
+doorcore = "doorcore.cli:main"
+doorcore-api = "doorcore.api:main"
+
+[tool.hatch.build]
+include = [
+    "src/doorcore/**/*.py",
+    "src/doorcore/data/**/*",
+]
+
+[tool.hatch.build.targets.wheel]
+packages = ["src/doorcore"]

doorcore-0.1.0/src/doorcore/__init__.py

@@ -0,0 +1,9 @@
+"""doorcore — door-Core v2 本地 AI 引擎（可 pip 安装的公开库）"""
+
+__version__ = "0.1.0"
+
+from .core import DoorCore
+from .api import application, get_ai
+from .cli import main as cli_main
+
+__all__ = ["DoorCore", "application", "get_ai", "cli_main", "__version__"]

doorcore-0.1.0/src/doorcore/api.py

@@ -0,0 +1,127 @@
+# door-core v2 —— 零依赖 OpenAI 兼容 API + 可部署 WSGI application
+# 本地运行：  python api.py            -> http://localhost:8765/v1
+# 公网部署：  平台 import 本文件的 application（WSGI）即可，模型名 door-core
+import os
+import json
+import importlib.resources as _resources
+
+from http.server import ThreadingHTTPServer, BaseHTTPRequestHandler
+from doorcore.core import DoorCore
+
+
+def _pkg_path(*parts):
+    try:
+        return str(_resources.files("doorcore").joinpath(*parts))
+    except Exception:
+        return os.path.join(os.path.dirname(os.path.abspath(__file__)), *parts)
+
+
+def load_config():
+    p = _pkg_path("data", "config.json")
+    if os.path.exists(p):
+        try:
+            return json.load(open(p, encoding="utf-8"))
+        except Exception:
+            pass
+    return {"n": 3}
+
+
+cfg = load_config()
+_AI = None
+
+
+def get_ai():
+    global _AI
+    if _AI is None:
+        n = int(cfg.get("n", 3))
+        _AI = DoorCore(n=n)
+        _AI.load_or_train()
+    return _AI
+
+
+def _chat_reply(messages):
+    text = ""
+    for m in reversed(messages or []):
+        if m.get("role") == "user":
+            text = m.get("content", "")
+            break
+    return get_ai().ask(text)
+
+
+def _completion_obj(reply):
+    return {
+        "object": "chat.completion",
+        "model": "door-core",
+        "choices": [{
+            "index": 0,
+            "message": {"role": "assistant", "content": reply},
+            "finish_reason": "stop",
+        }],
+        "usage": {"prompt_tokens": 0, "completion_tokens": 0, "total_tokens": 0},
+    }
+
+
+def _body_str(obj):
+    return json.dumps(obj, ensure_ascii=False).encode("utf-8")
+
+
+class Handler(BaseHTTPRequestHandler):
+    def _send(self, code, obj):
+        body = _body_str(obj)
+        self.send_response(code)
+        self.send_header("Content-Type", "application/json; charset=utf-8")
+        self.send_header("Content-Length", str(len(body)))
+        self.end_headers()
+        self.wfile.write(body)
+
+    def do_POST(self):
+        if self.path.rstrip("/") not in ("/v1/chat/completions", "/chat/completions"):
+            self._send(404, {"error": "not found"})
+            return
+        try:
+            ln = int(self.headers.get("Content-Length", 0) or 0)
+            raw = self.rfile.read(ln) if ln else b""
+            data = json.loads(raw.decode("utf-8"))
+        except Exception as e:
+            self._send(400, {"error": str(e)})
+            return
+        self._send(200, _completion_obj(_chat_reply(data.get("messages", []))))
+
+    def log_message(self, *a):
+        pass
+
+
+def application(environ, start_response):
+    path = (environ.get("PATH_INFO", "") or "").rstrip("/")
+    if environ.get("REQUEST_METHOD") == "POST" and path in (
+            "/v1/chat/completions", "/chat/completions"):
+        try:
+            ln = int(environ.get("CONTENT_LENGTH", 0) or 0)
+            raw = environ["wsgi.input"].read(ln) if ln else b""
+            data = json.loads(raw.decode("utf-8"))
+            reply = _chat_reply(data.get("messages", []))
+        except Exception as e:
+            reply = "出错了：" + str(e)
+        body = _body_str(_completion_obj(reply))
+        start_response("200 OK", [
+            ("Content-Type", "application/json; charset=utf-8"),
+            ("Content-Length", str(len(body))),
+        ])
+        return [body]
+    body = b'{"error":"not found"}'
+    start_response("404 Not Found", [
+        ("Content-Type", "application/json"),
+        ("Content-Length", str(len(body))),
+    ])
+    return [body]
+
+
+def main():
+    port = int(os.environ.get("PORT", "8765"))
+    print(f"[door-core v2 API] http://localhost:{port}/v1  (模型名 door-core)")
+    httpd = ThreadingHTTPServer(("0.0.0.0", port), Handler)
+    httpd.serve_forever()
+
+
+if __name__ == "__main__":
+    main()

doorcore-0.1.0/src/doorcore/cli.py

@@ -0,0 +1,57 @@
+# door-core v2 —— 命令行聊天（训练并从 .txt 语料加载，结果存 model.pkl）
+import os
+import sys
+import json
+import importlib.resources as _resources
+
+from doorcore.core import DoorCore
+
+
+def _pkg_path(*parts):
+    try:
+        return str(_resources.files("doorcore").joinpath(*parts))
+    except Exception:
+        return os.path.join(os.path.dirname(os.path.abspath(__file__)), *parts)
+
+
+def load_config():
+    p = _pkg_path("data", "config.json")
+    if os.path.exists(p):
+        try:
+            return json.load(open(p, encoding="utf-8"))
+        except Exception:
+            pass
+    return {"n": 3}
+
+
+def main():
+    cfg = load_config()
+    n = int(cfg.get("n", 3))
+    ai = DoorCore(n=n)
+
+    force = "--retrain" in sys.argv
+    status = ai.load_or_train(force=force)
+    print(f"[door-core v2] 模型: {status} | 知识条目: {len(ai.kb)} | "
+          f"马尔可夫节点: {len(ai.model)}")
+    print("聊天开始（/retrain 重新训练, /quit 退出）")
+
+    while True:
+        try:
+            u = input("你> ").strip()
+        except (EOFError, KeyboardInterrupt):
+            print()
+            break
+        if not u:
+            continue
+        if u in ("/quit", "/exit"):
+            break
+        if u == "/retrain":
+            ai.load_corpus()
+            ai.save_model()
+            print(f"已重新训练：知识条目 {len(ai.kb)}，马尔可夫节点 {len(ai.model)}")
+            continue
+        print("AI> " + ai.ask(u))
+
+
+if __name__ == "__main__":
+    main()