Skip to content

Commit

Permalink
update uploading script
Browse files Browse the repository at this point in the history
  • Loading branch information
dongyuanjushi committed Nov 30, 2024
1 parent 2838ea0 commit a6806ad
Show file tree
Hide file tree
Showing 2 changed files with 26 additions and 6 deletions.
16 changes: 13 additions & 3 deletions cerebrum/example/upload_agent_demo.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,17 @@
from cerebrum.manager.agent import AgentManager

manager = AgentManager('https://my.aios.foundation')
import argparse

agent_package = manager.package_agent('/Users/rama2r/Cerebrum/example/agents/academic_agent')
if __name__ == "__main__":
parser = argparse.ArgumentParser(description="Upload agents")
parser.add_argument(
"--agent_path",
required=True
)
args = parser.parse_args()

manager.upload_agent(agent_package)
manager = AgentManager('https://my.aios.foundation')

agent_package = manager.package_agent(args.agent_path)

manager.upload_agent(agent_package)
16 changes: 13 additions & 3 deletions cerebrum/example/upload_tool_demo.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,17 @@
from cerebrum.manager.tool import ToolManager

manager = ToolManager('https://my.aios.foundation')
import argparse

tool_package = manager.package_tool('/Users/rama2r/Cerebrum/example/tools/arxiv')
if __name__ == "__main__":
parser = argparse.ArgumentParser(description="Upload tools")
parser.add_argument(
"--tool_path",
required=True
)
args = parser.parse_args()

manager.upload_tool(tool_package)
manager = ToolManager('https://my.aios.foundation')

tool_package = manager.package_tool(args.tool_path)

manager.upload_tool(tool_package)

0 comments on commit a6806ad

Please sign in to comment.