OllamaChatGenerator return error in 'streaming_callback' argument

[NILICK]

Describe the bug
Based on this document, I can use streaming_callback as argument in OllamaChatGenerator.

My code:

from haystack.dataclasses import ChatMessage
from haystack_integrations.components.generators.ollama import OllamaChatGenerator
from haystack.components.generators.utils import print_streaming_chunk

generator = OllamaChatGenerator(model="phi3",
                                url = "http://localhost:11434/api/chat",
                                streaming_callback = print_streaming_chunk,
                                generation_kwargs={
                                    "num_predict": 100,
                                    "temperature": 0.9,
                                    },
                                )

messages = [ChatMessage.from_system("\nYou are a helpful, respectful and honest assistant"),
ChatMessage.from_user("What's Natural Language Processing?")]

print(generator.run(messages=messages))

But it return error

Error message

TypeError: OllamaChatGenerator.init() got an unexpected keyword argument 'streaming_callback'

System:

• OS: Ubuntu
• Haystack version (commit or version number): 2.2.1

[anakin87]

Hello!

I have tried your code with the latest version of ollama-haystack (0.0.7) and works for me.

Please try reinstalling the latest version (pip install -U ollama-haystack) and let us know.

[NILICK]

Hi @anakin87, Thanks for your fast and correct reply.