Publish current ComfyUI O1Key code baseline

Replace the prior release tree with the current plugin, frontend, tests, and documentation. Document retired node IDs and the public Gitea update source.
This commit is contained in:
Jony
2026-09-24 19:56:48 +08:00
parent 3e337722ab
commit ba920f2b66
183 changed files with 49496 additions and 9934 deletions
+40
View File
@@ -0,0 +1,40 @@
import sys
import types
import unittest
from pathlib import Path
PACKAGE_DIR = Path(__file__).resolve().parents[1]
sys.path.insert(0, str(PACKAGE_DIR.parent))
# Import the node module without running the root registry, whose unrelated
# ComfyUI imports initialize CUDA during package import.
package = types.ModuleType("comfyui_o1key")
package.__path__ = [str(PACKAGE_DIR)]
sys.modules.setdefault("comfyui_o1key", package)
nodes_package = types.ModuleType("comfyui_o1key.nodes")
nodes_package.__path__ = [str(PACKAGE_DIR / "nodes")]
sys.modules.setdefault("comfyui_o1key.nodes", nodes_package)
from comfyui_o1key.nodes.universal_llm import ( # noqa: E402
DEFAULT_MODEL,
SUPPORTED_MODELS,
UniversalLLMChat,
)
class UniversalLLMModelTests(unittest.TestCase):
def test_gpt_6_sol_is_supported_and_default(self):
schema = UniversalLLMChat.GET_SCHEMA()
schema.validate()
model_input = next(item for item in schema.inputs if item.id == "模型")
self.assertEqual(DEFAULT_MODEL, "gpt-6-sol")
self.assertEqual(SUPPORTED_MODELS[0], DEFAULT_MODEL)
self.assertIn("gpt-6-astra", SUPPORTED_MODELS)
self.assertEqual(model_input.options, SUPPORTED_MODELS)
self.assertEqual(model_input.default, DEFAULT_MODEL)
if __name__ == "__main__":
unittest.main()