Data Transformations and Validation#

flowchart TD
    Start([User Code]) --> Init{Initialize Ollama}

    Init --> SetConfig[Set Configuration]
    SetConfig --> StoreModel[Store model name]
    StoreModel --> StoreURL[Store base_url]
    StoreURL --> StoreTimeout[Store timeout]
    StoreTimeout --> StoreTemp[Store temperature]
    StoreTemp --> StoreJSON[Store json_mode flag]
    StoreJSON --> StoreKwargs[Store additional_kwargs]

    StoreKwargs --> CreateMetadata[Create Metadata]
    CreateMetadata --> SetChatFlag[Set is_chat_model=True]
    SetChatFlag --> SetFnCallFlag[Set is_function_calling_model]
    SetFnCallFlag --> SetContext[Set context_window]
    SetContext --> Ready([Ollama Instance Ready])

    Ready --> CallType{Call Type?}

    CallType -->|chat| ChatPath[Chat Path]
    CallType -->|complete| CompletePath[Complete Path]
    CallType -->|generate_tool_calls| ToolsPath[Tools Path]
    CallType -->|stream_chat| StreamPath[Stream Path]

    %% Chat Path
    ChatPath --> ValidateMessages{Messages Valid?}
    ValidateMessages -->|No| Error1[Raise ValueError]
    ValidateMessages -->|Yes| BuildChatReq[Build Chat Request]

    BuildChatReq --> AddModel[Add model name]
    AddModel --> ConvertMessages[Convert Messages to dicts]
    ConvertMessages --> AddOptions[Add options: temperature, etc.]
    AddOptions --> CheckJSON{json_mode?}
    CheckJSON -->|True| AddFormat[Add format: json]
    CheckJSON -->|False| AddKeepAlive
    AddFormat --> AddKeepAlive[Add keep_alive]

    AddKeepAlive --> EnsureClient[Ensure client initialized]
    EnsureClient --> CheckClient{Client exists?}
    CheckClient -->|No| CreateClient[Create Client with base_url, timeout]
    CheckClient -->|Yes| SendRequest
    CreateClient --> SendRequest[Send HTTP POST /api/chat]

    SendRequest --> ReceiveRaw[Receive raw dict response]
    ReceiveRaw --> ParseResponse[_chat_from_response]

    ParseResponse --> ExtractMsg[Extract message dict]
    ExtractMsg --> ParseRole[Parse role: assistant]
    ParseRole --> ParseContent[Parse content: str]
    ParseContent --> CheckTools{tool_calls present?}
    CheckTools -->|Yes| ParseToolCalls[Parse tool_calls array]
    CheckTools -->|No| CreateMessage1
    ParseToolCalls --> CreateMessage1[Create Message object]

    CreateMessage1 --> ExtractMeta[Extract metadata: model, times, tokens]
    ExtractMeta --> CreateChatResp[Create ChatResponse]
    CreateChatResp --> ReturnChat([Return ChatResponse])

    %% Complete Path
    CompletePath --> Decorator[@chat_to_completion_decorator]
    Decorator --> WrapPrompt[Wrap prompt in Message]
    WrapPrompt --> SetRole[role=USER, content=prompt]
    SetRole --> CallChat[Delegate to chat method]
    CallChat --> ChatPath
    ReturnChat --> UnwrapDecorator[Decorator unwraps response]
    UnwrapDecorator --> ExtractText[Extract message.content as text]
    ExtractText --> CreateCompleteResp[Create CompletionResponse]
    CreateCompleteResp --> ReturnComplete([Return CompletionResponse])

    %% Tools Path
    ToolsPath --> PrepareTools[_prepare_chat_with_tools]
    PrepareTools --> ConvertToolsLoop[For each tool in tools]
    ConvertToolsLoop --> ExtractToolMeta[Extract tool.metadata]
    ExtractToolMeta --> GetSchema[Get fn_schema from metadata]
    GetSchema --> BuildToolDict[Build Ollama tool dict]
    BuildToolDict --> AddToolType[Add type: function]
    AddToolType --> AddToolFunc[Add function: name, description, parameters]

    AddToolFunc --> MergeKwargs[Merge tools into kwargs]
    MergeKwargs --> CallChatWithTools[Call chat with tools kwarg]
    CallChatWithTools --> SendRequestTools[HTTP POST with tools array]
    SendRequestTools --> ReceiveToolResp[Receive response with tool_calls]
    ReceiveToolResp --> ValidateTools[_validate_chat_with_tools_response]
    ValidateTools --> CheckParallel{allow_parallel?}
    CheckParallel -->|No| ForceSingle[force_single_tool_call]
    CheckParallel -->|Yes| ReturnToolResp
    ForceSingle --> ReturnToolResp([Return ChatResponse with tools])

    %% Stream Path
    StreamPath --> BuildStreamReq[Build chat request with stream=True]
    BuildStreamReq --> SendStreamReq[HTTP POST /api/chat streaming]
    SendStreamReq --> StreamLoop{For each chunk}

    StreamLoop --> ReceiveChunk[Receive chunk dict]
    ReceiveChunk --> ParseChunk[_chat_stream_from_response]
    ParseChunk --> ExtractDelta[Extract message delta]
    ExtractDelta --> AccumContent[Accumulate content]
    AccumContent --> CheckToolChunk{tool_calls in chunk?}
    CheckToolChunk -->|Yes| AccumTools[Accumulate tool_calls]
    CheckToolChunk -->|No| CreateStreamResp
    AccumTools --> CreateStreamResp[Create ChatResponse with delta]
    CreateStreamResp --> YieldResp[Yield ChatResponse]
    YieldResp --> CheckDone{done=True?}
    CheckDone -->|No| StreamLoop
    CheckDone -->|Yes| EndStream([Stream Complete])

    %% Error paths
    Error1 --> ErrorEnd([Raise Exception])

    %% Styling
    style Start fill:#e1f5ff
    style Ready fill:#e1f5ff
    style ReturnChat fill:#c8e6c9
    style ReturnComplete fill:#c8e6c9
    style ReturnToolResp fill:#c8e6c9
    style EndStream fill:#c8e6c9
    style Error1 fill:#ffcdd2
    style ErrorEnd fill:#ffcdd2
    style SendRequest fill:#fff9c4
    style SendRequestTools fill:#fff9c4
    style SendStreamReq fill:#fff9c4