from langchain_ollama import ChatOllama
import controlflow as cf
from controlflow.tasks.validators import between

cf.defaults.model = ChatOllama(
    model = "llama3.2",
)


optimist = cf.Agent()

def sentiment(text: str) -> float:
    return cf.run(
        "Classify the sentiment of the text as a value between 0 and 1",
        agents=[optimist],
        result_type=float,
        result_validator=between(0, 1),
        context={"text": text},
    )

sentiment("I love ControlFlow!")