Added some cases for dialogue2dialogue, topic_to_graph

deeppavlov · Nov 11, 2024 · e7aa17f · e7aa17f
1 parent 6dd403a
commit e7aa17f
Showing 1 changed file with 179 additions and 2 deletions.
diff --git a/dev_packages/chatsky_llm_autoconfig/chatsky_llm_autoconfig/autometrics/test_data/data.json b/dev_packages/chatsky_llm_autoconfig/chatsky_llm_autoconfig/autometrics/test_data/data.json
@@ -408,6 +408,183 @@
             }
         }
     ],
-    "dialogue_to_dialogue": [],
-    "graph_to_graph": []
+    "dialogue_to_dialogue": [
+        {
+            "dialogue": [
+                {
+                    "text": "How can I help?",
+                    "participant": "assistant"
+                },
+                {
+                    "text": "I need to make an order",
+                    "participant": "user"
+                },
+                {
+                    "text": "Which books would you like to order?",
+                    "participant": "assistant"
+                },
+                {
+                    "text": "I need Crime and Punishment",
+                    "participant": "user"
+                },
+                {
+                    "text": "Added it to your cart. Anything else?",
+                    "participant": "assistant"
+                },
+                {
+                    "text": "No, that's all",
+                    "participant": "user"
+                },
+                {
+                    "text": "Okay, everything is done, you can go to cart and finish the order.",
+                    "participant": "assistant"
+                },
+                {
+                    "text": "Thanks",
+                    "participant": "user"
+                },
+                {
+                    "text": "Goodbye",
+                    "participant": "assistant"
+                }
+            ]
+        },
+        {
+            "dialogue": [
+                {
+                    "text": "How can I help?",
+                    "participant": "assistant"
+                },
+                {
+                    "text": "I don't know what to do",
+                    "participant": "user"
+                }
+            ]
+        },
+        {
+            "dialogue": [
+                {
+                    "text": "How can I help?",
+                    "participant": "assistant"
+                },
+                {
+                    "text": "I need to make an order",
+                    "participant": "user"
+                },
+                {
+                    "text": "Which books would you like to order?",
+                    "participant": "assistant"
+                },
+                {
+                    "text": "One War and Piece in hard cover and one Pride and Prejudice",
+                    "participant": "user"
+                },
+                {
+                    "text": "Please enter delivery method would you like to use? We currently offer pickup or home delivery.",
+                    "participant": "assistant"
+                },
+                {
+                    "text": "I will pickup my order",
+                    "participant": "user"
+                },
+                {
+                    "text": "Please, enter the payment method you would like to use: cash or credit card.",
+                    "participant": "assistant"
+                },
+                {
+                    "text": "Cash",
+                    "participant": "user"
+                },
+                {
+                    "text": "Success",
+                    "participant": "assistant"
+                }
+            ]
+        }
+    ],
+    "graph_to_graph": [],
+    "dialogue_to_graph": [],
+    "topic_to_graph": [
+        {
+            "topic": "purchase in a book store",
+            "graph": {
+                "edges": [
+                    {
+                        "utterances": [
+                            "I want to order from you"
+                        ],
+                        "source": 1,
+                        "target": 2
+                    },
+                    {
+                        "utterances": [
+                            "One War and Piece in hard cover and one Pride and Prejudice"
+                        ],
+                        "source": 2,
+                        "target": 3
+                    },
+                    {
+                        "utterances": [
+                            "Delivery to the address: Pine Driveway, 45"
+                        ],
+                        "source": 3,
+                        "target": 4
+                    },
+                    {
+                        "utterances": [
+                            "With credit card, please"
+                        ],
+                        "source": 4,
+                        "target": 5
+                    }
+                ],
+                "nodes": [
+                    {
+                        "id": 1,
+                        "label": "start",
+                        "is_start": true,
+                        "utterances": [
+                            "How can I help?"
+                        ]
+                    },
+                    {
+                        "id": 2,
+                        "label": "ask_item",
+                        "is_start": false,
+                        "utterances": [
+                            "Which books would you like to order?"
+                        ]
+                    },
+                    {
+                        "id": 3,
+                        "label": "ask_delivery",
+                        "is_start": false,
+                        "utterances": [
+                            "Please enter delivery method would you like to use? We currently offer pickup or home delivery."
+                        ]
+                    },
+                    {
+                        "id": 4,
+                        "label": "ask_payment_method",
+                        "is_start": false,
+                        "utterances": [
+                            "Please, enter the payment method you would like to use: cash or credit card."
+                        ]
+                    },
+                    {
+                        "id": 5,
+                        "label": "success",
+                        "is_start": false,
+                        "utterances": [
+                            "Done"
+                        ]
+                    }
+                ]
+            }
+        },
+        {
+            "topic": "",
+            "graph": {}
+        }
+    ]
 }