refactor

Author: abolfazl8131

app/__pycache__/main.cpython-311.pyc

@@ -4,7 +4,7 @@
         write_bugfix,
         write_installation,
         edit_directory_generator,execute_pythonfile)
-from .models import (IaCBasicInput,
+from app.models import (IaCBasicInput,
         IaCBugfixInput, 
         Output,
         IaCInstallationInput,IaCTemplateGeneration,HelmTemplateGeneration)

app/__pycache__/prompt_generators.cpython-311.pyc

@@ -0,0 +1,3 @@
+from .helm_models import *
+from .terraform_models import *
+from .utils import *

app/main.py

@@ -0,0 +1,12 @@
+from pydantic import BaseModel
+from typing import List, Optional
+
+class Pod(BaseModel):
+    name:str
+    image:str
+    target_port:int
+
+class HelmTemplateGeneration(BaseModel):
+    api_version:int = 1
+    pods:List[Pod]
+    

app/models/__init__.py

@@ -0,0 +1,23 @@
+from pydantic import BaseModel
+from typing import List, Optional
+
+from .utils import BasicInput
+
+
+class IaCBasicInput(BasicInput):
+    input:str
+    service:Optional[str] = 'terraform'
+
+class IaCBugfixInput(BasicInput):
+    bug_description:str
+    version:str = 'latest'
+    service:Optional[str] = 'terraform'
+
+class IaCInstallationInput(BaseModel):
+    os:str = "ubuntu"
+    service:Optional[str] = 'terraform'
+
+class IaCTemplateGeneration(BaseModel):
+    CI_integration:bool = True
+    base_config:str = 'ec2'
+    service:str = 'terraform'