Update backend.py

backend.py

@@ -4,12 +4,12 @@ import io
 import requests
 import html  # For escaping HTML characters
 from bs4 import BeautifulSoup
-from openai import OpenAI
+from openai import OpenAI 
 
 # Initialize OpenAI API with Nvidia's Mistral model
 client = OpenAI(
-    base_url="https://api.llama-api.com",
-    api_key="LA-38879bbd57704371a151ce66b6186aa6e95ab69a3b974f5ca4459f2ce89c4de1"
+    base_url="https://integrate.api.nvidia.com/v1",
+    api_key="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"
 )
 
 def clean_test_case_output(text):
@@ -86,7 +86,7 @@ def generate_testcases(user_story):
     try:
         # Call the Nvidia Mistral API with the refined prompt
         completion = client.chat.completions.create(
-            model="llama3.1-405b",  # Using Mistral model
+            model="nv-mistralai/mistral-nemo-12b-instruct",  # Using Mistral model
             messages=[
                 {"role": "user", "content": prompt}
             ],