85 lines
2.3 KiB
Python
Executable File
85 lines
2.3 KiB
Python
Executable File
"""
|
|
Configuration Module
|
|
|
|
This module provides configuration settings for the Mars Toolkit.
|
|
It includes API keys, endpoints, paths, and other configuration parameters.
|
|
"""
|
|
|
|
from typing import Dict, Any
|
|
|
|
class Config:
|
|
|
|
@classmethod
|
|
def as_dict(cls) -> Dict[str, Any]:
|
|
"""Return all configuration settings as a dictionary"""
|
|
return {
|
|
key: value for key, value in cls.__dict__.items()
|
|
if not key.startswith('__') and not callable(value)
|
|
}
|
|
|
|
@classmethod
|
|
def update(cls, **kwargs):
|
|
"""Update configuration settings"""
|
|
for key, value in kwargs.items():
|
|
if hasattr(cls, key):
|
|
setattr(cls, key, value)
|
|
|
|
|
|
class General_Config(Config):
|
|
"""Configuration class for General MCP"""
|
|
|
|
# Searxng
|
|
SEARXNG_HOST="http://192.168.168.1:40032/"
|
|
SEARXNG_MAX_RESULTS=10
|
|
|
|
|
|
class Material_Config(Config):
|
|
|
|
|
|
# Materials Project
|
|
MP_API_KEY = 'PMASAg256b814q3OaSRWeVc7MKx4mlKI'
|
|
MP_ENDPOINT = 'https://api.materialsproject.org/'
|
|
MP_TOPK = 3
|
|
|
|
LOCAL_MP_PROPS_ROOT = '/home/ubuntu/sas0/LYT/paper_dataset/mp_cif/Props/'
|
|
LOCAL_MP_CIF_ROOT = '/home/ubuntu/sas0/LYT/paper_dataset/mp_cif/MPDatasets/'
|
|
# Proxy
|
|
HTTP_PROXY = ''#'http://192.168.168.1:20171'
|
|
HTTPS_PROXY = ''#'http://192.168.168.1:20171'
|
|
|
|
# FairChem
|
|
FAIRCHEM_MODEL_PATH = '/home/ubuntu/sas0/lzy/multi_mcp_server/sci_mcp/material_mcp/support/pretrained_models/fairchem_ckpt/eqV2_86M_omat_mp_salex.pt'
|
|
FMAX = 0.05
|
|
|
|
# MatterGen
|
|
MATTERGENMODEL_ROOT = '/home/ubuntu/sas0/lzy/multi_mcp_server/sci_mcp/material_mcp/support/pretrained_models/mattergen_ckpt'
|
|
MATTERGEN_ROOT='/home/ubuntu/sas0/lzy/multi_mcp_server/sci_mcp/material_mcp/mattergen_gen/mattergen'
|
|
MATTERGENMODEL_RESULT_PATH = 'results/'
|
|
|
|
# Dify
|
|
DIFY_ROOT_URL = 'http://192.168.191.101:6080'
|
|
DIFY_API_KEY = 'app-IKZrS1RqIyurPSzR73mz6XSA'
|
|
|
|
#temp root
|
|
TEMP_ROOT = '/home/ubuntu/sas0/lzy/multi_mcp_server/temp/material'
|
|
|
|
|
|
class Chemistry_Config(Config):
|
|
|
|
|
|
TEMP_ROOT = '/home/ubuntu/sas0/lzy/multi_mcp_server/temp/chemistry'
|
|
|
|
RXN4_CHEMISTRY_KEY='apk-8928522a146c2503f30b16d9909222d7583f412ee8f1049f08d32a089ba88d34'
|
|
|
|
|
|
|
|
|
|
|
|
|
|
material_config = Material_Config()
|
|
general_config = General_Config()
|
|
chemistry_config = Chemistry_Config()
|
|
|
|
|
|
|