ArvindSelvaraj commited on
Commit
a76211f
1 Parent(s): e71685b

Update backend.py

Browse files
Files changed (1) hide show
  1. backend.py +2 -1
backend.py CHANGED
@@ -6,7 +6,7 @@ import html # For escaping HTML characters
6
  from bs4 import BeautifulSoup
7
  from openai import OpenAI
8
 
9
- # Initialize OpenAI API with Nvidia's llama model
10
  client = OpenAI(
11
  base_url="https://integrate.api.nvidia.com/v1",
12
  api_key="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"
@@ -38,6 +38,7 @@ def generate_testcases(user_story):
38
  "Generate as many as testcases possible minimum 6 ,maximum it can be anything"
39
  "Understand the story thoroughly"
40
  "If it's a DropBury or ODAC Portal User Story, then we perform testing in ODAC Portal"
 
41
  """
42
 
43
  # Combine the few-shot examples with the user story for the model to process
 
6
  from bs4 import BeautifulSoup
7
  from openai import OpenAI
8
 
9
+ # Initialize OpenAI API with Nvidia's Llama model
10
  client = OpenAI(
11
  base_url="https://integrate.api.nvidia.com/v1",
12
  api_key="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"
 
38
  "Generate as many as testcases possible minimum 6 ,maximum it can be anything"
39
  "Understand the story thoroughly"
40
  "If it's a DropBury or ODAC Portal User Story, then we perform testing in ODAC Portal"
41
+ "Give minimum 5 steps for each testcase,steps should be valid and correct"
42
  """
43
 
44
  # Combine the few-shot examples with the user story for the model to process