genesis-flow 1.0.1__py3-none-any.whl → 1.0.3__py3-none-any.whl

{genesis_flow-1.0.1.dist-info → genesis_flow-1.0.3.dist-info}/METADATA

@@ -1,7 +1,7 @@
 Metadata-Version: 2.4
 Name: genesis-flow
-Version: 1.0.1
-Summary: Genesis-Flow: Lightweight, secure MLflow fork for Genesis platform
+Version: 1.0.3
+Summary: Genesis-Flow: MLflow v3.1.4 compatible fork for Genesis platform
 Maintainer-email: Databricks <mlflow-oss-maintainers@googlegroups.com>
 License: Copyright 2018 Databricks, Inc.  All rights reserved.
         
@@ -268,6 +268,36 @@ Requires-Dist: databricks-sdk<1,>=0.20.0; extra == "databricks"
 Requires-Dist: azure-storage-file-datalake>12; extra == "databricks"
 Requires-Dist: boto3>1; extra == "databricks"
 Requires-Dist: botocore; extra == "databricks"
+Requires-Dist: databricks-agents<2.0,>=1.0.0; extra == "databricks"
+Provides-Extra: mlserver
+Requires-Dist: mlserver!=1.3.1,<2.0.0,>=1.2.0; extra == "mlserver"
+Requires-Dist: mlserver-mlflow!=1.3.1,<2.0.0,>=1.2.0; extra == "mlserver"
+Provides-Extra: gateway
+Requires-Dist: fastapi<1; extra == "gateway"
+Requires-Dist: uvicorn[standard]<1; extra == "gateway"
+Requires-Dist: watchfiles<2; extra == "gateway"
+Requires-Dist: aiohttp<4; extra == "gateway"
+Requires-Dist: boto3<2,>=1.28.56; extra == "gateway"
+Requires-Dist: tiktoken<1; extra == "gateway"
+Requires-Dist: slowapi<1,>=0.1.9; extra == "gateway"
+Provides-Extra: genai
+Requires-Dist: fastapi<1; extra == "genai"
+Requires-Dist: uvicorn[standard]<1; extra == "genai"
+Requires-Dist: watchfiles<2; extra == "genai"
+Requires-Dist: aiohttp<4; extra == "genai"
+Requires-Dist: boto3<2,>=1.28.56; extra == "genai"
+Requires-Dist: tiktoken<1; extra == "genai"
+Requires-Dist: slowapi<1,>=0.1.9; extra == "genai"
+Provides-Extra: sqlserver
+Requires-Dist: mlflow-dbstore; extra == "sqlserver"
+Provides-Extra: aliyun-oss
+Requires-Dist: aliyunstoreplugin; extra == "aliyun-oss"
+Provides-Extra: xethub
+Requires-Dist: mlflow-xethub; extra == "xethub"
+Provides-Extra: jfrog
+Requires-Dist: mlflow-jfrog-plugin; extra == "jfrog"
+Provides-Extra: langchain
+Requires-Dist: langchain<=0.3.25,>=0.1.0; extra == "langchain"
 Provides-Extra: auth
 Requires-Dist: Flask-WTF<2; extra == "auth"
 Provides-Extra: extras
@@ -761,32 +791,6 @@ pre-commit install
 
 # Run tests
 pytest tests/
