Skip to content

Commit e028eb7

Browse files
add structure generating as one scene of db fetching
1 parent 1bcba28 commit e028eb7

1 file changed

Lines changed: 6 additions & 6 deletions

File tree

  • agents/matmaster_agent/sub_agents

agents/matmaster_agent/sub_agents/tools.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -831,7 +831,7 @@
831831
},
832832
'fetch_structures_with_filter': {
833833
'belonging_agent': OPTIMADE_DATABASE_AGENT_NAME,
834-
'scene': [SceneEnum.DATABASE_SEARCH],
834+
'scene': [SceneEnum.DATABASE_SEARCH, SceneEnum.STRUCTURE_GENERATE],
835835
'description': OptimadeFilterToolDescription,
836836
'args_setting': f"{OptimadeFilterArgsSetting}",
837837
'alternative': [
@@ -844,7 +844,7 @@
844844
},
845845
'fetch_structures_with_spg': {
846846
'belonging_agent': OPTIMADE_DATABASE_AGENT_NAME,
847-
'scene': [SceneEnum.DATABASE_SEARCH],
847+
'scene': [SceneEnum.DATABASE_SEARCH, SceneEnum.STRUCTURE_GENERATE],
848848
'description': OptimadeSpgToolDescription,
849849
'args_setting': f"{OptimadeSpgArgsSetting}",
850850
'alternative': [
@@ -857,7 +857,7 @@
857857
},
858858
'fetch_structures_with_bandgap': {
859859
'belonging_agent': OPTIMADE_DATABASE_AGENT_NAME,
860-
'scene': [SceneEnum.DATABASE_SEARCH],
860+
'scene': [SceneEnum.DATABASE_SEARCH, SceneEnum.STRUCTURE_GENERATE],
861861
'description': OptimadeBandgapToolDescription,
862862
'args_setting': f"{OptimadeBandgapArgsSetting}",
863863
'alternative': [
@@ -870,7 +870,7 @@
870870
},
871871
'fetch_bohrium_crystals': {
872872
'belonging_agent': BOHRIUMPUBLIC_DATABASE_AGENT_NAME,
873-
'scene': [SceneEnum.DATABASE_SEARCH],
873+
'scene': [SceneEnum.DATABASE_SEARCH, SceneEnum.STRUCTURE_GENERATE],
874874
'description': BohriumPublicAgentToolDescription,
875875
'args_setting': f"{BohriumPublicAgentArgsSetting}",
876876
'alternative': [
@@ -883,7 +883,7 @@
883883
},
884884
'fetch_openlam_structures': {
885885
'belonging_agent': OPENLAM_DATABASE_AGENT_NAME,
886-
'scene': [SceneEnum.DATABASE_SEARCH],
886+
'scene': [SceneEnum.DATABASE_SEARCH, SceneEnum.STRUCTURE_GENERATE],
887887
'description': OpenlamAgentToolDescription,
888888
'args_setting': f"{OpenlamAgentArgsSetting}",
889889
'alternative': [
@@ -896,7 +896,7 @@
896896
},
897897
'fetch_mofs_sql': {
898898
'belonging_agent': MOFDB_DATABASE_AGENT_NAME,
899-
'scene': [SceneEnum.DATABASE_SEARCH],
899+
'scene': [SceneEnum.DATABASE_SEARCH, SceneEnum.STRUCTURE_GENERATE],
900900
'description': MofdbAgentToolDescription,
901901
'args_setting': f"{MofdbAgentArgsSetting}",
902902
'alternative': ['web-search'],

0 commit comments

Comments
 (0)