mirror of
https://github.com/hansjone/oclaw.git
synced 2026-10-09 07:20:44 +08:00
oliver
ba3836f00f
- 在 oclaw/ 下重新初始化 Git 仓库 - 补齐子仓库 .gitignore，避免提交本地运行态数据（_local、node_modules、logs 等） - 提交当前工程代码与配置 Made-with: Cursor
40 lines
1.4 KiB
Python
40 lines
1.4 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
def _runtime_get(runtime: Any, key: str):
|
|
if isinstance(runtime, dict):
|
|
return runtime.get(key)
|
|
return getattr(runtime, key, None)
|
|
|
|
|
|
def list_runtime_image_generation_providers(runtime: Any = None) -> list[dict]:
|
|
providers = _runtime_get(runtime, "image_generation_providers")
|
|
if isinstance(providers, list):
|
|
return [p for p in providers if isinstance(p, dict)]
|
|
return []
|
|
|
|
def generate_image(*, prompt: str, provider_id: str | None = None, runtime: Any = None, **kwargs) -> dict:
|
|
providers = list_runtime_image_generation_providers(runtime)
|
|
if not providers:
|
|
return {"ok": False, "error": "no_image_generation_provider_registered"}
|
|
|
|
chosen = None
|
|
if provider_id:
|
|
chosen = next((p for p in providers if str(p.get("id")) == provider_id), None)
|
|
if chosen is None:
|
|
chosen = providers[0]
|
|
|
|
generator = chosen.get("generate")
|
|
if callable(generator):
|
|
result = generator(prompt=prompt, **kwargs)
|
|
if isinstance(result, dict):
|
|
return {"ok": True, "provider": chosen.get("id"), **result}
|
|
return {"ok": True, "provider": chosen.get("id"), "result": result}
|
|
|
|
return {
|
|
"ok": True,
|
|
"provider": chosen.get("id"),
|
|
"prompt": prompt,
|
|
"note": "provider_has_no_generate_callable",
|
|
}
|