# This file was auto-generated by `flavors update`.
# Please do not edit it manually.

_ML_PACKAGE_VERSIONS = {
    "semantic_kernel": {
        "package_info": {
            "pip_release": "semantic-kernel"
        },
        "autologging": {
            "minimum": "1.34.0",
            "maximum": "1.43.0"
        }
    },
    "openai": {
        "package_info": {
            "pip_release": "openai"
        },
        "models": {
            "minimum": "1.87.0",
            "maximum": "2.41.0"
        },
        "autologging": {
            "minimum": "1.87.0",
            "maximum": "2.41.0"
        }
    },
    "dspy": {
        "package_info": {
            "pip_release": "dspy"
        },
        "models": {
            "minimum": "3.0.0",
            "maximum": "3.2.1"
        },
        "autologging": {
            "minimum": "3.0.0",
            "maximum": "3.2.1"
        }
    },
    "langchain": {
        "package_info": {
            "pip_release": "langchain"
        },
        "models": {
            "minimum": "0.3.26",
            "maximum": "1.3.4"
        },
        "autologging": {
            "minimum": "0.3.26",
            "maximum": "1.3.4"
        }
    },
    "langgraph": {
        "package_info": {
            "pip_release": "langgraph"
        },
        "models": {
            "minimum": "0.4.9",
            "maximum": "1.2.4"
        },
        "autologging": {
            "minimum": "0.4.9",
            "maximum": "1.2.4"
        }
    },
    "llama_index": {
        "package_info": {
            "pip_release": "llama-index",
            "module_name": "llama_index.core"
        },
        "models": {
            "minimum": "0.12.43",
            "maximum": "0.14.22"
        },
        "autologging": {
            "minimum": "0.12.43",
            "maximum": "0.14.22"
        }
    },
    "ag2": {
        "package_info": {
            "pip_release": "ag2",
            "module_name": "autogen"
        },
        "autologging": {
            "minimum": "0.9.3",
            "maximum": "0.13.3"
        }
    },
    "autogen": {
        "package_info": {
            "pip_release": "autogen-agentchat",
            "module_name": "autogen_agentchat"
        },
        "autologging": {
            "minimum": "0.6.2",
            "maximum": "0.7.5"
        }
    },
    "gemini": {
        "package_info": {
            "pip_release": "google-genai",
            "module_name": "google.genai"
        },
        "autologging": {
            "minimum": "1.21.0",
            "maximum": "2.8.0"
        }
    },
    "anthropic": {
        "package_info": {
            "pip_release": "anthropic"
        },
        "autologging": {
            "minimum": "0.55.0",
            "maximum": "0.107.1"
        }
    },
    "crewai": {
        "package_info": {
            "pip_release": "crewai",
            "module_name": "crewai"
        },
        "autologging": {
            "minimum": "0.134.0",
            "maximum": "1.14.6"
        }
    },
    "agno": {
        "package_info": {
            "pip_release": "agno",
            "module_name": "agno"
        },
        "autologging": {
            "minimum": "1.7.0",
            "maximum": "2.6.12"
        }
    },
    "pydantic_ai": {
        "package_info": {
            "pip_release": "pydantic-ai",
            "module_name": "pydantic_ai"
        },
        "autologging": {
            "minimum": "0.2.19",
            "maximum": "1.94.0"
        }
    },
    "smolagents": {
        "package_info": {
            "pip_release": "smolagents",
            "module_name": "smolagents"
        },
        "autologging": {
            "minimum": "1.19.0",
            "maximum": "1.26.0"
        }
    },
    "strands": {
        "package_info": {
            "pip_release": "strands-agents",
            "module_name": "strands"
        },
        "autologging": {
            "minimum": "1.4.0",
            "maximum": "1.42.0"
        }
    },
    "mistral": {
        "package_info": {
            "pip_release": "mistralai",
            "module_name": "mistralai"
        },
        "autologging": {
            "minimum": "1.9.1",
            "maximum": "2.4.9"
        }
    },
    "groq": {
        "package_info": {
            "pip_release": "groq"
        },
        "autologging": {
            "minimum": "0.29.0",
            "maximum": "1.4.0"
        }
    },
    "bedrock": {
        "package_info": {
            "pip_release": "boto3",
            "module_name": "boto3"
        },
        "autologging": {
            "minimum": "1.38.37",
            "maximum": "1.43.24"
        }
    },
    "sklearn": {
        "package_info": {
            "pip_release": "scikit-learn"
        },
        "models": {
            "minimum": "1.5.1",
            "maximum": "1.9.0"
        },
        "autologging": {
            "minimum": "1.5.1",
            "maximum": "1.9.0"
        }
    },
    "pytorch": {
        "package_info": {
            "pip_release": "torch",
            "module_name": "torch"
        },
        "models": {
            "minimum": "2.4.0",
            "maximum": "2.12.0"
        },
        "autologging": {
            "minimum": "2.4.0",
            "maximum": "2.12.0"
        }
    },
    "pytorch-lightning": {
        "package_info": {
            "pip_release": "pytorch-lightning",
            "module_name": "lightning"
        },
        "models": {
            "minimum": "2.3.1",
            "maximum": "2.6.5"
        },
        "autologging": {
            "minimum": "2.3.1",
            "maximum": "2.6.5"
        }
    },
    "keras": {
        "package_info": {
            "pip_release": "keras"
        },
        "models": {
            "minimum": "3.4.0",
            "maximum": "3.14.1"
        },
        "autologging": {
            "minimum": "3.4.0",
            "maximum": "3.14.1"
        }
    },
    "tensorflow": {
        "package_info": {
            "pip_release": "tensorflow"
        },
        "models": {
            "minimum": "2.16.2",
            "maximum": "2.21.0"
        },
        "autologging": {
            "minimum": "2.16.2",
            "maximum": "2.21.0"
        }
    },
    "xgboost": {
        "package_info": {
            "pip_release": "xgboost"
        },
        "models": {
            "minimum": "2.1.0",
            "maximum": "3.2.0"
        },
        "autologging": {
            "minimum": "2.1.0",
            "maximum": "3.2.0"
        }
    },
    "lightgbm": {
        "package_info": {
            "pip_release": "lightgbm"
        },
        "models": {
            "minimum": "4.5.0",
            "maximum": "4.6.0"
        },
        "autologging": {
            "minimum": "4.5.0",
            "maximum": "4.6.0"
        }
    },
    "catboost": {
        "package_info": {
            "pip_release": "catboost"
        },
        "models": {
            "minimum": "1.2.6",
            "maximum": "1.2.10"
        }
    },
    "onnx": {
        "package_info": {
            "pip_release": "onnx"
        },
        "models": {
            "minimum": "1.17.0",
            "maximum": "1.21.0"
        }
    },
    "spacy": {
        "package_info": {
            "pip_release": "spacy"
        },
        "models": {
            "minimum": "3.8.2",
            "maximum": "3.8.14"
        }
    },
    "statsmodels": {
        "package_info": {
            "pip_release": "statsmodels"
        },
        "models": {
            "minimum": "0.14.3",
            "maximum": "0.14.6"
        },
        "autologging": {
            "minimum": "0.14.3",
            "maximum": "0.14.6"
        }
    },
    "spark": {
        "package_info": {
            "pip_release": "pyspark",
            "module_name": "pyspark"
        },
        "models": {
            "minimum": "3.2.1",
            "maximum": "4.1.2"
        },
        "autologging": {
            "minimum": "3.3.0",
            "maximum": "4.1.2"
        }
    },
    "prophet": {
        "package_info": {
            "pip_release": "prophet"
        },
        "models": {
            "minimum": "1.1.6",
            "maximum": "1.3.0"
        }
    },
    "pmdarima": {
        "package_info": {
            "pip_release": "pmdarima"
        },
        "models": {
            "minimum": "2.1.0",
            "maximum": "2.1.1"
        }
    },
    "h2o": {
        "package_info": {
            "pip_release": "h2o"
        },
        "models": {
            "minimum": "3.46.0.4",
            "maximum": "3.46.0.11"
        }
    },
    "shap": {
        "package_info": {
            "pip_release": "shap"
        },
        "models": {
            "minimum": "0.46.0",
            "maximum": "0.52.0"
        }
    },
    "paddle": {
        "package_info": {
            "pip_release": "paddlepaddle"
        },
        "models": {
            "minimum": "2.6.2",
            "maximum": "3.3.1"
        },
        "autologging": {
            "minimum": "2.6.2",
            "maximum": "3.3.1"
        }
    },
    "transformers": {
        "package_info": {
            "pip_release": "transformers"
        },
        "models": {
            "minimum": "4.42.0",
            "maximum": "5.10.2"
        },
        "autologging": {
            "minimum": "4.42.0",
            "maximum": "5.10.2"
        }
    },
    "diffusers": {
        "package_info": {
            "pip_release": "diffusers"
        },
        "models": {
            "minimum": "0.37.0",
            "maximum": "0.38.0"
        }
    },
    "haystack": {
        "package_info": {
            "pip_release": "haystack-ai",
            "module_name": "haystack"
        },
        "autologging": {
            "minimum": "2.2.2",
            "maximum": "2.30.0"
        }
    },
    "sentence_transformers": {
        "package_info": {
            "pip_release": "sentence-transformers"
        },
        "models": {
            "minimum": "3.1.0",
            "maximum": "5.5.1"
        }
    },
    "johnsnowlabs": {
        "package_info": {
            "pip_release": "johnsnowlabs"
        },
        "models": {
            "minimum": "5.4.0",
            "maximum": "6.4.0"
        }
    }
}

