A script to push a new LLM (Large Language Model) model to JFrog Artifactory (under 30 lines of code :)

LLM_2_Artifactory.py

import requests

# Artifactory configuration - Change it to your own settings
ARTIFACTORY_URL = 'https://your-artifactory-url.com'
ARTIFACTORY_REPO = 'your-repo-name'
ARTIFACTORY_USERNAME = 'your-username'
ARTIFACTORY_API_KEY = 'your-api-key'

# LLM model file path
LLM_MODEL_FILE = 'path/to/your/llm_model.tar.gz'

def push_to_artifactory():
    url = f'{ARTIFACTORY_URL}/{ARTIFACTORY_REPO}/llm_model.tar.gz'
    headers = {
        'Authorization': f'Bearer {ARTIFACTORY_API_KEY}',
        'X-JFrog-Art-Api': ARTIFACTORY_API_KEY
    }
    with open(LLM_MODEL_FILE, 'rb') as file:
        data = file.read()
    try:
        response = requests.put(url, headers=headers, data=data)
        if response.status_code == 201:
            print("LLM model pushed to Artifactory successfully!")
        else:
            print(f"Failed to push LLM model. Status Code: {response.status_code}, Message: {response.text}")
    except requests.exceptions.RequestException as e:
        print(f"Error while pushing the LLM model to Artifactory: {e}")
if __name__ == "__main__":
    push_to_artifactory()