local-llm-server/llm_server/llm/info.py

import requests

from llm_server import opts


def get_running_model():
    if opts.mode == 'oobabooga':
        try:
            backend_response = requests.get(f'{opts.backend_url}/api/v1/model', timeout=3)
        except Exception as e:
            return False, e
        try:
            r_json = backend_response.json()
            return r_json['result']
        except Exception as e:
            return False, e
    elif opts.mode == 'hf-textgen':
        try:
            backend_response = requests.get(f'{opts.backend_url}/info')
        except Exception as e:
            return False, e
        try:
            r_json = backend_response.json()
            return r_json['model_id'].replace('/', '_')
        except Exception as e:
            return False, e
    else:
        raise Exception