diff --git a/docs/assets/fstring.png b/docs/assets/fstring.png index 260199d..0dd02bd 100644 Binary files a/docs/assets/fstring.png and b/docs/assets/fstring.png differ diff --git a/docs/assets/ollama_chat.png b/docs/assets/ollama_chat.png index c062a26..72ae97c 100644 Binary files a/docs/assets/ollama_chat.png and b/docs/assets/ollama_chat.png differ diff --git a/docs/assets/ollama_workflow.png b/docs/assets/ollama_workflow.png index 7d3cb4f..de17548 100644 Binary files a/docs/assets/ollama_workflow.png and b/docs/assets/ollama_workflow.png differ diff --git a/scripts/take_screenshots.py b/scripts/take_screenshots.py index 94e4f5d..c40bbfc 100644 --- a/scripts/take_screenshots.py +++ b/scripts/take_screenshots.py @@ -29,16 +29,18 @@ _STORAGE_STATE = { "localStorage": [ { "name": "workflow", - "value": json.dumps({ - "last_node_id": 0, - "last_link_id": 0, - "nodes": [], - "links": [], - "groups": [], - "config": {}, - "extra": {}, - "version": 0.4, - }), + "value": json.dumps( + { + "last_node_id": 0, + "last_link_id": 0, + "nodes": [], + "links": [], + "groups": [], + "config": {}, + "extra": {}, + "version": 0.4, + } + ), }, {"name": "Comfy.OpenWorkflowsPaths", "value": "[]"}, {"name": "Comfy.ActiveWorkflowIndex", "value": "0"}, @@ -609,7 +611,9 @@ async def main() -> int: async with async_playwright() as pw: browser = await pw.chromium.launch(headless=True) - context = await browser.new_context(viewport=VIEWPORT, storage_state=_STORAGE_STATE) + context = await browser.new_context( + viewport=VIEWPORT, storage_state=_STORAGE_STATE + ) page = await context.new_page() print(f"Opening {COMFYUI_URL} …") diff --git a/tests/test_ollama.py b/tests/test_ollama.py index 2ab9b86..50a8bc1 100644 --- a/tests/test_ollama.py +++ b/tests/test_ollama.py @@ -295,7 +295,9 @@ class TestUS3ModelLifecycle: monkeypatch.setattr(ollama_mod, "_post_json", fake_post) - OllamaLoadModel().load_model(client="http://localhost:11434", model="test-model") + OllamaLoadModel().load_model( + client="http://localhost:11434", model="test-model" + ) assert captured.get("url", "").endswith("/api/generate"), ( f"Expected URL ending in /api/generate but got: {captured.get('url')}" @@ -335,7 +337,9 @@ class TestUS3ModelLifecycle: monkeypatch.setattr(ollama_mod, "_post_json", fake_post) - OllamaLoadModel().load_model(client="http://localhost:11434", model="test-model") + OllamaLoadModel().load_model( + client="http://localhost:11434", model="test-model" + ) payload = captured.get("payload", {}) assert payload.get("keep_alive") == -1, (