Update backend.py
Browse files- backend.py +4 -1
backend.py
CHANGED
|
@@ -33,6 +33,9 @@ def generate_testcases(user_story):
|
|
| 33 |
|
| 34 |
# Few-shot learning examples to guide the model
|
| 35 |
few_shot_examples = """
|
|
|
|
|
|
|
|
|
|
| 36 |
Example 1:
|
| 37 |
User Story:
|
| 38 |
Ensure the "Key Details" feature is enabled in the backend.
|
|
@@ -81,7 +84,7 @@ def generate_testcases(user_story):
|
|
| 81 |
"""
|
| 82 |
|
| 83 |
# Combine the few-shot examples with the user story for the model to process
|
| 84 |
-
prompt = few_shot_examples + f"\nUser Story: {user_story}\n"
|
| 85 |
|
| 86 |
try:
|
| 87 |
# Call the Nvidia llama API with the refined prompt
|
|
|
|
| 33 |
|
| 34 |
# Few-shot learning examples to guide the model
|
| 35 |
few_shot_examples = """
|
| 36 |
+
|
| 37 |
+
"App is Tech360 iOS generate more testcases"
|
| 38 |
+
|
| 39 |
Example 1:
|
| 40 |
User Story:
|
| 41 |
Ensure the "Key Details" feature is enabled in the backend.
|
|
|
|
| 84 |
"""
|
| 85 |
|
| 86 |
# Combine the few-shot examples with the user story for the model to process
|
| 87 |
+
prompt = few_shot_examples + f"\nUser Story: {user_story}\n"
|
| 88 |
|
| 89 |
try:
|
| 90 |
# Call the Nvidia llama API with the refined prompt
|