|
2 | 2 | # |
3 | 3 | # SPDX-License-Identifier: Apache-2.0 |
4 | 4 |
|
5 | | -from typing import Any, Optional |
| 5 | +from typing import Any |
6 | 6 |
|
7 | 7 | from haystack import component, default_from_dict, default_to_dict, logging |
8 | 8 | from haystack.components.generators.chat import OpenAIChatGenerator |
@@ -59,15 +59,15 @@ def __init__( |
59 | 59 | self, |
60 | 60 | *, |
61 | 61 | model: str, |
62 | | - api_base_url: str = "http://localhost:8321/v1/openai/v1", |
63 | | - organization: Optional[str] = None, |
64 | | - streaming_callback: Optional[StreamingCallbackT] = None, |
65 | | - generation_kwargs: Optional[dict[str, Any]] = None, |
66 | | - timeout: Optional[int] = None, |
67 | | - tools: Optional[ToolsType] = None, |
| 62 | + api_base_url: str = "http://localhost:8321/v1", |
| 63 | + organization: str | None = None, |
| 64 | + streaming_callback: StreamingCallbackT | None = None, |
| 65 | + generation_kwargs: dict[str, Any] | None = None, |
| 66 | + timeout: int | None = None, |
| 67 | + tools: ToolsType | None = None, |
68 | 68 | tools_strict: bool = False, |
69 | | - max_retries: Optional[int] = None, |
70 | | - http_client_kwargs: Optional[dict[str, Any]] = None, |
| 69 | + max_retries: int | None = None, |
| 70 | + http_client_kwargs: dict[str, Any] | None = None, |
71 | 71 | ): |
72 | 72 | """ |
73 | 73 | Creates an instance of LlamaStackChatGenerator. To use this chat generator, |
|
0 commit comments