Wendong-Fan commited on
Commit
57bb690
·
1 Parent(s): 0b8f215

update run.py

Browse files
Files changed (1) hide show
  1. owl/run.py +6 -6
owl/run.py CHANGED
@@ -24,14 +24,14 @@ def construct_society(question: str) -> OwlRolePlaying:
24
  assistant_role_name = "assistant"
25
 
26
  user_model = ModelFactory.create(
27
- model_platform=ModelPlatformType.DEFAULT,
28
- model_type=ModelType.DEFAULT,
29
  model_config_dict=ChatGPTConfig(temperature=0, top_p=1).as_dict(), # [Optional] the config for model
30
  )
31
 
32
  assistant_model = ModelFactory.create(
33
- model_platform=ModelPlatformType.DEFAULT,
34
- model_type=ModelType.DEFAULT,
35
  model_config_dict=ChatGPTConfig(temperature=0, top_p=1).as_dict(), # [Optional] the config for model
36
  )
37
 
@@ -42,10 +42,10 @@ def construct_society(question: str) -> OwlRolePlaying:
42
  planning_agent_model=assistant_model
43
  ).get_tools(),
44
  *DocumentProcessingToolkit().get_tools(),
45
- *VideoAnalysisToolkit().get_tools(), # This requires OpenAI and Qwen Key
 
46
  *CodeExecutionToolkit().get_tools(),
47
  *ImageAnalysisToolkit(model=assistant_model).get_tools(),
48
- *AudioAnalysisToolkit().get_tools(), # This requires OpenAI Key
49
  *SearchToolkit(model=assistant_model).get_tools(),
50
  *ExcelToolkit().get_tools()
51
  ]
 
24
  assistant_role_name = "assistant"
25
 
26
  user_model = ModelFactory.create(
27
+ model_platform=ModelPlatformType.OPENAI,
28
+ model_type=ModelType.GPT_4O,
29
  model_config_dict=ChatGPTConfig(temperature=0, top_p=1).as_dict(), # [Optional] the config for model
30
  )
31
 
32
  assistant_model = ModelFactory.create(
33
+ model_platform=ModelPlatformType.OPENAI,
34
+ model_type=ModelType.GPT_4O,
35
  model_config_dict=ChatGPTConfig(temperature=0, top_p=1).as_dict(), # [Optional] the config for model
36
  )
37
 
 
42
  planning_agent_model=assistant_model
43
  ).get_tools(),
44
  *DocumentProcessingToolkit().get_tools(),
45
+ *VideoAnalysisToolkit(model=assistant_model).get_tools(), # This requires OpenAI Key
46
+ *AudioAnalysisToolkit().get_tools(), # This requires OpenAI Key
47
  *CodeExecutionToolkit().get_tools(),
48
  *ImageAnalysisToolkit(model=assistant_model).get_tools(),
 
49
  *SearchToolkit(model=assistant_model).get_tools(),
50
  *ExcelToolkit().get_tools()
51
  ]