Library

If you want to use TaskWeaver as a library, you can refer to the following code example:

from taskweaver.app.app import TaskWeaverApp

# This is the folder that contains the taskweaver_config.json file and not the repo root. Defaults to "./project/"
app_dir = "./project/"
app = TaskWeaverApp(app_dir=app_dir)
session = app.get_session()

user_query = "hello, what can you do?"
response_round = session.send_message(user_query)
print(response_round.to_dict())

Essentially, you need to create a TaskWeaverApp object and then get a session object from it. Each time, you can send a message to TaskWeaver by calling session.send_message(user_query). The return value of session.send_message(user_query) is a Round object, which contains the response from TaskWeaver. A round is a conversation round between the user and TaskWeaver, which contains a list of posts.

An example of the Round object is shown below. To better understand the structure, you can refer to the Concepts section.

{
    "id": "round-20231201-043134-218a2681",
    "user_query": "hello, what can you do?",
    "state": "finished",
    "post_list": [
        {
            "id": "post-20231201-043134-10eedcca",
            "message": "hello, what can you do?",
            "send_from": "User",
            "send_to": "Planner",
            "attachment_list": []
        },
        {
            "id": "post-20231201-043141-86a2aaff",
            "message": "I can help you with various tasks, such as counting rows in a data file, detecting anomalies in a dataset, searching for products on Klarna, summarizing research papers, and pulling data from a SQL database. Please provide more information about the task you want to accomplish, and I'll guide you through the process.",
            "send_from": "Planner",
            "send_to": "User",
            "attachment_list": [
                {
                    "id": "atta-20231201-043141-6bc4da86",
                    "type": "init_plan",
                    "content": "1. list the available functions"
                },
                {
                    "id": "atta-20231201-043141-6f29f6c9",
                    "type": "plan",
                    "content": "1. list the available functions"
                },
                {
                    "id": "atta-20231201-043141-76186c7a",
                    "type": "current_plan_step",
                    "content": "1. list the available functions"
                }
            ]
        }
    ]
}

tip

If you need to see the intermediate states of the conversation, you need to implement a SessionEventHandler class and pass it at calling session.send_message(user_query, event_handler=your_event_handler). Find more information about the event handler in this section.