Wendong-Fan commited on
Commit
19ebfc5
·
1 Parent(s): 09a1a1e

update example

Browse files
Files changed (2) hide show
  1. owl/run.py +6 -1
  2. owl/run_qwen_zh.py +6 -1
owl/run.py CHANGED
@@ -74,6 +74,11 @@ def construct_society(question: str) -> OwlRolePlaying:
74
  model_type=ModelType.GPT_4O,
75
  model_config_dict={"temperature": 0},
76
  ),
 
 
 
 
 
77
  }
78
 
79
  # Configure toolkits
@@ -91,7 +96,7 @@ def construct_society(question: str) -> OwlRolePlaying:
91
  SearchToolkit().search_google, # Comment this out if you don't have google search
92
  SearchToolkit().search_wiki,
93
  *ExcelToolkit().get_tools(),
94
- *DocumentProcessingToolkit().get_tools(),
95
  ]
96
 
97
  # Configure agent roles and parameters
 
74
  model_type=ModelType.GPT_4O,
75
  model_config_dict={"temperature": 0},
76
  ),
77
+ "document": ModelFactory.create(
78
+ model_platform=ModelPlatformType.OPENAI,
79
+ model_type=ModelType.GPT_4O,
80
+ model_config_dict={"temperature": 0},
81
+ ),
82
  }
83
 
84
  # Configure toolkits
 
96
  SearchToolkit().search_google, # Comment this out if you don't have google search
97
  SearchToolkit().search_wiki,
98
  *ExcelToolkit().get_tools(),
99
+ *DocumentProcessingToolkit(model=models["document"]).get_tools(),
100
  ]
101
 
102
  # Configure agent roles and parameters
owl/run_qwen_zh.py CHANGED
@@ -80,6 +80,11 @@ def construct_society(question: str) -> OwlRolePlaying:
80
  model_type=ModelType.QWEN_VL_MAX,
81
  model_config_dict={"temperature": 0},
82
  ),
 
 
 
 
 
83
  }
84
 
85
  # Configure toolkits
@@ -97,7 +102,7 @@ def construct_society(question: str) -> OwlRolePlaying:
97
  SearchToolkit().search_google, # Comment this out if you don't have google search
98
  SearchToolkit().search_wiki,
99
  *ExcelToolkit().get_tools(),
100
- *DocumentProcessingToolkit().get_tools(),
101
  ]
102
 
103
  # Configure agent roles and parameters
 
80
  model_type=ModelType.QWEN_VL_MAX,
81
  model_config_dict={"temperature": 0},
82
  ),
83
+ "document": ModelFactory.create(
84
+ model_platform=ModelPlatformType.QWEN,
85
+ model_type=ModelType.QWEN_VL_MAX,
86
+ model_config_dict={"temperature": 0},
87
+ ),
88
  }
89
 
90
  # Configure toolkits
 
102
  SearchToolkit().search_google, # Comment this out if you don't have google search
103
  SearchToolkit().search_wiki,
104
  *ExcelToolkit().get_tools(),
105
+ *DocumentProcessingToolkit(model=models["document"]).get_tools(),
106
  ]
107
 
108
  # Configure agent roles and parameters