From e16c819e832aac0324a1b1e824dc9a82d6d3021d Mon Sep 17 00:00:00 2001 From: Yiorgis Gozadinos Date: Thu, 16 Apr 2026 11:50:55 +0300 Subject: [PATCH] Adapt to new haiku.skills --- app/backend/main.py | 4 +++- examples/custom_agent_agui.py | 2 +- haiku_rag_slim/haiku/rag/chat/app.py | 2 +- haiku_rag_slim/pyproject.toml | 2 +- uv.lock | 8 ++++---- 5 files changed, 10 insertions(+), 8 deletions(-) diff --git a/app/backend/main.py b/app/backend/main.py index 29a28957..d76b33e7 100644 --- a/app/backend/main.py +++ b/app/backend/main.py @@ -91,7 +91,9 @@ async def stream_chat(request: Request) -> Response: adapter = AGUIAdapter(agent=agent, run_input=run_input, accept=accept) async def event_stream(): - async with run_agui_stream(toolset, adapter, deps=SkillDeps()) as stream: + async with run_agui_stream( + adapter, toolset=toolset, deps=SkillDeps() + ) as stream: async for chunk in adapter.encode_stream(stream): yield chunk diff --git a/examples/custom_agent_agui.py b/examples/custom_agent_agui.py index bc27df25..c0d7f141 100644 --- a/examples/custom_agent_agui.py +++ b/examples/custom_agent_agui.py @@ -52,7 +52,7 @@ async def stream_chat(request: Request) -> Response: adapter = AGUIAdapter(agent=agent, run_input=run_input, accept=accept) async def event_stream(): - async with run_agui_stream(toolset, adapter) as stream: + async with run_agui_stream(adapter, toolset=toolset) as stream: async for chunk in adapter.encode_stream(stream): yield chunk diff --git a/haiku_rag_slim/haiku/rag/chat/app.py b/haiku_rag_slim/haiku/rag/chat/app.py index da58774f..5d6bcb14 100644 --- a/haiku_rag_slim/haiku/rag/chat/app.py +++ b/haiku_rag_slim/haiku/rag/chat/app.py @@ -221,7 +221,7 @@ class ChatApp(App): tool_args_deltas: dict[str, str] = {} try: - async with run_agui_stream(self._toolset, adapter) as stream: + async with run_agui_stream(adapter, toolset=self._toolset) as stream: async for event in stream: if event.type == EventType.TEXT_MESSAGE_START: chat_history.hide_thinking() diff --git a/haiku_rag_slim/pyproject.toml b/haiku_rag_slim/pyproject.toml index 1424ecef..7817e0eb 100644 --- a/haiku_rag_slim/pyproject.toml +++ b/haiku_rag_slim/pyproject.toml @@ -24,7 +24,7 @@ classifiers = [ dependencies = [ "cachetools>=7.0.5", "docling-core>=2.71.0", - "haiku.skills>=0.13.3", + "haiku.skills>=0.14.0", "httpx>=0.28.1", "jinja2>=3.1.0", "jsonpatch>=1.33", diff --git a/uv.lock b/uv.lock index b08917c4..85a303d2 100644 --- a/uv.lock +++ b/uv.lock @@ -1572,7 +1572,7 @@ requires-dist = [ { name = "cohere", marker = "extra == 'cohere'", specifier = ">=5.21.1" }, { name = "docling", marker = "extra == 'docling'", specifier = ">=2.84.0" }, { name = "docling-core", specifier = ">=2.71.0" }, - { name = "haiku-skills", specifier = ">=0.13.3" }, + { name = "haiku-skills", specifier = ">=0.14.0" }, { name = "httpx", specifier = ">=0.28.1" }, { name = "jinja2", specifier = ">=3.1.0" }, { name = "jsonpatch", specifier = ">=1.33" }, @@ -1606,7 +1606,7 @@ provides-extras = ["docling", "voyageai", "mxbai", "cohere", "zeroentropy", "jin [[package]] name = "haiku-skills" -version = "0.13.3" +version = "0.14.0" source = { registry = "https://pypi.org/simple" } dependencies = [ { name = "ag-ui-protocol" }, @@ -1616,9 +1616,9 @@ dependencies = [ { name = "pyyaml" }, { name = "skills-ref" }, ] -sdist = { url = "https://files.pythonhosted.org/packages/9c/65/44140f3289ebf3647c6f3dbf051de2162bc4ade11f020f98676e14d1d8b7/haiku_skills-0.13.3.tar.gz", hash = "sha256:4c918d6c8183c01274a34111159eeb56d1befed5d38a39779d9e511f92d4e088", size = 250458, upload-time = "2026-04-14T09:13:29.975Z" } +sdist = { url = "https://files.pythonhosted.org/packages/89/c4/82a6b82f70726a2e759aad4c6c553309f2cc2ca7f3c157b8a15fd14da709/haiku_skills-0.14.0.tar.gz", hash = "sha256:27074a171060a0ecae6b89c6b7756b3b8ed0dfb957a3f5076ab1d158e68feb82", size = 250637, upload-time = "2026-04-16T08:47:33.152Z" } wheels = [ - { url = "https://files.pythonhosted.org/packages/b9/c6/e3aaed844b80c94b8e8462b80751b372061b19bf17f57e531f327e0366f7/haiku_skills-0.13.3-py3-none-any.whl", hash = "sha256:2bf0d28fd3e7c65b627f1bec33245bc36dc997d95d6efcea949cb9260aec2d73", size = 31530, upload-time = "2026-04-14T09:13:28.787Z" }, + { url = "https://files.pythonhosted.org/packages/e5/7a/bc53abcbae8bf1aa013379f49f6690fbaff55f80a7b997824103a5f44384/haiku_skills-0.14.0-py3-none-any.whl", hash = "sha256:698d0012bcf06f43499c30aaf6a3b7b772c66737696bfa3f93a6b113fb0f6b17", size = 31613, upload-time = "2026-04-16T08:47:31.68Z" }, ] [[package]]