Ollama in Flows

CrewAI Community Support
1

In the documentation of Flows at Flows - CrewAI

The example given uses OpenAI as follows:

from crewai.flow.flow import Flow, listen, start
from dotenv import load_dotenv
from litellm import completion


class ExampleFlow(Flow):
    model = "gpt-4o-mini"

    @start()
    def generate_city(self):
        print("Starting flow")

        response = completion(
            model=self.model,
            messages=[
                {
                    "role": "user",
                    "content": "Return the name of a random city in the world.",
                },
            ],
        )

Can flows use other LLM’s like Ollama? How would that look?