# A mapping of flavor name to the module name to be imported for autologging.
# This is used for checking version compatibility in autologging.
# DO NOT EDIT MANUALLY

# GenAI packages
GENAI_FLAVOR_TO_MODULE_NAME = {
    "semantic_kernel": "semantic_kernel",
    "openai": "openai",
    "dspy": "dspy",
    "langchain": "langchain",
    "llama_index": "llama_index.core",
    "ag2": "autogen",
    "autogen": "autogen_agentchat",
    "gemini": "google.genai",
    "anthropic": "anthropic",
    "crewai": "crewai",
    "agno": "agno",
    "pydantic_ai": "pydantic_ai",
    "smolagents": "smolagents",
    "strands": "strands",
    "mistral": "mistralai",
    "groq": "groq",
    "bedrock": "boto3"
}

# Non-GenAI packages
NON_GENAI_FLAVOR_TO_MODULE_NAME = {
    "sklearn": "sklearn",
    "pytorch": "torch",
    "pytorch-lightning": "lightning",
    "keras": "keras",
    "tensorflow": "tensorflow",
    "xgboost": "xgboost",
    "lightgbm": "lightgbm",
    "statsmodels": "statsmodels",
    "spark": "pyspark",
    "paddle": "paddle",
    "transformers": "transformers",
    "haystack": "haystack",
    "pyspark.ml": "pyspark"
}

# Combined mapping for backward compatibility
FLAVOR_TO_MODULE_NAME = NON_GENAI_FLAVOR_TO_MODULE_NAME | GENAI_FLAVOR_TO_MODULE_NAME