It doesn't support Ollama but it does work with Ollama. I'm on MacOS and I use Podman.

#!/usr/bin/env python
import autogen
import os
import sys
import logging
import requests
import subprocess

# Set up logging
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)

# Set Docker host to use Podman socket
os.environ["DOCKER_HOST"] = "unix:///var/run/docker.sock"

BASE_URL = "http://localhost:11434/v1"

# Configuration for Ollama models
config_list = [
    {
        "base_url": BASE_URL,
        "api_key": "fakekey",
        "model": "qwen2.5:32b-instruct",
    }
]

llm_config = {
    "config_list": config_list,
}


# Function to validate Ollama server
def validate_ollama_server():
    try:
        response = requests.get(f"{BASE_URL}/models")
        response.raise_for_status()
        logger.info("Ollama server is running and accessible.")
    except requests.RequestException as e:
        logger.error(f"Failed to connect to Ollama server: {e}")
        sys.exit(1)


# Function to pull Python image
def pull_python_image():
    try:
        subprocess.run(["podman", "pull", "python:3-slim"], check=True)
        logger.info("Python image pulled successfully.")
    except subprocess.CalledProcessError as e:
        logger.error(f"Failed to pull Python image: {e}")
        sys.exit(1)


# Set up agents
assistant = autogen.AssistantAgent(name="assistant", llm_config=llm_config)
coder = autogen.AssistantAgent(name="coder", llm_config=llm_config)

# Code execution configuration
code_execution_config = {
    "work_dir": "coding",
    "use_docker": True,
}

# Set up user proxy agent
user_proxy = autogen.UserProxyAgent(
    name="user_proxy",
    human_input_mode="TERMINATE",
    max_consecutive_auto_reply=10,
    is_termination_msg=lambda x: x.get("content", "").rstrip().endswith("TERMINATE"),
    code_execution_config=code_execution_config,
)

# Main execution
if __name__ == "__main__":
    # Validate Ollama server
    validate_ollama_server()

    # Pull Python image
    pull_python_image()

    # Initiate chat
    logger.info("Initiating chat with assistant...")
    try:
        user_proxy.initiate_chat(
            assistant,
            message="Write a Python function to the answer to life the universe and everything.",
        )
    except Exception as e:
        logger.error(f"An error occurred during chat: {e}")
        sys.exit(1)

    logger.info("Chat completed successfully.")