-
-# 1. Install build tools
-  pip install build twine
-
-  # 2. Build the package
-  python -m build
-
-  # 3. Upload to PyPI
-  python -m twine upload dist/*
-
-  # Or upload to TestPyPI first
-  python -m twine upload --repository testpypi dist/*
-
-  Before uploading, make sure you have:
-  - A PyPI account and API token
-  - Configure your credentials in ~/.pypirc:
-
-  [pypi]
-  username = __token__
-  password = pypi-<your-token>
-
-  [testpypi]
-  repository = https://test.pypi.org/legacy/
-  username = __token__
-  password = pypi-<your-token>
-
 ```
 
 ### Code Quality

{genesis_flow-1.0.1.dist-info → genesis_flow-1.0.3.dist-info}/RECORD

@@ -1,4 +1,4 @@
-genesis_flow-1.0.1.dist-info/licenses/LICENSE.txt,sha256=Y5U1Xebzka__NZlqMPtBsYm0mRpMtUmTrONatpoL-ig,11382
+genesis_flow-1.0.3.dist-info/licenses/LICENSE.txt,sha256=Y5U1Xebzka__NZlqMPtBsYm0mRpMtUmTrONatpoL-ig,11382
 mlflow/__init__.py,sha256=-_r__N5Afed81pLVtr2wKbHQIA0aj9u9n_7kWGxLWi4,11194
 mlflow/__main__.py,sha256=_PcdoxKehR_a2MI6GqBfzYzRCXZhVyDCSdbxDWVlWd4,39
 mlflow/cli.py,sha256=f1ObrWZ03HgRiRoVEE1Gffe-dGcSY7CxJyEFgb5VUMM,26137
@@ -11,7 +11,7 @@ mlflow/mismatch.py,sha256=G8YMN05L2x3k0NZpRtVNygEcGUXxc3WBzR-VdJw2iao,1050
 mlflow/ml_package_versions.py,sha256=jCYVUZQ82hMWeskEjGDBk-EX-EnThuMM_qBhrI7SsuI,10533
 mlflow/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 mlflow/runs.py,sha256=qd8EtlSKj9499tSpzhtXp9Vm1iO5KK8P85KwZShyGH0,2519
-mlflow/version.py,sha256=ZRrRr7fEH9NnTqdz8RHUdIATR7mSDWNjh-8RWVilGDs,744
+mlflow/version.py,sha256=Wrx1SBDS4H0U3Z7Yc728mt1X9vzv3FeCnpiA0ABdGu8,755
 mlflow/ag2/__init__.py,sha256=M298Ynewc2rbH6qNq_eGzuameavonOBLPy6ypgNdbQY,2225
 mlflow/ag2/ag2_logger.py,sha256=vnKExbqJerKBrcHyAatjPYe9buIjndu2530J4bhSo_8,11357
 mlflow/anthropic/__init__.py,sha256=qCo8IhTJHUomooKQrLKbYv-I8hHHS1iziTDanwS4wE4,1306
@@ -43,7 +43,7 @@ mlflow/data/code_dataset_source.py,sha256=YtetLm5O-U-8AeRzt_CUcqfDvmyNOQZ0FlDG-k
 mlflow/data/dataset.py,sha256=ZcFl9TQVVzv0hJH_xn44qVC2H6WlcmoWD7_bDr6WUho,4119
 mlflow/data/dataset_registry.py,sha256=jm4L9P6GnKJ728O_cp6vbNewv5Kl-rp-Z8dXyKR0bw8,6491
 mlflow/data/dataset_source.py,sha256=5ta8hL8Wnalsmf0-qU65SB0apyFq8TItviOCfs5cwKY,3550
-mlflow/data/dataset_source_registry.py,sha256=QxQ1PiMvHqvrkDsccTydBYFCslYFuGSURLH0U_uitbQ,8342
+mlflow/data/dataset_source_registry.py,sha256=GsWoxTbQ13YdPgMkd642mpoqAwKNuf6qGZ7fWA4ysHg,8576
 mlflow/data/delta_dataset_source.py,sha256=1chBnweHmtRkWPzi1_2itPFzDguYYaGVOnTiGww5Wu8,6027
 mlflow/data/digest_utils.py,sha256=O4NP6z2m604uPqbTuk-1ksCUhTJCNMOQb419FM4mzeY,2937
 mlflow/data/evaluation_dataset.py,sha256=FygCtfd_Ccfq2ohHRlq1lghdRGiy-YzF-0bMea3Smuk,20707
@@ -135,13 +135,15 @@ mlflow/evaluation/evaluation.py,sha256=YAO2LRKEt74W29cIZ2tSaEs_qymOpt1MkjrNCHuth
 mlflow/evaluation/evaluation_tag.py,sha256=Ynl0fxz49lFPTv7kIvi4CePKlg3yXG2Ci6nT9PmtUNI,1682
 mlflow/evaluation/fluent.py,sha256=Qj67ni9dupg40XsPDb9q3FduxlFWfITYayPBt5CExaI,1849
 mlflow/evaluation/utils.py,sha256=wrb1CtJhjVJB1ITRTlav2WisfqO8gPU8_nzx0CTQ8zo,6375
+mlflow/gateway/providers/bedrock.py,sha256=UVsTKNkSaJsCTjPYQmsSNwG1CPkM4trTwZIsF-AINZw,10475
 mlflow/gemini/__init__.py,sha256=-mPzQ9hFBLsNMkP3RjOp7cQAzyhHl2hcqVNYle46kKM,2481
 mlflow/gemini/autolog.py,sha256=DPFsW22c8yb8Z_Iv5wuTleyOrnfZQbmMWR34FN6PzcE,6585
 mlflow/gemini/chat.py,sha256=BqPjA2BAplw5JSqKJ87pGZC0Rasij28Ae57uGU6YnGI,9475
 mlflow/genai/__init__.py,sha256=Lc_eDeUQpIlGGGZbCs_Kj3dpn9NHm6QydFV3XsFJwE4,1612
 mlflow/genai/scheduled_scorers.py,sha256=lLd9LVTeC3AfUqVo_52t_8FIXHlI4ubd1q-qjTtLE00,18097
 mlflow/genai/datasets/__init__.py,sha256=UXgb_Ces-hYVvv7WwWw42Kbz3_RN9cW8GUUMiTwpq9A,2095
-mlflow/genai/datasets/evaluation_dataset.py,sha256=w1Bfp0VxMZpAHZoh_RsQA79PI79oeY9OznU7Dfgqh2I,4705
+mlflow/genai/datasets/databricks_evaluation_dataset_source.py,sha256=27P-CGE1lEtsRAJisaywliGWAzq9BBGjUhN3QpXe-xI,2405
+mlflow/genai/datasets/evaluation_dataset.py,sha256=1EzhmMupmlRkyjVuOPIE_axJrFAAVhPT_U2WEfGfnd0,4772
 mlflow/genai/evaluation/__init__.py,sha256=aHMqUtH-4Bvgae2wlQMXL5cLYjwtI3U7KhTZ_v_dGHE,106
 mlflow/genai/evaluation/base.py,sha256=Eu8yqoEapy3qxAe3AzCBREMKsH7RU9B0UeSMEXpjChA,15974
 mlflow/genai/evaluation/constant.py,sha256=CKhzmcf67O9ZJjrEIIKUp74-Lc3IlXNnBcorGcCotCc,736
@@ -164,7 +166,7 @@ mlflow/genai/optimize/optimizers/utils/dspy_mipro_callback.py,sha256=RsRnAuEK5_5
 mlflow/genai/optimize/optimizers/utils/dspy_mipro_utils.py,sha256=VHzDqQU5VUndGBCh1kA9cZpZpwLmkkqrnna5QCeUqD0,500
 mlflow/genai/prompts/__init__.py,sha256=EGzQX6JVnxX9Ie8gqJnI9CFjQJYHJp1ltviQhM9vO44,7165
 mlflow/genai/scorers/__init__.py,sha256=lU2WYlye6rYi3xMV_5V204QeZv7htvJmAt_Khcq2EGI,553
-mlflow/genai/scorers/base.py,sha256=FkzT7wLZzQOZMDCCGgtX__DH8B_zaQKhiHhIL6eisfk,19861
+mlflow/genai/scorers/base.py,sha256=NxsOIigau84kUORAQ7ebv9WnIK5kARsSa4WKdW5TyxI,20208
 mlflow/genai/scorers/builtin_scorers.py,sha256=WB3P58wbPk8XM2UHwUxw2tXKWllv5Nr5KdUxiKENn-o,27727
 mlflow/genai/scorers/scorer_utils.py,sha256=NTUUVkikAIQO52wOLwwSCsM8JBiuz1ah7K7sbJmjUEE,4849
 mlflow/genai/scorers/validation.py,sha256=28tIEy2j2NxUx-ooWR2Go4_gk3akGVB3hq1RTaOcaf0,6390
@@ -186,7 +188,7 @@ mlflow/langchain/model.py,sha256=RF8Vy61hVem4-ULsHRVfEzzLV6BFgcsNnEw1uMaO6G8,391
 mlflow/langchain/output_parsers.py,sha256=U7sYzic0ZBCoU-NCrMZWAePprn5PG_ZldfQEvRScyrQ,4957
 mlflow/langchain/retriever_chain.py,sha256=jjLhPLNn7kSEJNTCGKNDyF7HbiVAQ3TY6fbunjJYc_Q,5272
 mlflow/langchain/runnables.py,sha256=Hn2dcDtFAykLdXB6Q-6wRoqLIg1rO3ztBwhu2_iu3wQ,19770
-mlflow/langchain/utils/chat.py,sha256=HUzYQ5maPaARIVNqhDoWhR8vNDwCFpg-H407GCGGESA,14195
+mlflow/langchain/utils/chat.py,sha256=1UfQ0JazRwxFQUYbzsliCWshIO8P-09_R7i95BuE8Ss,14563
 mlflow/langchain/utils/logging.py,sha256=nzGjLVr-0JgRnVNTpmwz0oNGS4E5_GtF2sULxndOX3U,23411
 mlflow/langchain/utils/serialization.py,sha256=VZbuOS2USnzBIR3Mvm4xsL7v-g1L9W9PftEBn6pBja4,1233
 mlflow/legacy_databricks_cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -232,7 +234,7 @@ mlflow/models/resources.py,sha256=blqbEthnPNqFwUdtRYdyDOZ2nR6Q-3R6Jwf9eotegQw,11
 mlflow/models/signature.py,sha256=A-odLBU1TI1u-1GQrru4pJ0HFIGiCdmad3_Y9vCdz7g,25978
 mlflow/models/utils.py,sha256=7pr3WX8uNvaX36GXhM32DJjk-2-q6_7vDbE5jzNNYpQ,82897
 mlflow/models/wheeled_model.py,sha256=U8zqEZtnFjwNuji_6DBCaL3sOF3Hczm_PJSUGwGOGI0,11886
-mlflow/models/container/__init__.py,sha256=P-uQFU5TQb3PZONbr_gGuweGwcXFBbav2j2Kllem56Y,10636
+mlflow/models/container/__init__.py,sha256=Y1HlcvGwOkYTQbjIMIJLIUd9gtR6Q5jSbhoiYMgoWLw,10636
 mlflow/models/container/scoring_server/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 mlflow/models/container/scoring_server/nginx.conf,sha256=XwBf_NWOylw3MulHAZgyVMjorgoLt0EkRtf4lPJYj80,710
 mlflow/models/evaluation/__init__.py,sha256=MbagaX5WlRiWeysM1RO0q0SEe0nRhBZG2tyc7vo1VP8,528
@@ -367,6 +369,7 @@ mlflow/sklearn/utils.py,sha256=CM9z-UD65RBN4sIdEoJWhazkdmE9Wa9InjsU61y1XfE,38912
 mlflow/smolagents/__init__.py,sha256=V2sfkuWxQ090CaIjwBh7icD0bSgYP3jKD5MRzy__uW0,2202
 mlflow/smolagents/autolog.py,sha256=5GJpqmXs8dd2tf4jkL6fk7Fxb0TsxAMykB865M-BpK8,4770
 mlflow/smolagents/chat.py,sha256=tsrS9dm-iIigSbZQ3A6fA_vUXlnr77hMAI65PNY9Ono,980
+mlflow/spark/__init__.py,sha256=-h5YdE-BVQPW9ke9Lfu33W9qlNqrn_FXPHkNsVVL8ic,54655
 mlflow/store/__init__.py,sha256=9uQ8sjjcbpTkeNTV1RjPFLlW8b6oDaeae1nDwbLVtNs,281
 mlflow/store/_unity_catalog/__init__.py,sha256=idiiLNYc7lzfOaPwY_QwTfvH1A6XMM0pkPzi-voVUsw,63
 mlflow/store/_unity_catalog/lineage/__init__.py,sha256=X-C_OnRoaueY0AoIty-WokUpnoRgoDGxhZBkDzYkquo,62
@@ -379,8 +382,8 @@ mlflow/store/_unity_catalog/registry/utils.py,sha256=aOfKcdh8liGODGzRTMs6R_ddItU
 mlflow/store/artifact/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 mlflow/store/artifact/artifact_repo.py,sha256=2FBg7jEIpXLYF8vGDwyuV6VyyHJsDZHD4P2Hzi8KH8Y,18654
 mlflow/store/artifact/artifact_repository_registry.py,sha256=dMUX-uZLlNDFRaASHnMnEcHQpk5UvtSWSBNRQId6Ldk,7099
-mlflow/store/artifact/azure_blob_artifact_repo.py,sha256=Ud-ccqA1wTUdn-q5y-mGg_Xd2lYu9hRafKLpfPOJCJU,12417
-mlflow/store/artifact/azure_data_lake_artifact_repo.py,sha256=pwrm5iLeIe1uA6jjB-6ZKsob4Oj07PpHoMTezK5a4Ag,12138
+mlflow/store/artifact/azure_blob_artifact_repo.py,sha256=kB-GgvyaKIHKL4CirmdRzw3tX-uc5OYoNbsxUtSeyO4,12417
+mlflow/store/artifact/azure_data_lake_artifact_repo.py,sha256=9jG3rYUR4FSOqpLlo7xcBMKha3pTlTsvt_lDNUUREg0,12138
 mlflow/store/artifact/cli.py,sha256=TsUlxBTxx4Iyudyd-zZmhbsHL9zVyjAyJMnqqYzss-U,5375
 mlflow/store/artifact/cloud_artifact_repo.py,sha256=qQRuWgrKU7USTqoG7Urf6g4OEaJaNraEuwFrt_tjfPM,13332
 mlflow/store/artifact/databricks_artifact_repo.py,sha256=tGlaVlGcBvppDIGr-no1X7kSdzgcwtmiOJcO8jFP9t0,30377
@@ -391,17 +394,17 @@ mlflow/store/artifact/databricks_sdk_artifact_repo.py,sha256=UytO9rYFuQk1c_nWBFE
 mlflow/store/artifact/databricks_sdk_models_artifact_repo.py,sha256=izMSMpimTMFUWzVlmrWxWWs74B0kfeCYurT_fwNhP_4,3458
 mlflow/store/artifact/dbfs_artifact_repo.py,sha256=Z449LgfNJ1UverObdceuZF6FIBUOIG1Lb9545u6W1W4,10758
 mlflow/store/artifact/ftp_artifact_repo.py,sha256=A5aCDhRpdpbbxXGkFTHJgYpCMS7R7KDJ6CWxRX0jYhI,5291
-mlflow/store/artifact/gcs_artifact_repo.py,sha256=F77lALp9SDh3X2NwcKhEWYoGyQ2o9m1bVnMySTmu-QM,12098
+mlflow/store/artifact/gcs_artifact_repo.py,sha256=v_msi2tLs76raxOGoBnEUsU0uf5BbgFEXr9ifY3Z1Mc,12098
 mlflow/store/artifact/hdfs_artifact_repo.py,sha256=1YFjdF51av_Zm0B_unfSlOme15hsULWjL9bvtou5nGg,7956
 mlflow/store/artifact/http_artifact_repo.py,sha256=FSQ-IAoiMVc0K3mgHMz6ALVgqFRXfXXU43IBKxAdps4,9088
-mlflow/store/artifact/local_artifact_repo.py,sha256=LPVNoCyS-QHdCY4F0RsuGPsCD7hClv2ZYNJ-azaGoIY,6083
+mlflow/store/artifact/local_artifact_repo.py,sha256=u8GNoZZhtau2khzgjNKOLKRtdnDLl7dsvguZPTK8Qiw,6126
 mlflow/store/artifact/mlflow_artifacts_repo.py,sha256=vHF4f-IB4xB7arcGHZXy5C4SqUrjzNNX30OLH0bQwvs,3647
 mlflow/store/artifact/models_artifact_repo.py,sha256=TXyj7uMBk_e8IMlu79xaQo3jttU_FME8kXbeGW2jfvc,11093
 mlflow/store/artifact/optimized_s3_artifact_repo.py,sha256=QMNvnvcPmYwSM_PzM1KWmvuElQRwqg-nV1QFaErGs2g,15510
 mlflow/store/artifact/presigned_url_artifact_repo.py,sha256=p9I3V3hGn73Em6hRI6PflN9DWVDWAeL-X7CDa-XIfYg,7378
 mlflow/store/artifact/r2_artifact_repo.py,sha256=IQNMA5MUffsMOSxYyiF16lGnxWviU7br__utXjCmuJ8,2713
 mlflow/store/artifact/runs_artifact_repo.py,sha256=GPhn92ODs-POctsY5jxh5A18V5FXbldMEHxzWxN2NbE,11450
-mlflow/store/artifact/s3_artifact_repo.py,sha256=F7pWPTu-cm0c1v0-Id7WMLhnuYLZdata6tOR9soPwWE,12878
+mlflow/store/artifact/s3_artifact_repo.py,sha256=_4cyQdvY15LQWkcWDiaLZNO2thJK3QJWyoXhACoch6c,21253
 mlflow/store/artifact/sftp_artifact_repo.py,sha256=1Sn0w_VNEepHka1YpRao7Cq4u-SRKxfu7AZ3jv8PK-U,5508
 mlflow/store/artifact/uc_volume_artifact_repo.py,sha256=6gCjr_usIAXBi-Cd6-TF1XqMJEEnW9mnu4IXJXGeXWA,3457
 mlflow/store/artifact/unity_catalog_models_artifact_repo.py,sha256=BGk5QEYwoNWSSGa3dXNnu7lqKz9LBy1LOEzeopDKL5Y,7534
@@ -479,18 +482,18 @@ mlflow/system_metrics/metrics/network_monitor.py,sha256=przQFWg7P8iY7pZHpihj97Ip
 mlflow/system_metrics/metrics/rocm_monitor.py,sha256=n-UvfDCLjfXKrwBLqVr2WR-BCbqaLsspZAKjpPIm6Pw,4426
 mlflow/tracing/__init__.py,sha256=b_ZzQtaP5ctjnIeIFUG7OJ1iK7_ULePJOgFK5A4aKTk,499
 mlflow/tracing/assessment.py,sha256=USb21LblbH4p2bqe6KnNWHtaBxINXOgQuBNDrI2hf74,12236
-mlflow/tracing/client.py,sha256=6-dSnFNePnEAjvwN1meciku8e6tAJnalX3MWpBo3-Ns,21610
+mlflow/tracing/client.py,sha256=vrwje1RuXifyM8fbFwqloA_3_zpSlLPLY-pLK9CBMEw,25075
 mlflow/tracing/config.py,sha256=JNuIFBdetSctdN0zLRqDosAR-g86Ux5P2MIQYjSkx8g,4507
 mlflow/tracing/constant.py,sha256=_RnKWkv0RBr-ik3GHJpX7GxN0pPcz-HC1iuednnVPeo,4120
 mlflow/tracing/destination.py,sha256=ZtHYxGx9oLzSdpLqDpMYmpL9tClcSqVAD8fgYII5kto,2757
 mlflow/tracing/fluent.py,sha256=HNA-cmm7WeDmLUxOBNXxb9Bcqt7jwoq5W48YanU785k,53397
-mlflow/tracing/provider.py,sha256=db0_8yu3YoBTMZiyFbUbVBdQrTX2pGtuvA3tAWmNcGc,16677
+mlflow/tracing/provider.py,sha256=L-rLpRB4YZVrjr_u8Y57iIk_g1b-RWR-Y6KHWAy0Zuk,16977
 mlflow/tracing/trace_manager.py,sha256=sjMg-V4kaON-xz5jl5cPF53H4ZNMqW1QSMz0_1WI2XM,7281
 mlflow/tracing/display/__init__.py,sha256=XoiwHbWNbQVuDklpcm-X3cG8pgdg8vRZYtjYC2os4m0,1265
 mlflow/tracing/display/display_handler.py,sha256=ui39ZbJ5PthiEYfDTYlPP44UpWaMmL5GhcBsqMcjuDA,6282
 mlflow/tracing/export/async_export_queue.py,sha256=2Slz5Ab_6FM8GcJRhZ1G7jqRGnRlIGYziGzWDT0NOcU,6766
 mlflow/tracing/export/inference_table.py,sha256=nI2MjgaJQbKhwFKtMGnpcupPLWLAuCJWilY5j-_Asq0,6044
-mlflow/tracing/export/mlflow_v3.py,sha256=PaRtWX5BqCoxsQTcFK2lIwYbwFr4J_ynzSnBHSCmUm4,5898
+mlflow/tracing/export/mlflow_v3.py,sha256=8LaWwy2vc8Hxu1ttECJJvFhAB4TceBOg7XFsWiXTYvQ,5797
 mlflow/tracing/export/utils.py,sha256=C9xB1wAqopblJ1MSoUZVHHYnS2o_MhACM_j3Qg01ZZs,2201
 mlflow/tracing/processor/base_mlflow.py,sha256=vPpcFaQQOsA_9FUKqexPDKRCQluGsv9sQHSKXZLdkiQ,8389
 mlflow/tracing/processor/inference_table.py,sha256=VmzCWAOQyFHrWmZVfxO--CsYg7czaAsW205mWD5wuCw,7505
@@ -517,7 +520,7 @@ mlflow/tracking/metric_value_conversion_utils.py,sha256=AXxTd9Cot55d9bs4LUQvoeiF
 mlflow/tracking/multimedia.py,sha256=vm11i1lxw-gXh8j5kxmdXOFSwQrj8JmdV0AjPiw79Y4,6279
 mlflow/tracking/registry.py,sha256=I1Dzf3mXyhN0CDcOD3J_Y8bHuWv9fXmLsjABWPQhuNk,3524
 mlflow/tracking/_model_registry/__init__.py,sha256=_HtGwD9WY1x2-02TgEftcWrKKqBnJNjB9YdRc6sqLe4,41
-mlflow/tracking/_model_registry/client.py,sha256=MeyyyxCndQNRdRaAj-AKsp5mny6kE10vkB2OT9Z9r2M,27122
+mlflow/tracking/_model_registry/client.py,sha256=6URbRwOCiNSKl6_LO1so6tStMTOHCTVhvYEX77xfiOA,27371
 mlflow/tracking/_model_registry/fluent.py,sha256=DYXqDICZvjpiSLxcb2LqOVp165phGHr__nOHQ3kCA6M,32343
 mlflow/tracking/_model_registry/registry.py,sha256=szFyA0NMxtsD-aAGVDmag0yxXIhMUeUttf_DcEGmuu4,3251
 mlflow/tracking/_model_registry/utils.py,sha256=0oyWU_iKleLVeEaWLb-glhkJlArKX0U3kj5iVWv3zdI,9377
@@ -587,7 +590,7 @@ mlflow/utils/env_manager.py,sha256=iLX1KxlTmoslXixTTKO896_SOrGEpiDIwMYcbthnb0E,4
 mlflow/utils/env_pack.py,sha256=QnuDhCjcqsXsYRZFPn1l0S8zHMs2wxbSSlkbzg-P7WY,4972
 mlflow/utils/environment.py,sha256=yODdh3keEvsl0l8cgm5r6pLcnDKOpWqJJXK7RhRGSeo,37390
 mlflow/utils/exception_utils.py,sha256=BHedTMmi3NvVbHzppKXmPubxLQq6OS7Q44fnpnT4bCw,365
-mlflow/utils/file_utils.py,sha256=1mzHEmVx7tWH0TVgyTuFWzfWpfzrevEnW4M1ZgVPDoo,31465
+mlflow/utils/file_utils.py,sha256=joq-v2D5ffSV762mPbI__fABwQrIy0ELQ3z7AzkZTVo,31518
 mlflow/utils/git_utils.py,sha256=_uKXn2h58Xg2KvnanOf0eoZd23muWY__Lnl3nYmyblQ,2361
 mlflow/utils/gorilla.py,sha256=tAAbZgNVoCQwdGfPli-8yuK-fL1liNjjuMp-QuW6SwI,24049
 mlflow/utils/lazy_load.py,sha256=VJF0HV1fDEjenu8BFHOWfmDgJWME9GI8Fc9NG1VsN2o,1726
@@ -607,7 +610,7 @@ mlflow/utils/proto_json_utils.py,sha256=mku-o9v-uk_KVOAY6HFTVLDuAWtpWAg5NY_E1lMB
 mlflow/utils/pydantic_utils.py,sha256=-wsOgoARRTCYJEaTzEH0LVMiJX6ichzMHE2s4DiEJgw,1798
 mlflow/utils/request_utils.py,sha256=ZMctmkm0SuucaMwu_AO36v09VjY3Ql97WXPM5o4gl4w,10127
 mlflow/utils/requirements_utils.py,sha256=79BL9iFqg9H4VRYiqk-gTLlrfLyUY7KVG4BDV8NRZyo,26669
-mlflow/utils/rest_utils.py,sha256=yLeFOP135qV2OFcvtb_fFtj18MATRoxV4wJ5UshnEyk,26540
+mlflow/utils/rest_utils.py,sha256=R5MZk0Ck358VeW-DKqpfSyDHbWjg_slQTpRfr4ZLDLM,26664
 mlflow/utils/search_logged_model_utils.py,sha256=JWczQxogo4u9Gr3gfoSVf7tVny1e-rxfAchh3Vqy6RU,4289
 mlflow/utils/search_utils.py,sha256=PgXtwlViHvmahY63otykh5iJVW0PR9DIwetrvCCBw2U,86142
 mlflow/utils/secure_loading.py,sha256=yJpKwrhzINmo0yVdp6fXvbNkZu_HEdwyB8deljGDXxk,10095
@@ -638,8 +641,8 @@ mlflow/utils/autologging_utils/metrics_queue.py,sha256=bwpMX7Go6xFxrpYROi6rDBdVt
 mlflow/utils/autologging_utils/safety.py,sha256=IwbTbusyE87Hc4qkhhvMikoaZqX9kpr972FWS2B8goc,51465
 mlflow/utils/autologging_utils/versioning.py,sha256=2hSN4KXFWEJCcopDdLG6BiPeqSoqjETeNMuUBsCCwlI,3762
 mlflow/utils/import_hooks/__init__.py,sha256=werje98Woelkbwrhtlb8wmRdt3RtiL--LqGru7Xh3YU,13589
-genesis_flow-1.0.1.dist-info/METADATA,sha256=QOs7W_wCIn4bYsjQVwLX8JE9zoW1dBO4rEJsf_qQmzU,32586
-genesis_flow-1.0.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-genesis_flow-1.0.1.dist-info/entry_points.txt,sha256=fwlE6mYWea9uCNIsmXdT5qMIOZ8rYbTvO6XcZR8fBbE,495
-genesis_flow-1.0.1.dist-info/top_level.txt,sha256=wm8UqYyUHI21EvrTDHb3eYICy0dOVDLBhAL-jp5zbuI,7
-genesis_flow-1.0.1.dist-info/RECORD,,
+genesis_flow-1.0.3.dist-info/METADATA,sha256=aNXjmAeUTG0TkJzMp0BF4CFNJgcpMw8yuMeFQ2Eyh90,33401
+genesis_flow-1.0.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+genesis_flow-1.0.3.dist-info/entry_points.txt,sha256=fwlE6mYWea9uCNIsmXdT5qMIOZ8rYbTvO6XcZR8fBbE,495
+genesis_flow-1.0.3.dist-info/top_level.txt,sha256=wm8UqYyUHI21EvrTDHb3eYICy0dOVDLBhAL-jp5zbuI,7
+genesis_flow-1.0.3.dist-info/RECORD,,

mlflow/data/dataset_source_registry.py

@@ -217,3 +217,11 @@ try:
     _dataset_source_registry.register(UCVolumeDatasetSource)
 except ImportError:
     pass
+try:
+    from mlflow.genai.datasets.databricks_evaluation_dataset_source import (
+        DatabricksEvaluationDatasetSource,
+    )
+
+    _dataset_source_registry.register(DatabricksEvaluationDatasetSource)
+except ImportError:
+    pass

mlflow/gateway/providers/bedrock.py

@@ -0,0 +1,298 @@
+import json
+import time
+from enum import Enum
+
+from mlflow.gateway.config import AmazonBedrockConfig, AWSIdAndKey, AWSRole, RouteConfig
+from mlflow.gateway.constants import (
+    MLFLOW_AI_GATEWAY_ANTHROPIC_DEFAULT_MAX_TOKENS,
+)
+from mlflow.gateway.exceptions import AIGatewayConfigException, AIGatewayException
+from mlflow.gateway.providers.anthropic import AnthropicAdapter
+from mlflow.gateway.providers.base import BaseProvider, ProviderAdapter
+from mlflow.gateway.providers.cohere import CohereAdapter
+from mlflow.gateway.providers.utils import rename_payload_keys
+from mlflow.gateway.schemas import completions
+
+AWS_BEDROCK_ANTHROPIC_MAXIMUM_MAX_TOKENS = 8191
+
+
+class AmazonBedrockAnthropicAdapter(AnthropicAdapter):
+    @classmethod
+    def chat_to_model(cls, payload, config):
+        payload = super().chat_to_model(payload, config)
+        # "model" keys are not supported in Bedrock"
+        payload.pop("model", None)
+        return payload
+
+    @classmethod
+    def completions_to_model(cls, payload, config):
+        payload = super().completions_to_model(payload, config)
+
+        if "\n\nHuman:" not in payload.get("stop_sequences", []):
+            payload.setdefault("stop_sequences", []).append("\n\nHuman:")
+
+        payload["max_tokens_to_sample"] = min(
+            payload.get("max_tokens_to_sample", MLFLOW_AI_GATEWAY_ANTHROPIC_DEFAULT_MAX_TOKENS),
+            AWS_BEDROCK_ANTHROPIC_MAXIMUM_MAX_TOKENS,
+        )
+
+        # "model" keys are not supported in Bedrock"
+        payload.pop("model", None)
+        return payload
+
+    @classmethod
+    def model_to_completions(cls, payload, config):
+        payload["model"] = config.model.name
+        return super().model_to_completions(payload, config)
+
+
+class AWSTitanAdapter(ProviderAdapter):
+    # TODO handle top_p, top_k, etc.
+    @classmethod
+    def completions_to_model(cls, payload, config):
+        n = payload.pop("n", 1)
+        if n != 1:
+            raise AIGatewayException(
+                status_code=422,
+                detail=f"'n' must be '1' for AWS Titan models. Received value: '{n}'.",
+            )
+
+        # The range of Titan's temperature is 0-1, but ours is 0-2, so we halve it
+        if "temperature" in payload:
+            payload["temperature"] = 0.5 * payload["temperature"]
+        return {
+            "inputText": payload.pop("prompt"),
+            "textGenerationConfig": rename_payload_keys(
+                payload, {"max_tokens": "maxTokenCount", "stop": "stopSequences"}
+            ),
+        }
+
+    @classmethod
+    def model_to_completions(cls, resp, config):
+        return completions.ResponsePayload(
+            created=int(time.time()),
+            object="text_completion",
+            model=config.model.name,
+            choices=[
+                completions.Choice(
+                    index=idx,
+                    text=candidate.get("outputText"),
+                    finish_reason=None,
+                )
+                for idx, candidate in enumerate(resp.get("results", []))
+            ],
+            usage=completions.CompletionsUsage(
+                prompt_tokens=None,
+                completion_tokens=None,
+                total_tokens=None,
+            ),
+        )
+
+    @classmethod
+    def embeddings_to_model(cls, payload, config):
+        raise NotImplementedError
+
+    @classmethod
+    def model_to_embeddings(cls, resp, config):
+        raise NotImplementedError
+
+
+class AI21Adapter(ProviderAdapter):
+    # TODO handle top_p, top_k, etc.
+    @classmethod
+    def completions_to_model(cls, payload, config):
+        return rename_payload_keys(
+            payload,
+            {
+                "stop": "stopSequences",
+                "n": "numResults",
+                "max_tokens": "maxTokens",
+            },
+        )
+
+    @classmethod
+    def model_to_completions(cls, resp, config):
+        return completions.ResponsePayload(
+            created=int(time.time()),
+            object="text_completion",
+            model=config.model.name,
+            choices=[
+                completions.Choice(
+                    index=idx,
+                    text=candidate.get("data", {}).get("text"),
+                    finish_reason=None,
+                )
+                for idx, candidate in enumerate(resp.get("completions", []))
+            ],
+            usage=completions.CompletionsUsage(
+                prompt_tokens=None,
+                completion_tokens=None,
+                total_tokens=None,
+            ),
+        )
+
+    @classmethod
+    def embeddings_to_model(cls, payload, config):
+        raise NotImplementedError
+
+    @classmethod
+    def model_to_embeddings(cls, resp, config):
+        raise NotImplementedError
+
+
+class AmazonBedrockModelProvider(Enum):
+    AMAZON = "amazon"
+    COHERE = "cohere"
+    AI21 = "ai21"
+    ANTHROPIC = "anthropic"
+
+    @property
+    def adapter_class(self) -> type[ProviderAdapter]:
+        return AWS_MODEL_PROVIDER_TO_ADAPTER.get(self)
+
+    @classmethod
+    def of_str(cls, name: str):
+        name = name.lower()
+
+        for opt in cls:
+            if opt.name.lower() in name or opt.value.lower() in name:
+                return opt
+
+
+AWS_MODEL_PROVIDER_TO_ADAPTER = {
+    AmazonBedrockModelProvider.COHERE: CohereAdapter,
+    AmazonBedrockModelProvider.ANTHROPIC: AmazonBedrockAnthropicAdapter,
+    AmazonBedrockModelProvider.AMAZON: AWSTitanAdapter,
+    AmazonBedrockModelProvider.AI21: AI21Adapter,
+}
+
+
+class AmazonBedrockProvider(BaseProvider):
+    NAME = "Amazon Bedrock"
+    CONFIG_TYPE = AmazonBedrockConfig
+
+    def __init__(self, config: RouteConfig):
+        super().__init__(config)
+
+        if config.model.config is None or not isinstance(config.model.config, AmazonBedrockConfig):
+            raise TypeError(f"Invalid config type {config.model.config}")
+        self.bedrock_config: AmazonBedrockConfig = config.model.config
+        self._client = None
+        self._client_created = 0
+
+    def _client_expired(self):
+        if not isinstance(self.bedrock_config.aws_config, AWSRole):
+            return False
+
+        return (
+            (time.monotonic_ns() - self._client_created)
+            >= (self.bedrock_config.aws_config.session_length_seconds) * 1_000_000_000,
+        )
+
+    def get_bedrock_client(self):
+        import boto3
+        import botocore.exceptions
+
+        if self._client is not None and not self._client_expired():
+            return self._client
+
+        session = boto3.Session(**self._construct_session_args())
+
+        try:
+            self._client, self._client_created = (
+                session.client(
+                    service_name="bedrock-runtime",
+                    **self._construct_client_args(session),
+                ),
+                time.monotonic_ns(),
+            )
+            return self._client
+        except botocore.exceptions.UnknownServiceError as e:
+            raise AIGatewayConfigException(
+                "Cannot create Amazon Bedrock client; ensure boto3/botocore "
+                "linked from the Amazon Bedrock user guide are installed. "
+                "Otherwise likely missing credentials or accessing account without to "
+                "Amazon Bedrock Private Preview"
+            ) from e
+
+    def _construct_session_args(self):
+        session_args = {
+            "region_name": self.bedrock_config.aws_config.aws_region,
+        }
+
+        return {k: v for k, v in session_args.items() if v}
+
+    def _construct_client_args(self, session):
+        aws_config = self.bedrock_config.aws_config
+
+        if isinstance(aws_config, AWSRole):
+            role = session.client(service_name="sts").assume_role(
+                RoleArn=aws_config.aws_role_arn,
+                RoleSessionName="ai-gateway-bedrock",
+                DurationSeconds=aws_config.session_length_seconds,
+            )
+            return {
+                "aws_access_key_id": role["Credentials"]["AccessKeyId"],
+                "aws_secret_access_key": role["Credentials"]["SecretAccessKey"],
+                "aws_session_token": role["Credentials"]["SessionToken"],
+            }
+        elif isinstance(aws_config, AWSIdAndKey):
+            return {
+                "aws_access_key_id": aws_config.aws_access_key_id,
+                "aws_secret_access_key": aws_config.aws_secret_access_key,
+                "aws_session_token": aws_config.aws_session_token,
+            }
+        else:
+            return {}
+
+    @property
+    def _underlying_provider(self):
+        if (not self.config.model.name) or "." not in self.config.model.name:
+            return None
+        return AmazonBedrockModelProvider.of_str(self.config.model.name)
+
+    @property
+    def adapter_class(self) -> type[ProviderAdapter]:
+        provider = self._underlying_provider
+        if not provider:
+            raise AIGatewayException(
+                status_code=422,
+                detail=f"Unknown Amazon Bedrock model type {self._underlying_provider}",
+            )
+        adapter = provider.adapter_class
+        if not adapter:
+            raise AIGatewayException(
+                status_code=422,
+                detail=f"Don't know how to handle {self._underlying_provider} for Amazon Bedrock",
+            )
+        return adapter
+
+    def _request(self, body):
+        import botocore.exceptions
+
+        try:
+            response = self.get_bedrock_client().invoke_model(
+                body=json.dumps(body).encode(),
+                modelId=self.config.model.name,
+                # defaults
+                # save=False,
+                accept="application/json",
+                contentType="application/json",
+            )
+            return json.loads(response.get("body").read())
+
+        # TODO work though botocore.exceptions to make this catchable.
+        # except botocore.exceptions.ValidationException as e:
+        #     raise HTTPException(status_code=422, detail=str(e)) from e
+
+        except botocore.exceptions.ReadTimeoutError as e:
+            raise AIGatewayException(status_code=408) from e
+
+    async def completions(self, payload: completions.RequestPayload) -> completions.ResponsePayload:
+        from fastapi.encoders import jsonable_encoder
+
+        self.check_for_model_field(payload)
+        payload = jsonable_encoder(payload, exclude_none=True, exclude_defaults=True)
+        payload = self.adapter_class.completions_to_model(payload, self.config)
+        response = self._request(payload)
+        return self.adapter_class.model_to_completions(response, self.config)

mlflow/genai/datasets/databricks_evaluation_dataset_source.py

@@ -0,0 +1,77 @@
+from typing import Any
+
+from mlflow.data.dataset_source import DatasetSource
+
+
+class DatabricksEvaluationDatasetSource(DatasetSource):
+    """
+    Represents a Databricks Evaluation Dataset source.
+
+    This source is used for datasets managed by the Databricks agents SDK.
+    """
+
+    def __init__(self, table_name: str, dataset_id: str):
+        """
+        Args:
+            table_name: The three-level UC table name of the dataset
+            dataset_id: The unique identifier of the dataset
+        """
+        self._table_name = table_name
+        self._dataset_id = dataset_id
+
+    @property
+    def table_name(self) -> str:
+        """The UC table name of the dataset."""
+        return self._table_name
+
+    @property
+    def dataset_id(self) -> str:
+        """The unique identifier of the dataset."""
+        return self._dataset_id
+
+    @staticmethod
+    def _get_source_type() -> str:
+        return "databricks_evaluation_dataset"
+
+    def load(self, **kwargs) -> Any:
+        """
+        Loads the dataset from the source.
+
+        This method is not implemented as the dataset should be loaded through
+        the databricks.agents.datasets API.
+        """
+        raise NotImplementedError(
+            "Loading a Databricks Evaluation Dataset from source is not supported"
+        )
+
+    @staticmethod
+    def _can_resolve(raw_source: dict[str, Any]) -> bool:
+        """
+        Determines whether the source can be resolved from a dictionary representation.
+        """
+        # Resolution from a dictionary representation is not supported for Databricks Evaluation
+        # Datasets
+        return False
+
+    @classmethod
+    def _resolve(cls, raw_source: dict[str, Any]):
+        """
+        Resolves the source from a dictionary representation.
+        """
+        raise NotImplementedError("Resolution from a source dictionary is not supported")
+
+    def to_dict(self) -> dict[str, Any]:
+        """
+        Returns a dictionary representation of the source.
+        """
+        return {
+            "table_name": self._table_name,
+            "dataset_id": self._dataset_id,
+        }
+
+    @classmethod
+    def from_dict(cls, source_dict: dict[str, Any]) -> "DatabricksEvaluationDatasetSource":
+        """
+        Creates an instance from a dictionary representation.
+        """
+        return cls(table_name=source_dict["table_name"], dataset_id=source_dict["dataset_id"])