config.cpython-310.pyc•3.52 kB
o
6��h< � @ s d Z ddlZddlmZ ddlmZ e� ee�jjZed Z e d Z
e� d�Z e� dd �Z
d
Zd Ze� d d
�Ze� dd��� dkZe� dd��� dkZee� dd��Zee� dd��Ze� dd�Zee� dd��Ze� dd��� dkZe� dd ��d �Ze� d d��� dkZ ee� d d ��Z ee� d!d��Z e� d"d#�Z ee� d$d%��Z e!e� d&d'��Z"d(e#e$e%e& f fd)d*�Z'd(e(fd+d,�Z)e
j*d-d-d.� e+d/k�r ddl,Z,e-d0� e-d1� e-d0� e'� \Z.Z/e.r�e-d2� ne-d3� e/D ] Z0e-d4e0� �� q�e-� e-d5� e-e,j1e)� d6d7�� dS dS )8zb
Configuration settings for RTIdeas MCP Server
Handles MongoDB connection and MCP server settings
� N)�Path)� load_dotenv�data�raw� MONGODB_URI�MONGODB_DATABASEZrtideasz rtideas-mcpz1.0.0� LOG_LEVEL�INFO�ENABLE_AUTO_ANALYSIS�true�ENABLE_WEBSOCKET� API_TIMEOUTZ30� MAX_RESULTSZ100�API_HOSTz0.0.0.0�API_PORTZ8000�
API_RELOAD�false� CORS_ORIGINS�*�,�CORS_CREDENTIALS�DEFAULT_SEARCH_LIMIT�10�MAX_SEARCH_LIMIT�DEFAULT_CLUSTER_METHOD�hdbscan�DEFAULT_CLUSTER_K�8� DEFAULT_SIMILARITY_THRESHOLDz0.6�returnc C s� g } t s | �d� tdkr| �d� tdkr| �d� tdkr$| �d� tdkr-| �d� tdk r6| �d� d t kr@d
ksFn | �d � g d �}t|vrV| �d
|� �� t | �dk| fS )zh
Validate the configuration settings.
Returns:
Tuple of (is_valid, list_of_errors)
z,MONGODB_URI environment variable is requiredr z&API_TIMEOUT must be a positive integerz&MAX_RESULTS must be a positive integerz/DEFAULT_SEARCH_LIMIT must be a positive integerz+MAX_SEARCH_LIMIT must be a positive integer� z$DEFAULT_CLUSTER_K must be at least 2g g �?z8DEFAULT_SIMILARITY_THRESHOLD must be between 0.0 and 1.0)r ZkmeansZlouvainz'DEFAULT_CLUSTER_METHOD must be one of: )
r �appendr
r r r r r r �len)�errorsZvalid_cluster_methods� r$ �$C:\Users\angel\Desktop\mcp\config.py�validate_config9 s&
r& c C sj t rtt �dkrt dd� d nt td�ttttttd�t t
d�t t d�dt
ittd �ttd
�d tid �S )
zq
Get a summary of the current configuration.
Returns:
Dictionary with configuration details
�2 Nz...)�uriZdatabase)�name�version�timeout�host�port�reload)Z
default_limitZ max_limit)Zdefault_methodZ default_kZdefault_threshold)Z
auto_analysis� websocket)Zorigins� credentials�level)�mongodb�server�searchZ
clustering�
similarityZfeatures�cors�logging)r r" r �MCP_SERVER_NAME�MCP_SERVER_VERSIONr
r r r r r r r r r
r r r r r$ r$ r$ r% �get_config_summarya s6 " �� �������r: T)�parents�exist_ok�__main__z<============================================================z RTIdeas MCP Server Configurationu ✅ Configuration is valid
u ❌ Configuration has errors:
z - zConfiguration Summary:r )�indent)2�__doc__�os�pathlibr �dotenvr �__file__�parentZ PROJECT_ROOTZDATA_DIRZ RAW_DATA_DIR�getenvr r r8 r9 r �lowerr
r �intr
r r r r �splitr r r r r r �floatr �tuple�bool�list�strr& �dictr: �mkdir�__name__�json�print�is_validr# �error�dumpsr$ r$ r$ r% �<module> sZ
(.
