Commit
·
19ebfc5
1
Parent(s):
09a1a1e
update example
Browse files- owl/run.py +6 -1
- 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
|