diff --git a/examples/run_novita_ai.py b/examples/run_novita_ai.py new file mode 100644 index 0000000..c981d22 --- /dev/null +++ b/examples/run_novita_ai.py @@ -0,0 +1,135 @@ +# ========= Copyright 2023-2024 @ CAMEL-AI.org. All Rights Reserved. ========= +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. +# ========= Copyright 2023-2024 @ CAMEL-AI.org. All Rights Reserved. ========= +import sys +import pathlib +from dotenv import load_dotenv +from camel.models import ModelFactory +from camel.toolkits import ( + CodeExecutionToolkit, + ExcelToolkit, + ImageAnalysisToolkit, + BrowserToolkit, + FileWriteToolkit, +) +from camel.types import ModelPlatformType, ModelType +from camel.logger import set_log_level +from camel.societies import RolePlaying + +from owl.utils import run_society, DocumentProcessingToolkit + +base_dir = pathlib.Path(__file__).parent.parent +env_path = base_dir / "owl" / ".env" +load_dotenv(dotenv_path=str(env_path)) + +set_log_level(level="DEBUG") + + +def construct_society(question: str) -> RolePlaying: + r"""Construct a society of agents based on the given question. + + Args: + question (str): The task or question to be addressed by the society. + + Returns: + RolePlaying: A configured society of agents ready to address the question. + """ + + # Create models for different components + models = { + "user": ModelFactory.create( + model_platform=ModelPlatformType.NOVITA, + model_type=ModelType.NOVITA_LLAMA_4_MAVERICK_17B, + model_config_dict={"temperature": 0}, + ), + "assistant": ModelFactory.create( + model_platform=ModelPlatformType.NOVITA, + model_type=ModelType.NOVITA_LLAMA_4_MAVERICK_17B, + model_config_dict={"temperature": 0}, + ), + "browsing": ModelFactory.create( + model_platform=ModelPlatformType.NOVITA, + model_type=ModelType.NOVITA_LLAMA_4_MAVERICK_17B, + model_config_dict={"temperature": 0}, + ), + "planning": ModelFactory.create( + model_platform=ModelPlatformType.NOVITA, + model_type=ModelType.NOVITA_LLAMA_4_MAVERICK_17B, + model_config_dict={"temperature": 0}, + ), + "image": ModelFactory.create( + model_platform=ModelPlatformType.NOVITA, + model_type=ModelType.NOVITA_LLAMA_4_MAVERICK_17B, + model_config_dict={"temperature": 0}, + ), + "document": ModelFactory.create( + model_platform=ModelPlatformType.NOVITA, + model_type=ModelType.NOVITA_LLAMA_4_MAVERICK_17B, + model_config_dict={"temperature": 0}, + ), + } + + # Configure toolkits + tools = [ + *BrowserToolkit( + headless=False, # Set to True for headless mode (e.g., on remote servers) + web_agent_model=models["browsing"], + planning_agent_model=models["planning"], + ).get_tools(), + *CodeExecutionToolkit(sandbox="subprocess", verbose=True).get_tools(), + *ImageAnalysisToolkit(model=models["image"]).get_tools(), + *ExcelToolkit().get_tools(), + *DocumentProcessingToolkit(model=models["document"]).get_tools(), + *FileWriteToolkit(output_dir="./").get_tools(), + ] + + # Configure agent roles and parameters + user_agent_kwargs = {"model": models["user"]} + assistant_agent_kwargs = {"model": models["assistant"], "tools": tools} + + # Configure task parameters + task_kwargs = { + "task_prompt": question, + "with_task_specify": False, + } + + # Create and return the society + society = RolePlaying( + **task_kwargs, + user_role_name="user", + user_agent_kwargs=user_agent_kwargs, + assistant_role_name="assistant", + assistant_agent_kwargs=assistant_agent_kwargs, + ) + + return society + + +def main(): + r"""Main function to run the OWL system with an example question.""" + # Default research question + default_task = "Open Brave search, summarize the github stars, fork counts, etc. of camel-ai's camel framework, and write the numbers into a python file using the plot package, save it locally, and run the generated python file." + + # Override default task if command line argument is provided + task = sys.argv[1] if len(sys.argv) > 1 else default_task + + # Construct and run the society + society = construct_society(task) + answer, chat_history, token_count = run_society(society) + + # Output the result + print(f"\033[94mAnswer: {answer}\033[0m") + + +if __name__ == "__main__": + main() diff --git a/owl/.env_template b/owl/.env_template index eb4901d..8d5ea06 100644 --- a/owl/.env_template +++ b/owl/.env_template @@ -32,6 +32,9 @@ OPENAI_API_KEY="Your_Key" #GOOGLE GEMINI API (https://ai.google.dev/gemini-api/docs/api-key) # GEMINI_API_KEY ="Your_Key" +# NOVITA API (https://novita.ai/settings/key-management?utm_source=github_owl&utm_medium=github_readme&utm_campaign=github_link) +# NOVITA_API_KEY="Your_Key" + #=========================================== # Tools & Services API #=========================================== diff --git a/owl/webapp.py b/owl/webapp.py index 33f06b6..7846e99 100644 --- a/owl/webapp.py +++ b/owl/webapp.py @@ -255,6 +255,7 @@ MODULE_DESCRIPTIONS = { "run_groq": "Using groq model to process tasks", "run_ppio": "Using ppio model to process tasks", "run_together_ai": "Using together ai model to process tasks", + "run_novita_ai": "Using novita ai model to process tasks", } @@ -640,6 +641,8 @@ def get_api_guide(key: str) -> str: return "https://chunkr.ai/" elif "firecrawl" in key_lower: return "https://www.firecrawl.dev/" + elif "novita" in key_lower: + return "https://novita.ai/settings/key-management?utm_source=github_owl&utm_medium=github_readme&utm_campaign=github_link" else: return "" diff --git a/owl/webapp_zh.py b/owl/webapp_zh.py index 966af01..ceb1fd8 100644 --- a/owl/webapp_zh.py +++ b/owl/webapp_zh.py @@ -254,6 +254,7 @@ MODULE_DESCRIPTIONS = { "run_groq": "使用groq模型处理任务", "run_ppio": "使用ppio模型处理任务", "run_together_ai": "使用together ai模型处理任务", + "run_novita_ai": "使用novita ai模型处理任务", } @@ -623,6 +624,8 @@ def get_api_guide(key: str) -> str: return "https://chunkr.ai/" elif "firecrawl" in key_lower: return "https://www.firecrawl.dev/" + elif "novita" in key_lower: + return "https://novita.ai/settings/key-management?utm_source=github_owl&utm_medium=github_readme&utm_campaign=github_link" else: return ""