fix: update API endpoints, logging, and data types for enhanced compatibility and clarity

Author: Rasic2

agents/matmaster_agent/base_agents/job_agent.py

@@ -404,6 +404,9 @@ async def _run_async_impl(
                         ):
                             raw_result = part.function_response.response['result']
                             results = json.loads(raw_result.content[0].text)
+                            logger.info(
+                                f"[SubmitCoreCalculationMCPLlmAgent] results = {results}"
+                            )
                             origin_job_id = results['job_id']
                             job_name = part.function_response.name
                             job_status = results['status']

agents/matmaster_agent/constant.py

@@ -36,7 +36,7 @@
     DFLOW_K8S_API_SERVER = 'https://workflows.deepmodeling.com'
     BOHRIUM_API_URL = 'https://bohrium-api.dp.tech'
 
-OpenAPIJobAPI = f"{OPENAPI_HOST}/openapi/v1/job"
+OpenAPIJobAPI = f"{OPENAPI_HOST}/openapi/v1/sandbox/job"
 
 MATERIALS_ACCESS_KEY = str(os.getenv('MATERIALS_ACCESS_KEY'))
 MATERIALS_PROJECT_ID = int(os.getenv('MATERIALS_PROJECT_ID'))

agents/matmaster_agent/model.py

@@ -51,7 +51,7 @@ class JobResult(BaseModel):
 
 class BohrJobInfo(BaseModel):
     origin_job_id: str
-    job_id: int
+    job_id: Union[int, str]
     job_query_url: str
     job_detail_url: str
     job_status: JobStatus

agents/matmaster_agent/structure_generate_agent/agent.py

@@ -26,7 +26,7 @@
 ] = 'registry.dp.tech/dptech/dp/native/prod-788025/structure-generate-agent:small'
 StructureGenerateBohriumExecutor['machine']['remote_profile'][
     'machine_type'
-] = 'c8_m31_1 * NVIDIA T4'
+] = 'c8_m32_1 * NVIDIA 4090'
 
 sse_params = SseServerParams(url=StructureGenerateServerUrl)
 

agents/matmaster_agent/structure_generate_agent/constant.py

@@ -3,6 +3,6 @@
 StructureGenerateAgentName = 'structure_generate_agent'
 
 if CURRENT_ENV in ['test', 'uat']:
-    StructureGenerateServerUrl = 'http://pfmx1355864.bohrium.tech:50003/sse'
+    StructureGenerateServerUrl = 'http://hftp1387064.bohrium.tech:50001/sse'
 else:
     StructureGenerateServerUrl = 'https://cystalformer-uuid1754551471.app-space.dplink.cc/sse?token=1750cd294e6c4270946ae37107a725ff'