From 96ef111afae17ea2cd58ac46cd3ad2c49edb92c5 Mon Sep 17 00:00:00 2001 From: anakin87 Date: Mon, 12 Jan 2026 15:43:28 +0100 Subject: [PATCH] chore: pin llama-stack; X|Y typing --- integrations/llama_stack/pyproject.toml | 2 +- .../llama_stack/chat/chat_generator.py | 16 ++++++++-------- 2 files changed, 9 insertions(+), 9 deletions(-) diff --git a/integrations/llama_stack/pyproject.toml b/integrations/llama_stack/pyproject.toml index ea96a7fdc6..093654f6c8 100644 --- a/integrations/llama_stack/pyproject.toml +++ b/integrations/llama_stack/pyproject.toml @@ -20,7 +20,7 @@ classifiers = [ "Programming Language :: Python :: Implementation :: CPython", "Programming Language :: Python :: Implementation :: PyPy", ] -dependencies = ["haystack-ai>=2.19.0", "llama-stack>=0.2.17"] +dependencies = ["haystack-ai>=2.22.0", "llama-stack>=0.2.17,<0.4.0"] [project.urls] Documentation = "https://github.com/deepset-ai/haystack-core-integrations/tree/main/integrations/llama-stack#readme" diff --git a/integrations/llama_stack/src/haystack_integrations/components/generators/llama_stack/chat/chat_generator.py b/integrations/llama_stack/src/haystack_integrations/components/generators/llama_stack/chat/chat_generator.py index 82a638b858..6f2d58507a 100644 --- a/integrations/llama_stack/src/haystack_integrations/components/generators/llama_stack/chat/chat_generator.py +++ b/integrations/llama_stack/src/haystack_integrations/components/generators/llama_stack/chat/chat_generator.py @@ -2,7 +2,7 @@ # # SPDX-License-Identifier: Apache-2.0 -from typing import Any, Optional +from typing import Any from haystack import component, default_from_dict, default_to_dict, logging from haystack.components.generators.chat import OpenAIChatGenerator @@ -60,14 +60,14 @@ def __init__( *, model: str, api_base_url: str = "http://localhost:8321/v1/openai/v1", - organization: Optional[str] = None, - streaming_callback: Optional[StreamingCallbackT] = None, - generation_kwargs: Optional[dict[str, Any]] = None, - timeout: Optional[int] = None, - tools: Optional[ToolsType] = None, + organization: str | None = None, + streaming_callback: StreamingCallbackT | None = None, + generation_kwargs: dict[str, Any] | None = None, + timeout: int | None = None, + tools: ToolsType | None = None, tools_strict: bool = False, - max_retries: Optional[int] = None, - http_client_kwargs: Optional[dict[str, Any]] = None, + max_retries: int | None = None, + http_client_kwargs: dict[str, Any] | None = None, ): """ Creates an instance of LlamaStackChatGenerator. To use this chat generator,