guohanghui commited on
Commit
54baa0b
·
verified ·
1 Parent(s): e9bcb71

Update ASE/mcp_output/mcp_plugin/mcp_service.py

Browse files
ASE/mcp_output/mcp_plugin/mcp_service.py CHANGED
@@ -2,10 +2,15 @@ import os
2
  import sys
3
  from typing import List, Optional, Dict, Any
4
 
 
 
 
 
 
5
  from fastmcp import FastMCP
6
  import numpy as np
7
 
8
- # Import ASE modules (use pip installed version)
9
  from ase import Atoms
10
  from ase.build import bulk, molecule, fcc111, bcc100, hcp0001, surface
11
  from ase.io import read, write
 
2
  import sys
3
  from typing import List, Optional, Dict, Any
4
 
5
+ # Add the local source directory to sys.path
6
+ source_path = os.path.join(os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))), "source")
7
+ if source_path not in sys.path:
8
+ sys.path.insert(0, source_path)
9
+
10
  from fastmcp import FastMCP
11
  import numpy as np
12
 
13
+ # Import ASE modules from local source
14
  from ase import Atoms
15
  from ase.build import bulk, molecule, fcc111, bcc100, hcp0001, surface
16
  from ase.io import read, write