Get models from LM Studio server and prepare them for llm. On Mac the output goes to `Library/Application\ Support/io.datasette.llm/extra-openai-models.yaml`

lms_to_llm.py

import requests
import subprocess
import yaml

def get_data_from_api():
    base_url = "http://localhost:1234/v1"
    response = requests.get(base_url + "/models")

    if response.status_code == 200:
        json_data = response.json()

        models_data = []

        for data in json_data['data']:
            model = {
                "api_base": base_url,
                "model_name": data['id'],
                "model_id": data['id']
            }

            models_data.append(model)

        yaml_output = yaml.dump(models_data, default_flow_style=False)

        return yaml_output

    else:
        print("Failed to retrieve data from API. Status code: ", response.status_code)
        return None

def get_other_models_path():
    result = subprocess.run(['llm', 'keys', 'path'], capture_output=True, text=True)

    if result.returncode == 0:
        output = result.stdout.strip()
        output = output[:-len('keys.json')]
        
        return output
    else:
        return None
        # print(f"Command failed with return code {result.returncode}")
        # print(result.stderr)


yaml_output = get_data_from_api()
omp = get_other_models_path()

if omp:
    print(f"Copy the following code to {omp + 'extra-openai-models.yaml'}\n")

if yaml_output:
    print(yaml_output)