Initial Draft

dto/release_notes.py

@@ -11,6 +11,16 @@ class Enhancement(BaseModel):
     benefits: str = Field(description="The benefit or impact on the user or system as a result of the enhancement")
     reason: str = Field(description="The reason for implementing the change.")
 
+    def get_json_template(model: BaseModel):
+        """helper function to get the json schema from the pydantic model"""
+        
+        schema = model.model_json_schema()
+        json_schema = json.dumps(schema)
+        json_schema = Template(json_schema).substitute(
+            {"defs": "definitions", "ref": "$ref"}
+        )
+        return json.loads(json_schema)
+
 
 class ReleaseNotes(BaseModel):
     """Release Notes Pydantic Schema"""
@@ -18,4 +28,14 @@ class ReleaseNotes(BaseModel):
     release_date: str = Field(description="Provide release date",default=datetime.now().strftime('%d-%b-%Y').upper(), const=True)
     product_name: str = Field(description="Provide product name",default="Oracle Banking Retail Lending", const=True)
     summary: str = Field(description="A brief introduction highlighting the key focus of this release")
-    enhancements: List[Enhancement] = Field(description="List of enhancements in this release")
+    enhancements: List[Enhancement] = Field(description="List of enhancements in this release")
+
+    def get_json_template(model: BaseModel):
+        """helper function to get the json schema from the pydantic model"""
+        
+        schema = model.model_json_schema()
+        json_schema = json.dumps(schema)
+        json_schema = Template(json_schema).substitute(
+            {"defs": "definitions", "ref": "$ref"}
+        )
+        return json.loads(json_schema)

dto/requirement_gathering.py

@@ -11,8 +11,28 @@ class RequirementGathering(BaseModel):
     priority: str = Field(description="The priority of the requirement. Return only one from list of values [High, Medium and Low]. Ground result based on the document.")
     tags: List[str] = Field(description="List of tags for the requirement to identify into relevant categories(e.g., Product Specifications, Validations and Quality Standards)")
 
+    def get_json_template(model: BaseModel):
+        """helper function to get the json schema from the pydantic model"""
+        
+        schema = model.model_json_schema()
+        json_schema = json.dumps(schema)
+        json_schema = Template(json_schema).substitute(
+            {"defs": "definitions", "ref": "$ref"}
+        )
+        return json.loads(json_schema)
+
 class RequirementGatheringDetails(BaseModel):
     """Requirement Gathering Details Pydantic Schema"""
 
     header: str = Field(description="Provide the header of the document.")
-    requirements: List[RequirementGathering] = Field(description="List of requirements gathered from the document.")
+    requirements: List[RequirementGathering] = Field(description="List of requirements gathered from the document.")
+
+    def get_json_template(model: BaseModel):
+        """helper function to get the json schema from the pydantic model"""
+        
+        schema = model.model_json_schema()
+        json_schema = json.dumps(schema)
+        json_schema = Template(json_schema).substitute(
+            {"defs": "definitions", "ref": "$ref"}
+        )
+        return json.loads(json_schema)

dto/user_story.py

@@ -8,6 +8,16 @@ class UserStoryScenarios(BaseModel):
     pre_conditions: str = Field(description="Provide a precondition details") 
     action_details: str = Field(description="Provide action details to be taken in business application by the user role specified")
     expected_outcome: str = Field(description="Provide list of activities to be allowed by business user role")
+
+    def get_json_template(model: BaseModel):
+        """helper function to get the json schema from the pydantic model"""
+        
+        schema = model.model_json_schema()
+        json_schema = json.dumps(schema)
+        json_schema = Template(json_schema).substitute(
+            {"defs": "definitions", "ref": "$ref"}
+        )
+        return json.loads(json_schema)
     
 class UserStory(BaseModel):
     """User Story Pydantic Schema"""
@@ -17,3 +27,13 @@ class UserStory(BaseModel):
     feature: str = Field(description="Provide application feature/functionality")
     benefit: str = Field(description="Provide benefits of the feature to the user role")
     user_story_scenarios : List[UserStoryScenarios] = Field(description="Provide list of user story scenarios")
+
+    def get_json_template(model: BaseModel):
+        """helper function to get the json schema from the pydantic model"""
+        
+        schema = model.model_json_schema()
+        json_schema = json.dumps(schema)
+        json_schema = Template(json_schema).substitute(
+            {"defs": "definitions", "ref": "$ref"}
+        )
+        return json.loads(json_schema)