mirror of
https://github.com/hansjone/oclaw.git
synced 2026-10-17 06:11:01 +08:00
初始化:独立 oclaw 仓库首提交
- 在 oclaw/ 下重新初始化 Git 仓库 - 补齐子仓库 .gitignore,避免提交本地运行态数据(_local、node_modules、logs 等) - 提交当前工程代码与配置 Made-with: Cursor
This commit is contained in:
commit
ba3836f00f
579 changed files with 83112 additions and 0 deletions
48
tests/test_gateway_image_generation_chain.py
Normal file
48
tests/test_gateway_image_generation_chain.py
Normal file
|
|
@ -0,0 +1,48 @@
|
|||
from __future__ import annotations
|
||||
|
||||
import importlib.util
|
||||
from pathlib import Path
|
||||
|
||||
from oclaw.gateway.server_plugins import load_gateway_plugins
|
||||
|
||||
|
||||
def _load_image_core_generate_image():
|
||||
file_path = (Path(__file__).resolve().parents[1] / "extensions" / "image-generation-core" / "api.py").resolve()
|
||||
spec = importlib.util.spec_from_file_location("test_image_generation_core_api", str(file_path))
|
||||
if spec is None or spec.loader is None:
|
||||
raise RuntimeError("failed to load image-generation-core api module")
|
||||
module = importlib.util.module_from_spec(spec)
|
||||
spec.loader.exec_module(module)
|
||||
return getattr(module, "generate_image")
|
||||
|
||||
|
||||
def test_openai_registers_image_generation_provider_in_gateway_registry() -> None:
|
||||
out = load_gateway_plugins(
|
||||
cfg={"plugins": {"enabled": ["openai"]}},
|
||||
workspace_dir=".",
|
||||
log={},
|
||||
core_gateway_handlers={},
|
||||
base_methods=["ping"],
|
||||
)
|
||||
providers = out.plugin_registry.get("image_generation_providers") or []
|
||||
assert isinstance(providers, list)
|
||||
assert any(isinstance(p, dict) and p.get("id") == "openai" for p in providers)
|
||||
|
||||
|
||||
def test_image_generation_core_uses_registered_openai_provider() -> None:
|
||||
out = load_gateway_plugins(
|
||||
cfg={"plugins": {"enabled": ["openai"]}},
|
||||
workspace_dir=".",
|
||||
log={},
|
||||
core_gateway_handlers={},
|
||||
base_methods=["ping"],
|
||||
)
|
||||
runtime = {"image_generation_providers": out.plugin_registry.get("image_generation_providers") or []}
|
||||
generate_image = _load_image_core_generate_image()
|
||||
result = generate_image(prompt="sunset over city", provider_id="openai", runtime=runtime, size="1024x1024")
|
||||
assert result.get("ok") is True
|
||||
assert result.get("provider") == "openai"
|
||||
assert result.get("model") == "gpt-image-1"
|
||||
assert result.get("prompt") == "sunset over city"
|
||||
assert result.get("size") == "1024x1024"
|
||||
|
||||
Loading…
Add table
Add a link
Reference in a new issue