Update openai_client.py

openai_client.py

@@ -11,20 +11,21 @@ client = OpenAI(
     base_url=os.getenv("https://api.aimlapi.com"),
 )
 
+# Initialize the Anthropic client
+anthropic_client = anthropic.Anthropic(api_key=os.getenv("ANTHROPIC_API_KEY"))
 
 # Function to get GPT-4o Mini response
 def get_code_review_response(prompt, max_tokens=1000):
     try:
-        response = client.chat.completions.create(
-            model="gpt-4o-mini",
+        response = anthropic_client.messages.create(
+            model="claude-3-5-sonnet-20240620",
+            max_tokens=1024,
             messages=[
                 {
-                    "role": "system",
-                    "content": "You are an AI assistant who helps users in code reviews by deep thinking in points max 5-6 point shortly.",
+                    "role": "user",
+                    "content": f"You are an AI assistant who helps users in code reviews by deep thinking in points max 5-6 point shortly:\n{prompt}",
                 },
-                {"role": "user", "content": prompt},
             ],
-            max_tokens=max_tokens,
         )
         return response.choices[0].message.content
     except Exception as e:
@@ -34,16 +35,13 @@ def get_code_review_response(prompt, max_tokens=1000):
 # Function to refactor code
 def refactor_code(code_snippet):
     try:
-        response = client.chat.completions.create(
-            model="gpt-4o-2024-08-06",
+        response = anthropic_client.messages.create(
+            model="claude-3-5-sonnet-20240620",
+            max_tokens=1024,
             messages=[
-                {
-                    "role": "system",
-                    "content": "You are an expert code refactoring assistant.",
-                },
                 {
                     "role": "user",
-                    "content": f"Refactor the following code. Do not provide any explanation or comments, just return the refactored code.\n{code_snippet}",
+                    "content": f"Refactor the following code. Do not provide any explanation or comments, just return the refactored code:\n{code_snippet}",
                 },
             ],
         )
@@ -53,10 +51,6 @@ def refactor_code(code_snippet):
         return "Sorry, an error occurred while refactoring your code. Please try again later."
 
 
-# Initialize the Anthropic client
-anthropic_client = anthropic.Anthropic(api_key=os.getenv("ANTHROPIC_API_KEY"))
-
-
 # Function to get feedback on code using Anthropic
 def code_feedback(code_snippet):
     try:
@@ -129,13 +123,10 @@ def suggest_best_practices(code_snippet):
 # Function to remove code errors
 def remove_code_errors(code_snippet):
     try:
-        response = client.chat.completions.create(
-            model="gpt-4o-mini",
+        response = anthropic_client.messages.create(
+            model="claude-3-5-sonnet-20240620",
+            max_tokens=1024,
             messages=[
-                {
-                    "role": "system",
-                    "content": "You are an expert in debugging code. Provide concise suggestions to remove errors from the following code snippet.",
-                },
                 {
                     "role": "user",
                     "content": f"Identify and suggest fixes for errors in the following code:\n{code_snippet}",