生成sft数据,设置OQMD的代理,测试mars-t1

mars_toolkit/query/oqmd_query.py

@@ -1,11 +1,13 @@
 import logging
+import os
 import httpx
 import pandas as pd
 from bs4 import BeautifulSoup
 from io import StringIO
 from typing import Annotated
 
-from mars_toolkit.core.llm_tools import llm_tool
+from ..core import config
+from ..core.llm_tools import llm_tool
 
 logger = logging.getLogger(__name__)
 
@@ -23,6 +25,8 @@ async def fetch_chemical_composition_from_OQMD(
         Formatted text with material information and property tables
     """
     # Fetch data from OQMD
+    os.environ['HTTP_PROXY'] = config.HTTP_PROXY or ''
+    os.environ['HTTPS_PROXY'] = config.HTTPS_PROXY or ''
     url = f"https://www.oqmd.org/materials/composition/{composition}"
     try:
         async with httpx.AsyncClient(timeout=100.0) as client: