openbee 0.1.0__tar.gz

openbee-0.1.0/PKG-INFO

@@ -0,0 +1,70 @@
+Metadata-Version: 2.4
+Name: openbee
+Version: 0.1.0
+Summary: Python CLI for basic OpenBee commands
+Author: openbee
+License-Expression: MIT
+Project-URL: Homepage, https://github.com/sanpiekankan/openbee
+Project-URL: Repository, https://github.com/sanpiekankan/openbee
+Keywords: openbee,cli,ai,agent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3 :: Only
+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: Environment :: Console
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+
+# OpenBee Python CLI
+
+OpenBee Python CLI 是 OpenBee 的 Python 版本基础命令行工具，提供最小可用能力：
+
+- 查看内置 Bee 角色
+- 配置 LLM 参数（API Key、模型、Base URL、温度）
+- 向指定角色发送任务并获取回复
+
+## 安装
+
+```bash
+pip install openbee
+```
+
+## 快速开始
+
+### 1) 查看角色
+
+```bash
+openbee list
+```
+
+### 2) 配置模型
+
+```bash
+openbee config --api-key <YOUR_API_KEY> --model gpt-4o
+```
+
+### 3) 发起任务
+
+```bash
+openbee ask worker "帮我规划一个后端服务结构"
+```
+
+## 配置说明
+
+默认配置文件位置：
+
+```text
+~/.openbee/config.json
+```
+
+你也可以通过环境变量覆盖配置目录：
+
+```bash
+export OPENBEE_CONFIG_HOME=/path/to/custom/config
+```
+
+## 许可证
+
+MIT

openbee-0.1.0/README.md

@@ -0,0 +1,51 @@
+# OpenBee Python CLI
+
+OpenBee Python CLI 是 OpenBee 的 Python 版本基础命令行工具，提供最小可用能力：
+
+- 查看内置 Bee 角色
+- 配置 LLM 参数（API Key、模型、Base URL、温度）
+- 向指定角色发送任务并获取回复
+
+## 安装
+
+```bash
+pip install openbee
+```
+
+## 快速开始
+
+### 1) 查看角色
+
+```bash
+openbee list
+```
+
+### 2) 配置模型
+
+```bash
+openbee config --api-key <YOUR_API_KEY> --model gpt-4o
+```
+
+### 3) 发起任务
+
+```bash
+openbee ask worker "帮我规划一个后端服务结构"
+```
+
+## 配置说明
+
+默认配置文件位置：
+
+```text
+~/.openbee/config.json
+```
+
+你也可以通过环境变量覆盖配置目录：
+
+```bash
+export OPENBEE_CONFIG_HOME=/path/to/custom/config
+```
+
+## 许可证
+
+MIT

openbee-0.1.0/pyproject.toml

