From 671db3324ea3085afa3c21cbecf088383e0fc2f5 Mon Sep 17 00:00:00 2001 From: lazychih114 <55657767+Aaron617@users.noreply.github.com> Date: Sun, 9 Mar 2025 12:59:36 +0800 Subject: [PATCH] clean code --- owl/run.py | 9 +++------ owl/run_deepseek_example.py | 3 --- owl/run_gaia_roleplaying.py | 2 -- owl/run_mini.py | 19 ------------------- 4 files changed, 3 insertions(+), 30 deletions(-) diff --git a/owl/run.py b/owl/run.py index 6aea767..cf30504 100644 --- a/owl/run.py +++ b/owl/run.py @@ -13,15 +13,12 @@ from camel.toolkits import ( ExcelToolkit ) from camel.types import ModelPlatformType, ModelType -from camel.configs import ChatGPTConfig +# from camel.configs import ChatGPTConfig -from typing import List, Dict -from retry import retry from loguru import logger from utils import OwlRolePlaying, run_society -import os @@ -35,13 +32,13 @@ def construct_society(question: str) -> OwlRolePlaying: user_model = ModelFactory.create( model_platform=ModelPlatformType.OPENAI, model_type=ModelType.GPT_4O, - model_config_dict=ChatGPTConfig(temperature=0, top_p=1).as_dict(), # [Optional] the config for model + # model_config_dict=ChatGPTConfig(temperature=0, top_p=1).as_dict(), # [Optional] the config for model ) assistant_model = ModelFactory.create( model_platform=ModelPlatformType.OPENAI, model_type=ModelType.GPT_4O, - model_config_dict=ChatGPTConfig(temperature=0, top_p=1).as_dict(), # [Optional] the config for model + # model_config_dict=ChatGPTConfig(temperature=0, top_p=1).as_dict(), # [Optional] the config for model ) tools_list = [ diff --git a/owl/run_deepseek_example.py b/owl/run_deepseek_example.py index da14d9e..c51e6a0 100644 --- a/owl/run_deepseek_example.py +++ b/owl/run_deepseek_example.py @@ -3,13 +3,10 @@ from camel.toolkits import * from camel.types import ModelPlatformType, ModelType from camel.configs import DeepSeekConfig -from typing import List, Dict from dotenv import load_dotenv -from retry import retry from loguru import logger from utils import OwlRolePlaying, run_society -import os load_dotenv() diff --git a/owl/run_gaia_roleplaying.py b/owl/run_gaia_roleplaying.py index 8464af6..577cfbd 100644 --- a/owl/run_gaia_roleplaying.py +++ b/owl/run_gaia_roleplaying.py @@ -5,11 +5,9 @@ from camel.configs import ChatGPTConfig from utils import GAIABenchmark from dotenv import load_dotenv -from retry import retry from loguru import logger import os -import shutil load_dotenv() diff --git a/owl/run_mini.py b/owl/run_mini.py index 5045682..e00ac88 100644 --- a/owl/run_mini.py +++ b/owl/run_mini.py @@ -4,26 +4,15 @@ load_dotenv() from camel.models import ModelFactory from camel.toolkits import ( WebToolkit, - DocumentProcessingToolkit, - VideoAnalysisToolkit, - AudioAnalysisToolkit, - CodeExecutionToolkit, - ImageAnalysisToolkit, SearchToolkit, - ExcelToolkit, FunctionTool ) from camel.types import ModelPlatformType, ModelType -from camel.configs import ChatGPTConfig -from typing import List, Dict -from retry import retry from loguru import logger from utils import OwlRolePlaying, run_society -import os - @@ -36,13 +25,11 @@ def construct_society(question: str) -> OwlRolePlaying: user_model = ModelFactory.create( model_platform=ModelPlatformType.OPENAI, model_type=ModelType.GPT_4O, - model_config_dict=ChatGPTConfig(temperature=0, top_p=1).as_dict(), # [Optional] the config for model ) assistant_model = ModelFactory.create( model_platform=ModelPlatformType.OPENAI, model_type=ModelType.GPT_4O, - model_config_dict=ChatGPTConfig(temperature=0, top_p=1).as_dict(), # [Optional] the config for model ) tools_list = [ @@ -51,13 +38,7 @@ def construct_society(question: str) -> OwlRolePlaying: web_agent_model=assistant_model, planning_agent_model=assistant_model ).get_tools(), - # *DocumentProcessingToolkit().get_tools(), - # *VideoAnalysisToolkit(model=assistant_model).get_tools(), # This requires OpenAI Key - # *AudioAnalysisToolkit().get_tools(), # This requires OpenAI Key - # *CodeExecutionToolkit().get_tools(), - # *ImageAnalysisToolkit(model=assistant_model).get_tools(), FunctionTool(SearchToolkit(model=assistant_model).search_duckduckgo), - # *ExcelToolkit().get_tools() ] user_role_name = 'user'