mattergen转服务

This commit is contained in:
lzy
2025-04-02 16:24:50 +08:00
parent a77c2cd377
commit 7034566ee6
30 changed files with 656 additions and 339 deletions

View File

@@ -1,21 +1,7 @@
"""
Mars Toolkit
A comprehensive toolkit for materials science research, providing tools for:
- Material generation and property prediction
- Structure optimization
- Database queries (Materials Project, OQMD)
- Knowledge base retrieval
- Web search
Author: Yutang LI
Institution: SIAT-MIC
Contact: yt.li2@siat.ac.cn
"""
# Core modules
from mars_toolkit.core.config import config
from mars_toolkit.core.utils import setup_logging
# Basic tools
from mars_toolkit.misc.misc_tools import get_current_time
@@ -35,12 +21,14 @@ from mars_toolkit.query.oqmd_query import fetch_chemical_composition_from_OQMD
from mars_toolkit.query.dify_search import retrieval_from_knowledge_base
from mars_toolkit.query.web_search import search_online
# Visualization modules
from mars_toolkit.core.llm_tools import llm_tool, get_tools, get_tool_schemas
# Initialize logging
setup_logging()
__version__ = "0.1.0"
__all__ = ["llm_tool", "get_tools", "get_tool_schemas"]