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