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
|