github/owl
1
0
Fork 0
mirror of https://github.com/camel-ai/owl.git synced 2025-12-26 02:06:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

update mistral example code

Browse Source
This commit is contained in:
Wendong 2025-05-23 13:47:02 +08:00
parent 5c6edec7e0
commit 6e01599595
1 changed files with 12 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
examples/run_mistral.py
View File

@ -82,7 +82,7 @@ from camel.models import ModelFactory
from camel.toolkits import FunctionTool
from camel.types import ModelPlatformType, ModelType
from camel.logger import set_log_level
from camel.toolkits import MCPToolkit
from camel.toolkits import MCPToolkit, FileWriteToolkit, CodeExecutionToolkit
from camel.societies import RolePlaying
from owl.utils.enhanced_role_playing import arun_society
@ -109,7 +109,7 @@ async def construct_society(
models = {
"user": ModelFactory.create(
model_platform=ModelPlatformType.MISTRAL,
model_type=ModelType.MISTRAL_LARGE,
model_type=ModelType.MISTRAL_MEDIUM_3,
model_config_dict={"temperature": 0},
),
"assistant": ModelFactory.create(
@ -149,17 +149,21 @@ async def main():
# Default task
default_task = (
"I'd like a academic report about Andrew Ng, including "
"his research direction, published papers (At least 3), "
"institutions, etc. You have been provided with tools to do "
"browser operation. Open browser to finish the task."
"Help me search the latest reports about smart city, "
"summarize them and help me generate a PDF file. You have "
"been provided with tools to do browser operation. Open "
"browser to finish the task."
)
# Override default task if command line argument is provided
task = sys.argv[1] if len(sys.argv) > 1 else default_task
# Connect to all MCP toolkits
tools = [*mcp_toolkit.get_tools()]
# Connect to toolkits
tools = [
*mcp_toolkit.get_tools(),
*FileWriteToolkit().get_tools(),
*CodeExecutionToolkit().get_tools(),
]
society = await construct_society(task, tools)
answer, chat_history, token_count = await arun_society(society)
print(f"\033[94mAnswer: {answer}\033[0m")