bstraehle commited on
Commit
a768bed
1 Parent(s): 8e54661

Update multi_agent.py

Browse files
Files changed (1) hide show
  1. multi_agent.py +15 -0
multi_agent.py CHANGED
@@ -3,6 +3,11 @@ import base64, datetime, json, os
3
  from autogen import ConversableAgent, AssistantAgent
4
  from autogen.coding import LocalCommandLineCodeExecutor
5
 
 
 
 
 
 
6
  def read_file(file_path: str) -> str:
7
  with open(file_path, "r", encoding="utf-8") as file:
8
  return file.read()
@@ -61,6 +66,16 @@ def run_multi_agent(llm, task):
61
  code_execution_config=False,
62
  human_input_mode="NEVER",
63
  )
 
 
 
 
 
 
 
 
 
 
64
 
65
  chat_result = code_executor_agent.initiate_chat(
66
  code_writer_agent,
 
3
  from autogen import ConversableAgent, AssistantAgent
4
  from autogen.coding import LocalCommandLineCodeExecutor
5
 
6
+ ###
7
+ def get_today() -> Annotated[str, "Get today's date"]:
8
+ return str(date.today())
9
+ ###
10
+
11
  def read_file(file_path: str) -> str:
12
  with open(file_path, "r", encoding="utf-8") as file:
13
  return file.read()
 
66
  code_execution_config=False,
67
  human_input_mode="NEVER",
68
  )
69
+
70
+ ###
71
+ register_function(
72
+ get_today,
73
+ caller=code_writer_agent,
74
+ executor=code_writer_agent,
75
+ name="get_today",
76
+ description="Call this tool to get today's date.",
77
+ )
78
+ ###
79
 
80
  chat_result = code_executor_agent.initiate_chat(
81
  code_writer_agent,