fix(token): fix openai token input for basic and bugfix sections

Author: abolfazl8131

app/gpt_services.py

@@ -2,10 +2,10 @@
 from openai import OpenAI
 from fastapi import HTTPException
 
-def gpt_service(prompt):
+def gpt_service(prompt,token):
 
     try:
-        client = OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))
+        client = OpenAI(api_key=token)
         chat_completion = client.chat.completions.create(
             messages=[
                 {

app/models/terraform_models.py

@@ -5,44 +5,27 @@
 
 class IaCBasicInput(BasicInput):
     input:str
-    service:Optional[str] = 'terraform'
+    token:str
 
     @validator("input")
     def validate_input(cls, value):
         if not value:
             raise ValueError("Input cannot be empty.")
         return value
 
-    @validator("service")
-    def validate_service(cls, value):
-        allowed_services = ['terraform']
-        if value not in allowed_services:
-            raise ValueError(f"Service must be one of {allowed_services}.")
-        return value
+    
 
 class IaCBugfixInput(BasicInput):
     bug_description:str
-    version:str = 'latest'
-    service:Optional[str] = 'terraform'
+    token:str
 
     @validator("bug_description")
     def validate_bug_description(cls, value):
         if not value:
             raise ValueError("Bug description cannot be empty.")
         return value
 
-    @validator("version")
-    def validate_version(cls, value):
-        if not value:
-            raise ValueError("Version cannot be empty.")
-        return value
-
-    @validator("service")
-    def validate_service(cls, value):
-        allowed_services = ['terraform']
-        if value not in allowed_services:
-            raise ValueError(f"Service must be one of {allowed_services}.")
-        return value
+   
 
 class IaCInstallationInput(BaseModel):
     os:str = "Ubuntu"

app/prompt_generators.py

@@ -5,21 +5,15 @@
 def IaC_basics_generator(input : IaCBasicInput) -> str:
 
     prompt = f"""
-                Write a robust answer about {input.service},
-                focusing on the latest update of {input.service} and based on this question:{input.input},
-                minimun length of answer is {input.min_tokens} and maximum length is {input.max_tokens}
-
+                {input.input}
             """
     return prompt
 
 
 def IaC_bugfix_generator(input : IaCBugfixInput) -> str:
 
     prompt = f"""
-                Write a clear answer to debug {input.service}
-                focusing on the version {input.version} of {input.service} and based on this bug:{input.bug_description},
-                generate a correct code that help us to solve this bug.
-                minimum length of answer is {input.min_tokens} and maximum length is {input.max_tokens}
+                {input.bug_description}
 
             """
     return prompt

app/routes/terraform.py

@@ -38,15 +38,15 @@ async def IaC_basic_generation(request:IaCBasicInput) -> Output:
         if os.environ.get("TEST"):
             return Output(output='Terraform developed by hashicorp and it is very usefull')
         generated_prompt = IaC_basics_generator(request)
-        output = gpt_service(generated_prompt)
+        output = gpt_service(generated_prompt,request.token)
         return Output(output=output)
 
 @app.post("/api/IaC-bugfix/")
 async def IaC_bugfix_generation(request:IaCBugfixInput) -> Output:
         if os.environ.get("TEST"):
             return Output(output='fix this bug by adding x to the y')
         generated_prompt = IaC_bugfix_generator(request)
-        output = gpt_service(generated_prompt)
+        output = gpt_service(generated_prompt,request.token)
         return Output(output=output)