@@ -0,0 +1,38 @@
+[build-system]
+requires = ["setuptools>=68", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "openbee"
+version = "0.1.0"
+description = "Python CLI for basic OpenBee commands"
+readme = "README.md"
+requires-python = ">=3.9"
+license = "MIT"
+authors = [{ name = "openbee" }]
+keywords = ["openbee", "cli", "ai", "agent"]
+classifiers = [
+  "Programming Language :: Python :: 3",
+  "Programming Language :: Python :: 3 :: Only",
+  "Programming Language :: Python :: 3.9",
+  "Programming Language :: Python :: 3.10",
+  "Programming Language :: Python :: 3.11",
+  "Programming Language :: Python :: 3.12",
+  "Environment :: Console",
+]
+
+[project.urls]
+Homepage = "https://github.com/sanpiekankan/openbee"
+Repository = "https://github.com/sanpiekankan/openbee"
+
+[project.scripts]
+openbee = "openbee_py.cli:main"
+
+[tool.setuptools]
+package-dir = {"" = "src"}
+
+[tool.setuptools.packages.find]
+where = ["src"]
+
+[tool.pytest.ini_options]
+testpaths = ["tests"]

openbee-0.1.0/setup.cfg

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

openbee-0.1.0/src/openbee.egg-info/PKG-INFO

@@ -0,0 +1,70 @@
+Metadata-Version: 2.4
+Name: openbee
+Version: 0.1.0
+Summary: Python CLI for basic OpenBee commands
+Author: openbee
+License-Expression: MIT
+Project-URL: Homepage, https://github.com/sanpiekankan/openbee
+Project-URL: Repository, https://github.com/sanpiekankan/openbee
+Keywords: openbee,cli,ai,agent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3 :: Only
+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: Environment :: Console
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+
+# OpenBee Python CLI
+
+OpenBee Python CLI 是 OpenBee 的 Python 版本基础命令行工具，提供最小可用能力：
+
+- 查看内置 Bee 角色
+- 配置 LLM 参数（API Key、模型、Base URL、温度）
+- 向指定角色发送任务并获取回复
+
+## 安装
+
+```bash
+pip install openbee
+```
+
+## 快速开始
+
+### 1) 查看角色
+
+```bash
+openbee list
+```
+
+### 2) 配置模型
+
+```bash
+openbee config --api-key <YOUR_API_KEY> --model gpt-4o
+```
+
+### 3) 发起任务
+
+```bash
+openbee ask worker "帮我规划一个后端服务结构"
+```
+
+## 配置说明
+
+默认配置文件位置：
+
+```text
+~/.openbee/config.json
+```
+
+你也可以通过环境变量覆盖配置目录：
+
+```bash
+export OPENBEE_CONFIG_HOME=/path/to/custom/config
+```
+
+## 许可证
+
+MIT

openbee-0.1.0/src/openbee.egg-info/SOURCES.txt

@@ -0,0 +1,13 @@
+README.md
+pyproject.toml
+src/openbee.egg-info/PKG-INFO
+src/openbee.egg-info/SOURCES.txt
+src/openbee.egg-info/dependency_links.txt
+src/openbee.egg-info/entry_points.txt
+src/openbee.egg-info/top_level.txt
+src/openbee_py/__init__.py
+src/openbee_py/__main__.py
+src/openbee_py/cli.py
+src/openbee_py/config.py
+src/openbee_py/core.py
+tests/test_cli.py

openbee-0.1.0/src/openbee.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

openbee-0.1.0/src/openbee.egg-info/entry_points.txt

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

openbee-0.1.0/src/openbee.egg-info/top_level.txt

@@ -0,0 +1 @@
+openbee_py

openbee-0.1.0/src/openbee_py/__init__.py

@@ -0,0 +1,3 @@
+__all__ = ["__version__"]
+
+__version__ = "0.1.0"

openbee-0.1.0/src/openbee_py/__main__.py

@@ -0,0 +1,9 @@
+from __future__ import annotations
+
+import sys
+
+from .cli import main
+
+
+if __name__ == "__main__":
+    raise SystemExit(main(sys.argv[1:]))

openbee-0.1.0/src/openbee_py/cli.py

@@ -0,0 +1,121 @@
+from __future__ import annotations
+
+import argparse
+import json
+import sys
+from typing import Sequence
+
+from .config import load_config, update_llm_config
+from .core import ask_openbee, get_role, list_roles
+
+
+def _build_parser() -> argparse.ArgumentParser:
+    parser = argparse.ArgumentParser(
+        prog="openbee",
+        description="OpenBee Python CLI (basic commands)",
+    )
+    subparsers = parser.add_subparsers(dest="command")
+
+    subparsers.add_parser("list", help="List available bee roles")
+
+    config_parser = subparsers.add_parser("config", help="Show or update LLM config")
+    config_parser.add_argument("--provider", help="LLM provider name")
+    config_parser.add_argument("--api-key", help="LLM API key")
+    config_parser.add_argument("--model", help="LLM model name")
+    config_parser.add_argument("--base-url", help="OpenAI-compatible base URL")
+    config_parser.add_argument("--temperature", type=float, help="Sampling temperature")
+
+    ask_parser = subparsers.add_parser("ask", help="Ask a role to perform a task")
+    ask_parser.add_argument("role", help="Role id, e.g. worker")
+    ask_parser.add_argument("task", nargs="+", help="Task text")
+    ask_parser.add_argument("--api-key", help="Override API key")
+    ask_parser.add_argument("--model", help="Override model")
+    ask_parser.add_argument("--base-url", help="Override base URL")
+    ask_parser.add_argument("--temperature", type=float, help="Override temperature")
+
+    return parser
+
+
+def _run_list() -> int:
+    roles = list_roles()
+    for role in roles:
+        print(f'{role["name"]} ({role["id"]})')
+        print(f'  {role["description"]}')
+    return 0
+
+
+def _run_config(args: argparse.Namespace) -> int:
+    has_updates = any(
+        value is not None
+        for value in (
+            args.provider,
+            args.api_key,
+            args.model,
+            args.base_url,
+            args.temperature,
+        )
+    )
+    if has_updates:
+        config = update_llm_config(
+            provider=args.provider,
+            api_key=args.api_key,
+            model=args.model,
+            base_url=args.base_url,
+            temperature=args.temperature,
+        )
+        print("Configuration updated.")
+        print(json.dumps(config, ensure_ascii=False, indent=2))
+        return 0
+
+    config = load_config()
+    print(json.dumps(config, ensure_ascii=False, indent=2))
+    return 0
+
+
+def _run_ask(args: argparse.Namespace) -> int:
+    role = get_role(args.role)
+    if role is None:
+        print(f'Error: role "{args.role}" not found.', file=sys.stderr)
+        return 1
+
+    config = load_config().get("llm", {})
+    api_key = args.api_key or config.get("api_key", "")
+    model = args.model or config.get("model", "gpt-4o")
+    base_url = args.base_url or config.get("base_url", "https://api.openai.com/v1")
+    temperature = args.temperature if args.temperature is not None else float(config.get("temperature", 0.7))
+
+    if not api_key:
+        print('Error: API key missing. Run "openbee config --api-key <key>" first.', file=sys.stderr)
+        return 1
+
+    task = " ".join(args.task)
+    try:
+        content = ask_openbee(
+            role=role,
+            task=task,
+            api_key=api_key,
+            model=model,
+            base_url=base_url,
+            temperature=temperature,
+        )
+    except RuntimeError as exc:
+        print(f"Error: {exc}", file=sys.stderr)
+        return 1
+
+    print(content)
+    return 0
+
+
+def main(argv: Sequence[str] | None = None) -> int:
+    parser = _build_parser()
+    args = parser.parse_args(argv)
+
+    if args.command == "list":
+        return _run_list()
+    if args.command == "config":
+        return _run_config(args)
+    if args.command == "ask":
+        return _run_ask(args)
+
+    parser.print_help()
+    return 0

openbee-0.1.0/src/openbee_py/config.py

@@ -0,0 +1,60 @@
+from __future__ import annotations
+
+import copy
+import json
+import os
+from pathlib import Path
+from typing import Any
+
+
+DEFAULT_CONFIG: dict[str, Any] = {
+    "llm": {
+        "provider": "openai",
+        "api_key": "",
+        "model": "gpt-4o",
+        "base_url": "https://api.openai.com/v1",
+        "temperature": 0.7,
+    }
+}
+
+
+def get_config_dir() -> Path:
+    custom = os.getenv("OPENBEE_CONFIG_HOME")
+    if custom:
+        return Path(custom).expanduser().resolve()
+    return Path.home() / ".openbee"
+
+
+def get_config_path() -> Path:
+    return get_config_dir() / "config.json"
+
+
+def load_config() -> dict[str, Any]:
+    path = get_config_path()
+    if not path.exists():
+        return copy.deepcopy(DEFAULT_CONFIG)
+    data = json.loads(path.read_text(encoding="utf-8"))
+    merged = copy.deepcopy(DEFAULT_CONFIG)
+    if isinstance(data, dict):
+        for key, value in data.items():
+            if isinstance(value, dict) and isinstance(merged.get(key), dict):
+                merged[key].update(value)
+            else:
+                merged[key] = value
+    return merged
+
+
+def save_config(config: dict[str, Any]) -> None:
+    path = get_config_path()
+    path.parent.mkdir(parents=True, exist_ok=True)
+    path.write_text(json.dumps(config, ensure_ascii=False, indent=2), encoding="utf-8")
+
+
+def update_llm_config(**kwargs: Any) -> dict[str, Any]:
+    config = load_config()
+    llm = config.setdefault("llm", {})
+    for key, value in kwargs.items():
+        if value is not None:
+            llm[key] = value
+    save_config(config)
+    return config

openbee-0.1.0/src/openbee_py/core.py

@@ -0,0 +1,95 @@
+from __future__ import annotations
+
+import json
+import urllib.error
+import urllib.request
+from typing import Any
+
+
+ROLES: list[dict[str, str]] = [
+    {
+        "id": "worker",
+        "name": "Worker Bee",
+        "description": "General-purpose assistant for everyday tasks.",
+        "system_prompt": (
+            "You are a diligent Worker Bee in the OpenBee Hive. "
+            "Assist users with general tasks efficiently and precisely."
+        ),
+    },
+    {
+        "id": "researcher",
+        "name": "Researcher Bee",
+        "description": "Specialized assistant for research and information synthesis.",
+        "system_prompt": (
+            "You are a Researcher Bee in the OpenBee Hive. "
+            "Provide structured and accurate research-focused answers."
+        ),
+    },
+    {
+        "id": "architect",
+        "name": "Architect Bee",
+        "description": "Specialized assistant for system design and planning.",
+        "system_prompt": (
+            "You are an Architect Bee in the OpenBee Hive. "
+            "Design robust, maintainable solutions with clear trade-offs."
+        ),
+    },
+]
+
+
+def list_roles() -> list[dict[str, str]]:
+    return ROLES
+
+
+def get_role(role_id: str) -> dict[str, str] | None:
+    for role in ROLES:
+        if role["id"] == role_id:
+            return role
+    return None
+
+
+def ask_openbee(
+    role: dict[str, str],
+    task: str,
+    api_key: str,
+    model: str,
+    base_url: str,
+    temperature: float = 0.7,
+    timeout: float = 60.0,
+) -> str:
+    endpoint = f"{base_url.rstrip('/')}/chat/completions"
+    payload = {
+        "model": model,
+        "temperature": temperature,
+        "messages": [
+            {"role": "system", "content": role["system_prompt"]},
+            {"role": "user", "content": task},
+        ],
+    }
+    req = urllib.request.Request(
+        endpoint,
+        data=json.dumps(payload).encode("utf-8"),
+        headers={
+            "Content-Type": "application/json",
+            "Authorization": f"Bearer {api_key}",
+        },
+        method="POST",
+    )
+    try:
+        with urllib.request.urlopen(req, timeout=timeout) as response:
+            raw = response.read().decode("utf-8")
+    except urllib.error.HTTPError as exc:
+        details = exc.read().decode("utf-8", errors="replace")
+        raise RuntimeError(f"request failed: {exc.code} {details}") from exc
+    except urllib.error.URLError as exc:
+        raise RuntimeError(f"request failed: {exc.reason}") from exc
+
+    data: dict[str, Any] = json.loads(raw)
+    choices = data.get("choices")
+    if not isinstance(choices, list) or not choices:
+        raise RuntimeError("invalid response: missing choices")
+    message = choices[0].get("message", {})
+    content = message.get("content")
+    if not isinstance(content, str) or not content.strip():
+        raise RuntimeError("invalid response: missing content")
+    return content

openbee-0.1.0/tests/test_cli.py

@@ -0,0 +1,55 @@
+from __future__ import annotations
+
+import io
+import os
+import tempfile
+import unittest
+from contextlib import redirect_stdout
+from unittest.mock import patch
+
+from openbee_py import cli
+from openbee_py.config import get_config_path, load_config, update_llm_config
+
+
+class TestOpenBeePyCli(unittest.TestCase):
+    def test_list_command(self) -> None:
+        buf = io.StringIO()
+        with redirect_stdout(buf):
+            code = cli.main(["list"])
+        output = buf.getvalue()
+        self.assertEqual(code, 0)
+        self.assertIn("Worker Bee (worker)", output)
+
+    def test_config_roundtrip(self) -> None:
+        with tempfile.TemporaryDirectory() as temp_dir:
+            with patch.dict(os.environ, {"OPENBEE_CONFIG_HOME": temp_dir}, clear=False):
+                code = cli.main(
+                    [
+                        "config",
+                        "--api-key",
+                        "test-key",
+                        "--model",
+                        "gpt-4o-mini",
+                    ]
+                )
+                self.assertEqual(code, 0)
+                config = load_config()
+                self.assertEqual(config["llm"]["api_key"], "test-key")
+                self.assertEqual(config["llm"]["model"], "gpt-4o-mini")
+                self.assertTrue(get_config_path().exists())
+
+    def test_ask_command_uses_client(self) -> None:
+        with tempfile.TemporaryDirectory() as temp_dir:
+            with patch.dict(os.environ, {"OPENBEE_CONFIG_HOME": temp_dir}, clear=False):
+                update_llm_config(api_key="k", model="m", base_url="https://api.openai.com/v1")
+                with patch("openbee_py.cli.ask_openbee", return_value="ok") as mocked:
+                    buf = io.StringIO()
+                    with redirect_stdout(buf):
+                        code = cli.main(["ask", "worker", "hello"])
+                    self.assertEqual(code, 0)
+                    self.assertIn("ok", buf.getvalue())
+                    mocked.assert_called_once()
+
+
+if __name__ == "__main__":
+    unittest.main()