|
Server : Apache System : Linux iad1-shared-b8-43 6.6.49-grsec-jammy+ #10 SMP Thu Sep 12 23:23:08 UTC 2024 x86_64 User : dh_edsupp ( 6597262) PHP Version : 8.2.26 Disable Function : NONE Directory : /lib/python3/dist-packages/sos/report/plugins/__pycache__/ |
Upload File : |
o
-�_g� � @ s2 d dl Z d dlmZmZmZ G dd� dee�ZdS )� N)�Plugin�IndependentPlugin� PluginOptc @ sX e Zd ZdZd ZdZdZdZdZe dde
dd �e d
d e
d d �e d
ddd�gZ dd� Z dS )� InstructlabaL
This plugin is used to capture information about
Instructlab installations.
InstructLab is an open source project for enhancing
large language models (LLMs) used in generative
artificial intelligence (gen AI) applications.
Instructlab can run either as a container, or directly
outside a container.
� instructlab)�ai)r �ilab)r � ilab-userz
cloud-userzuser that runs instructlab)�default�val_type�desc�
ilab-conf-dir� zinstructlab data directory� get-cacheFz#Capture models and osci cached data)r
r c s d}d}d}d}d}�� ||�}�� ||�}�� ||�� ���� � d��� � d��� |d��� |d �g� g d
�}g d �} z ��d �d
d }
W n tyS d }
Y nw �j|� d�|� d�g|
d� �j� �fdd�| D �|
d� �jdd� |D �|
d� �j|d|
d� ��d�r��j|� |
d� ��t �j
�� ��d�} zt � | �} W n t
y� ��d| � d�� Y d S w | �r| j}
��d�rψ� |
��d��}
�� |
|������ |
|�� ����fd d�| D �� �j�� |
|�dd � ��d��r ���� |
|�� d S d S d S ) Nz /usr/share/instructlab/configz/opt/app-root/srcz.cache/instructlabz.config/instructlabz.local/share/instructlabz
taxonomy/.gitztaxonomy/.githubz*src/.local/share/instructlab/taxonomy/.gitz-src/.local/share/instructlab/taxonomy/.github)z
taxonomy diffz#taxonomy diff --taxonomy-base=emptyz system infoz
model listz config show) �data� generated�taxonomy�
taxonomy_data�chatlogs� checkpoints�datasets�internal�phasedz instructlab*r � z/rhel_ai_config.yamlz /config.yaml)� containerc � g | ]}�� � |��qS � �� path_join��.0�data_dir)�cont_local_path�selfr �@/usr/lib/python3/dist-packages/sos/report/plugins/instructlab.py�
<listcomp>g s �z%Instructlab.setup.<locals>.<listcomp>c S s g | ]}d |� ��qS )zilab r )r �subr r r$ r% l s T)� recursiver r r zUser "z�" does not exist, will not collect Instructlab information. Use `-k instructlab.ilab-user` option to define the user to use to collect data for Instructlabr
c r r r r )�data_dirs_baser# r r$ r% � s
��)r' )r �add_forbidden_path�get_all_containers_by_regex� Exception�
add_copy_spec�add_cmd_output�add_dir_listing�
get_option�add_container_logs�list�
containers�pwd�getpwnam�KeyError� _log_warn�pw_dir)r# �cont_share_conf_path�
cont_opt_path� cache_dir�
config_dir�local_share_dir�cont_cache_path�cont_config_path�subcmds� data_dirs�ilab_con� ilab_user�user_pwd�ilab_dirr )r" r( r# r$ �setup) s� �
���
��� �� ��
�
��
� �
�
��zInstructlab.setupN)
�__name__�
__module__� __qualname__�__doc__�
short_desc� plugin_name�profilesr2 �commandsr �str� option_listrE r r r r$ r s$
���� r )r3 �sos.report.pluginsr r r r r r r r$ �<module> s