backend: store the text user selected as interesting snippet

fedora-copr · Mar 26, 2024 · 72fdd81 · 72fdd81
1 parent a36e8e8
commit 72fdd81
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 4 deletions.
diff --git a/backend/src/schema.py b/backend/src/schema.py
@@ -51,6 +51,7 @@ class SnippetSchema(BaseModel):
     start_index: int
     end_index: int
     user_comment: str
+    text: str
 
     # def _splitter(self, return_lines: bool) -> tuple[int, int]:
     #     fst, snd = self.log_part.split("-")

diff --git a/backend/tests/conftest.py b/backend/tests/conftest.py
@@ -118,7 +118,12 @@ def spec_feedback_input_output_schema_tuple():
                 "name": "log1",
                 "content": "log content 1",
                 "snippets": [
-                    {"start_index": 1, "end_index": 2, "user_comment": "comment1"}
+                    {
+                        "start_index": 1,
+                        "end_index": 2,
+                        "user_comment": "comment1",
+                        "text": "log content 1",
+                    }
                 ],
             },
         ],
@@ -132,7 +137,12 @@ def spec_feedback_input_output_schema_tuple():
                 "name": "log1",
                 "content": "log content 1",
                 "snippets": [
-                    {"start_index": 1, "end_index": 2, "user_comment": "comment1"}
+                    {
+                        "start_index": 1,
+                        "end_index": 2,
+                        "user_comment": "comment1",
+                        "text": "log content 1",
+                    }
                 ],
             }
         },
@@ -151,7 +161,12 @@ def container_feedback_input_output_schema_tuple():
                 "name": "log1",
                 "content": "log content 1",
                 "snippets": [
-                    {"start_index": 1, "end_index": 2, "user_comment": "comment1"}
+                    {
+                        "start_index": 1,
+                        "end_index": 2,
+                        "user_comment": "comment1",
+                        "text": "log content 1",
+                    }
                 ],
             },
         ],
@@ -168,7 +183,12 @@ def container_feedback_input_output_schema_tuple():
                 "name": "log1",
                 "content": "log content 1",
                 "snippets": [
-                    {"start_index": 1, "end_index": 2, "user_comment": "comment1"}
+                    {
+                        "start_index": 1,
+                        "end_index": 2,
+                        "user_comment": "comment1",
+                        "text": "log content 1",
+                    }
                 ],
             }
         },