Merge branch 'main' of https://github.com/Aaron617/owl into main
Browse files- 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 |
-
"
|
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
|