Create models.py

models.py

@@ -0,0 +1,23 @@
+from pydantic import BaseModel, Field
+from typing import List
+
+client = OctoAI()
+
+class Article(BaseModel):
+    number: str = Field(description="Số điều, ví dụ như: Điều 1, Điều 2, Điều 3")
+    title: str = Field(description="Tiêu đề điều đứng ngay sau Số điều, ví dụ như: Phạm vi điều chỉnh, Đối tượng áp dụng")
+    content: str = Field(description="Nội dung chương bên dưới tiêu đề điều")
+
+
+class Chapter(BaseModel):
+    number: str = Field(description="Số chương, ví dụ như: Chương 1, Chương 2, Chương 3")
+    title: str = Field(description="Tiêu đề chương đứng bên dưới Số chương, ví dụ như: Những quy định chung, Quyền trách nhiệm của người lao động")
+    content: str = Field(description="Nội dung chương bên dưới tiêu đề chương")
+    articles: List[Article] = Field(description="Danh sách các điều")
+    
+class Law(BaseModel):
+    number: str = Field(description="Luật số hoặc số, ví dụ như: 58/2014/QH13")
+    title: str = Field(description="Tên luật, ví dụ như: Luật bảo hiểm xã hội")
+    publisher: str = Field(description="Cơ quan Phát hành, ví dụ như: Chính phủ, Quốc Hội")
+    document_type: str = Field(description="Loại văn bản, ví dụ như: Luật, Thông tư, Nghị định")
+    chapters: List[Chapter] = Field(description="Danh sách các chương")