clean code (#95)

This commit is contained in:
Mengkang Hu
2025-03-09 13:05:03 +08:00
committed by GitHub
4 changed files with 3 additions and 30 deletions

View File

@@ -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
@@ -34,13 +31,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 = [

View File

@@ -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()

View File

@@ -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()

View File

@@ -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'