Update backend.py

backend.py

@@ -6,7 +6,7 @@ import html  # For escaping HTML characters
 from bs4 import BeautifulSoup
 from openai import OpenAI 
 
-# Initialize OpenAI API with Nvidia's Mistral model
+# Initialize OpenAI API with Nvidia's llama model
 client = OpenAI(
     base_url="https://integrate.api.nvidia.com/v1",
     api_key="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"
@@ -24,7 +24,7 @@ def clean_test_case_output(text):
 def generate_testcases(user_story):
     """
     Generates advanced QA test cases based on a provided user story by interacting 
-    with Nvidia's Mistral model API. The prompt is refined for clarity, 
+    with Nvidia's llama model API. The prompt is refined for clarity, 
     and the output is processed for better quality.
     
     :param user_story: A string representing the user story for which to generate test cases.
@@ -84,9 +84,9 @@ def generate_testcases(user_story):
     prompt = few_shot_examples + f"\nUser Story: {user_story}\n"
 
     try:
-        # Call the Nvidia Mistral API with the refined prompt
+        # Call the Nvidia llama API with the refined prompt
         completion = client.chat.completions.create(
-            model="meta/llama-3.1-405b-instruct",  # Using Mistral model
+            model="meta/llama-3.1-405b-instruct",  # Using llama3.1 405b model
             messages=[
                 {"role": "user", "content": prompt}
             ],