Build a knowledge graph from the api calls

[openchatai]

Build a knowledge graph from the api calls

[codebanesr]

Trello Example

import networkx as nx
def create_trello_graph(json_data):
    # Create an empty directed graph
    graph = nx.DiGraph()

    def traverse_list(list_data, parent_board_id):
        for card in list_data.get("cards", []):
            # Add cards as nodes and connect them to their parent lists
            graph.add_node(card["id"], label="Card", title=card["title"], description=card["description"])
            graph.add_edge(parent_board_id, card["id"], label="Contains")

    for board in json_data.get("boards", []):
        # Add boards as nodes
        graph.add_node(board["id"], label="Board", title=board["title"])
        
        for trello_list in board.get("lists", []):
            # Add lists as nodes and connect them to their parent boards
            graph.add_node(trello_list["id"], label="List", title=trello_list["title"])
            graph.add_edge(board["id"], trello_list["id"], label="Contains")
            traverse_list(trello_list, board["id"])

    return graph

# Example JSON data

trello_data = {
    "boards": [
        {
            "id": "board1",
            "title": "Project X",
            "lists": [
                {
                    "id": "list1",
                    "title": "To Do",
                    "cards": [
                        {
                            "id": "card1",
                            "title": "Task 1",
                            "description": "This is the first task."
                        }
                    ]
                }
            ]
        }
    ]
}

# Create and display the graph
graph = create_trello_graph(trello_data)
print("Nodes:", graph.nodes(data=True))
print("Edges:", graph.edges(data=True))

[codebanesr]

Input: To create a Trello card in the "ToDo" list of the "trello" board, the agent traverses the graph structure.
Agent locates the board with the name "trello" and then identifies the "ToDo" list. Subsequently, it initiates the "generate_openapi" function call to complete the task.

It's important to note that this functionality has been tested and verified using a non-optimized version of the code found in the "integration" folder.