wehub-resource-sync
490 lines
11 KiB
Python
490 lines
11 KiB
Python
# 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
|