update

app.py

@@ -16,20 +16,19 @@ class ContentFormatter:
         return json.dumps(data)
 
 class AzureAgent:
-    def __init__(self, api_key, azure_uri, deployment_name, api_version):
+    def __init__(self, api_key, azure_uri, deployment_name):
         self.azure_uri = azure_uri
         self.headers = {
             'Authorization': f"Bearer {api_key}",
             'Content-Type': 'application/json'
         }
         self.deployment_name = deployment_name
-        self.api_version = api_version
         self.chat_formatter = ContentFormatter
 
     def invoke(self, text, **kwargs):
         body = self.chat_formatter.chat_completions(text, {**kwargs})
         conn = http.client.HTTPSConnection(self.azure_uri)
-        conn.request("POST", f'/v{self.api_version}/chat/completions', body=body, headers=self.headers)
+        conn.request("POST", f'/v1/chat/completions', body=body, headers=self.headers)
         response = conn.getresponse()
         data = response.read()
         conn.close()
@@ -68,7 +67,7 @@ model_type = st.sidebar.radio("Select the type of agent", ('AzureAgent', 'GPTAge
 api_key = st.sidebar.text_input("API Key", type="password")
 endpoint_url = st.sidebar.text_input("Endpoint URL")
 deployment_name = st.sidebar.text_input("Model Name")
-api_version = st.sidebar.text_input("API Version", '2022-11-01')  # Default API version
+api_version = st.sidebar.text_input("API Version", '2024-02-15-preview')  # Default API version
 
 # Model invocation parameters
 temperature = st.sidebar.slider("Temperature", min_value=0.0, max_value=1.0, value=0.5, step=0.01)
@@ -86,7 +85,7 @@ if uploaded_file is not None:
     # Process data button
     if st.button('Process Data'):
         if model_type == 'AzureAgent':
-            agent = AzureAgent(api_key, endpoint_url, deployment_name, api_version)
+            agent = AzureAgent(api_key, endpoint_url, deployment_name)
         else:
             agent = GPTAgent(api_key, endpoint_url, deployment_name, api_version)