fix: Merge pull request #20 from abolfazl8131/refactor

refactor

app/main.py

@@ -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/models/__init__.py

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

app/models/helm_models.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/terraform_models.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'

app/models/utils.py

@@ -0,0 +1,11 @@
+from pydantic import BaseModel
+from typing import List, Optional
+
+class Output(BaseModel):
+    output:str
+
+class BasicInput(BaseModel):
+
+    max_tokens:int = 500
+    min_tokens:int = 100
+    service:str