config.cpython-310.pyc•5.3 kB
o
0z�g0 � @ s� d Z ddlZddlZddlZddlmZ ddlmZmZm Z ddddd �d
d
d �d d
d
d�ddd�d�Z
e� e �Z
G dd� d�Ze� ZdS )z�
Configuration module for the persistent-code MCP server.
Handles loading and saving configuration settings for:
- LlamaIndex integration
- Embedding models
- Storage settings
- Logging levels
� N)�Path)�Dict�Any�OptionalTz&sentence-transformers/all-MiniLM-L6-v2i � )�enabled�embedding_model�
chunk_sizeZ
chunk_overlapF)Z
use_sqliteZcompress_storage�INFOZlogs)�levelZ file_loggingZ
log_directoryi� � )�max_tokens_per_component�similarity_top_k)� llama_index�storage�logging�advancedc @ s� e Zd ZdZd dee fdd�Zdeeef fdd�Z d d eeeef ddfd
d �Z
d d ed
ededefdd�Z d ed
ededdfdd�Z defdd�Z
defdd�Zdefdd�Zdefdd�Zdefd d �ZdS ) �Configz9Configuration manager for the persistent-code MCP server.N�
config_dirc C sH |rt |�| _nt �� d | _tj| jdd� | jd | _| �� | _dS )z�Initialize configuration.
Args:
config_dir: Directory for configuration files (default: ~/.persistent_code)
z.persistent_codeT)�exist_okz config.jsonN)r r �home�os�makedirs� config_file� load_config�config)�selfr � r �Q/Users/sparshdrolia/Sparsh personal/persistent-code-mcp/persistent_code/config.py�__init__- s zConfig.__init__�returnc
C s | j �� rszNt| j d��
}t�|�}W d � n1 s w Y t�d| j � �� t�� D ]"\}}||vr;|||< q.|�� D ]\}}||| vrO||| |< q?q.|W S t yr } zt�
d|� d�� t� � W Y d}~S d}~ww t�d| j � �� t� � }| � |� |S )zwLoad configuration from file or create default.
Returns:
Configuration dictionary
�rNzLoaded configuration from zError loading config: z . Using default configuration.z"Creating default configuration at )
r �exists�open�json�load�logger�info�DEFAULT_CONFIG�items� Exception�warning�copy� save_config)r �fr �section�values�key�value�er r r r B s0
�
����
zConfig.load_configr c
C s� |du r| j }z*t| jd��}tj||dd� W d � n1 s!w Y t�d| j� �� W dS tyL } zt�d|� �� W Y d}~dS d}~ww )zSave configuration to file.
Args:
config: Configuration to save (default: current config)
N�w� )�indentzSaved configuration to zError saving config: ) r r# r r$ �dumpr& r' r* �error)r r r. r3 r r r r- c s � ��zConfig.save_configr/ r1 �defaultc C s( z| j | | W S ty | Y S w )z�Get a configuration value.
Args:
section: Configuration section
key: Configuration key
default: Default value if not found
Returns:
Configuration value
)r �KeyError)r r/ r1 r9 r r r �gets s
�z
Config.getr2 c C s. || j vr
i | j |< || j | |< | �� dS )z�Set a configuration value.
Args:
section: Configuration section
key: Configuration key
value: Value to set
N)r r- )r r/ r1 r2 r r r �set� s
z
Config.setc C � | � ddtd d �S )zgGet the configured embedding model.
Returns:
Embedding model name
r r �r; r( �r r r r �get_embedding_model� �
�zConfig.get_embedding_modelc C r= )z{Check if LlamaIndex integration is enabled.
Returns:
True if enabled, False otherwise
r r r> r? r r r �is_llama_index_enabled� rA z Config.is_llama_index_enabledc C r= )zhGet the configured logging level.
Returns:
Logging level as string
r r r> r? r r r �get_logging_level� rA zConfig.get_logging_levelc C r= )z}Get the number of similar components to retrieve.
Returns:
Number of similar components
r r r> r? r r r �get_similarity_top_k� rA zConfig.get_similarity_top_kc C r= )z{Get the maximum number of tokens per component.
Returns:
Maximum tokens per component
r r
r> r? r r r � get_max_tokens_per_component� rA z#Config.get_max_tokens_per_component)N)�__name__�
__module__� __qualname__�__doc__r �strr r r r r- r; r< r@ �boolrB rC �intrD rE r r r r r * s ! r )rI r r$ r �pathlibr �typingr r r r( � getLoggerrF r& r r r r r r �<module> s4
�����