config related change

.gitignore

@@ -1,2 +1,3 @@
 __pycache__/
+.env
 

auth.py

@@ -1,11 +1,15 @@
 from fastapi.security.api_key import APIKeyHeader
 from fastapi import Security, HTTPException
 import os
+# from dotenv import load_dotenv
 
 
 api_key_header = APIKeyHeader(name="x-api-key", auto_error=False)
 
 async def api_key_auth(api_key: str = Security(api_key_header)):
+    # for local
+    # load_dotenv()
+
     API_Keys = [os.environ.get('API_KEY')]
     if api_key not in API_Keys:
         raise HTTPException(status_code=401, detail="Missing or invalid API key")

client.py

@@ -1,53 +1,57 @@
-from openai import AzureOpenAI
+from openai import AzureOpenAI, OpenAI
 import os
 import model
+# from dotenv import load_dotenv
 
 def create_client():
     print("In create client")
-    client = AzureOpenAI(
-        api_key=os.environ.get('OPENAI_API_KEY'),  
-        api_version=os.environ.get('MODEL_VERSION'),
-        azure_endpoint = os.environ.get('END_POINT')
-    )
+    
+    # for local
+    # load_dotenv()
+    
+    # if using openai
+    client = OpenAI()
+
+    # if using azureopenai
+    # client = AzureOpenAI()
+
     return client
 
 def get_completion(ticket:model.Ticket):
     print("In get completion")
+    
     # hardcoding response for testing
-    result = "intent= Complaint\nsuggested_response= Issue Refund\nsummary= The customer is making a complaint. Suggested resolution is to issue a refund.\nmail=\nDear Customer,\n\nWe are sorry to hear that you have a complaint regarding your booking. We understand the inconvenience this has caused you and we would like to make things right. Our records show that you have requested a refund.\n\nWe have processed your refund request and you should receive the refund within 3-5 business days. We apologize"
-    # client = create_client()
-    # prompt = create_prompt(ticket.subject, ticket.description)
-    # response = client.completions.create(
-    #     model=os.environ.get('MODEL_NAME'),
-    #     prompt=prompt,
-    #     max_tokens=100,
-    #     n=1,
-    #     stop=None,
-    #     temperature=0.7
-    #     )
-    # result = response.choices[0].text.strip()
-    # print(result)
+    # result = "intent= Complaint\nsuggested_response= Issue Refund\nsummary= The customer is making a complaint. Suggested resolution is to issue a refund.\nmail=\nDear Customer,\n\nWe are sorry to hear that you have a complaint regarding your booking. We understand the inconvenience this has caused you and we would like to make things right. Our records show that you have requested a refund.\n\nWe have processed your refund request and you should receive the refund within 3-5 business days. We apologize"
+    # return result
+    
+    client = create_client()
+    prompt = create_prompt(ticket.subject, ticket.description)
+    response = client.completions.create(
+        model=os.environ.get('MODEL_NAME'),
+        prompt=prompt,
+        max_tokens=500,
+        n=1,
+        stop=None,
+        temperature=0.5
+        )
+    result = response.choices[0].text.strip()
+    print(result)
     return result
 
 def create_prompt(subject:str, desc:str):
     print("In create Prompt")
     # add booking data here
-    data = ""
     prompt = '''
         You are an AI Customer Service Agent for booking.com. Please use the below instructions to answer.
         What is the intent of this request ```
-        Subject: {{subject}} Description: {{desc}}``` in 1 single word? Use these options for intent : 
+        Subject: {{subject}} Description: {{desc}}``` in 1 single word? Use these options for intent: 
 
         Complaint
-        Information Request
         Service Request
-        Cancellation
-        Change
+        Cancellation Request
         Refund Request
 
-        In the context of the previous request, 
-        If the intent is Information Request, use the ``` Booking Details: {{data}}``` to provide information as part of resolution response. 
-        otherwise Suggest a resolution response to address the customer query from the following actions list:
+        In the context of the previous request, Suggest a resolution response to address the customer query from the following actions list:
 
         Issue Refund
         Cancellation