Wendong-Fan commited on
Commit
d2b7ea4
·
2 Parent(s): 690d27d f30cd60

Merge branch 'main' of https://github.com/Aaron617/owl into main

Browse files
Files changed (1) hide show
  1. owl/run_deepseek_zh.py +2 -30
owl/run_deepseek_zh.py CHANGED
@@ -23,10 +23,10 @@ from dotenv import load_dotenv
23
 
24
  from camel.models import ModelFactory
25
  from camel.toolkits import (
26
- CodeExecutionToolkit,
27
  ExcelToolkit,
28
  SearchToolkit,
29
  FileWriteToolkit,
 
30
  )
31
  from camel.types import ModelPlatformType, ModelType
32
 
@@ -62,31 +62,6 @@ def construct_society(question: str) -> OwlRolePlaying:
62
  model_type=ModelType.DEEPSEEK_CHAT,
63
  model_config_dict={"temperature": 0},
64
  ),
65
- "web": ModelFactory.create(
66
- model_platform=ModelPlatformType.DEEPSEEK,
67
- model_type=ModelType.DEEPSEEK_CHAT,
68
- model_config_dict={"temperature": 0},
69
- ),
70
- "planning": ModelFactory.create(
71
- model_platform=ModelPlatformType.DEEPSEEK,
72
- model_type=ModelType.DEEPSEEK_CHAT,
73
- model_config_dict={"temperature": 0},
74
- ),
75
- "video": ModelFactory.create(
76
- model_platform=ModelPlatformType.DEEPSEEK,
77
- model_type=ModelType.DEEPSEEK_CHAT,
78
- model_config_dict={"temperature": 0},
79
- ),
80
- "image": ModelFactory.create(
81
- model_platform=ModelPlatformType.DEEPSEEK,
82
- model_type=ModelType.DEEPSEEK_CHAT,
83
- model_config_dict={"temperature": 0},
84
- ),
85
- "document": ModelFactory.create(
86
- model_platform=ModelPlatformType.DEEPSEEK,
87
- model_type=ModelType.DEEPSEEK_CHAT,
88
- model_config_dict={"temperature": 0},
89
- ),
90
  }
91
 
92
  # Configure toolkits
@@ -95,7 +70,6 @@ def construct_society(question: str) -> OwlRolePlaying:
95
  SearchToolkit().search_duckduckgo,
96
  SearchToolkit().search_wiki,
97
  *ExcelToolkit().get_tools(),
98
- *DocumentProcessingToolkit(model=models["document"]).get_tools(),
99
  *FileWriteToolkit(output_dir="./").get_tools(),
100
  ]
101
 
@@ -126,9 +100,7 @@ def main():
126
  r"""Main function to run the OWL system with an example question."""
127
  # Example research question
128
  question = (
129
- "请分析GitHub上CAMEL-AI项目的最新统计数据。找出该项目的星标数量、"
130
- "贡献者数量和最近的活跃度。然后,创建一个简单的Excel表格来展示这些数据,"
131
- "并生成一个柱状图来可视化这些指标。最后,总结CAMEL项目的受欢迎程度和发展趋势。"
132
  )
133
 
134
  # Construct and run the society
 
23
 
24
  from camel.models import ModelFactory
25
  from camel.toolkits import (
 
26
  ExcelToolkit,
27
  SearchToolkit,
28
  FileWriteToolkit,
29
+ CodeExecutionToolkit,
30
  )
31
  from camel.types import ModelPlatformType, ModelType
32
 
 
62
  model_type=ModelType.DEEPSEEK_CHAT,
63
  model_config_dict={"temperature": 0},
64
  ),
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
65
  }
66
 
67
  # Configure toolkits
 
70
  SearchToolkit().search_duckduckgo,
71
  SearchToolkit().search_wiki,
72
  *ExcelToolkit().get_tools(),
 
73
  *FileWriteToolkit(output_dir="./").get_tools(),
74
  ]
75
 
 
100
  r"""Main function to run the OWL system with an example question."""
101
  # Example research question
102
  question = (
103
+ "搜索OWL项目最近的新闻并生成一篇报告,最后保存到本地。"
 
 
104
  )
105
 
106
  # Construct and run the society