|
- from autogen import AssistantAgent, UserProxyAgent, config_list_from_json
-
-
- async def solve_task(task):
- config_list = config_list_from_json(env_or_file="OAI_CONFIG_LIST")
- assistant = AssistantAgent("assistant", llm_config={"config_list": config_list})
- user_proxy = UserProxyAgent(
- "user_proxy",
- code_execution_config={"work_dir": "coding", "use_docker": False},
- human_input_mode="NEVER",
- is_termination_msg=lambda msg: "TERMINATE" in msg.get("content", ""),
- )
- await user_proxy.a_initiate_chat(assistant, message=task)
-
- await user_proxy.a_send(
- f"""Based on the results in above conversation, create a response for the user.
- While computing the response, remember that this conversation was your inner mono-logue.
- The user does not need to know every detail of the conversation.
- All they want to see is the appropriate result for their task (repeated below) in
- a manner that would be most useful. Response should be less than 1500 characters.
-
- The task was: {task}
-
- There is no need to use the word TERMINATE in this response.
-
- """,
- assistant,
- request_reply=False,
- silent=True,
- )
- response = await assistant.a_generate_reply(assistant.chat_messages[user_proxy], user_proxy)
- await assistant.a_send(response, user_proxy, request_reply=False, silent=True)
-
- last_message = assistant.chat_messages[user_proxy][-1]["content"]
-
- return last_message[:2000]
|
