Skip to content

Commit 1438a3b

Browse files
committed
Update financial templates and adjust dataset configurations for improved performance
1 parent 0b5bdd2 commit 1438a3b

15 files changed

+19940
-6518
lines changed

.DS_Store

0 Bytes
Binary file not shown.

Usage_Examples/basic_usage.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55
import sys
66
import json
77
from dotenv import load_dotenv
8+
TEMPLATE_NAME= "templates/financial01/"
89
TEMPLATE_NAME= "templates/financial_retail_banking/"
910
# Add parent directory to path to import the library
1011
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
@@ -28,8 +29,8 @@ def main():
2829
API_KEY=api_key, # Use API key from environment variable
2930
template_client_prompt=TEMPLATE_NAME + "prompts/financial_client_01.j2",
3031
template_advisor_prompt=TEMPLATE_NAME + "prompts/financial_advisor_01.j2",
31-
file_profiles=TEMPLATE_NAME + "profiles/retail_banking_5.json",
32-
file_output=TEMPLATE_NAME + "outputs/test_dataset01_5b.json",
32+
file_profiles=TEMPLATE_NAME + "profiles/retail_banking_1.json",
33+
file_output=TEMPLATE_NAME + "outputs/test_dataset01_1b.json",
3334
model_client="gpt-4o",
3435
model_advisor="gpt-4o",
3536
# Optional parameters with custom values

templates/.DS_Store

6 KB
Binary file not shown.
6 KB
Binary file not shown.

templates/financial_retail_banking/create_dataset.py

Lines changed: 9 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -23,9 +23,11 @@
2323
# Define the template directory
2424
TEMPLATE_NAME = "templates/financial_retail_banking/"
2525

26-
# Add parent directory to path to import the library
27-
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
26+
# Explicitly point to the project root for importing wizardsdata
27+
PROJECT_ROOT = os.path.abspath(os.path.join(os.path.dirname(__file__), '../..'))
28+
sys.path.insert(0, PROJECT_ROOT)
2829

30+
# Now import the library - this will use the latest version
2931
import wizardsdata as wsd
3032

3133
def main():
@@ -68,16 +70,17 @@ def main():
6870
template_advisor_prompt=TEMPLATE_NAME + "prompts/financial_advisor_01.j2",
6971

7072
# Input/Output file paths
71-
file_profiles=TEMPLATE_NAME + "profiles/retail_banking_1.json",
72-
file_output=TEMPLATE_NAME + "outputs/retail_banking_dataset_11.json",
73+
file_profiles=TEMPLATE_NAME + "profiles/retail_banking_160.json",
74+
file_output=TEMPLATE_NAME + "outputs/retail_banking_dataset_b.json",
7375

7476
# Model configuration
7577
model_client="gpt-4o",
7678
model_advisor="gpt-4o",
7779

7880
# Conversation parameters
79-
temperature_client=0.8,
80-
max_recommended_questions=4
81+
temperature_client=0.6,
82+
temperature_advisor=0.2,
83+
max_recommended_questions=10
8184
)
8285

8386
# Check for missing parameters

templates/financial_retail_banking/outputs/retail_banking_dataset.json

Lines changed: 9963 additions & 0 deletions
Large diffs are not rendered by default.

templates/financial_retail_banking/outputs/retail_banking_dataset_1.json

Lines changed: 0 additions & 5679 deletions
This file was deleted.

templates/financial_retail_banking/outputs/retail_banking_dataset_11.json

Lines changed: 0 additions & 23 deletions
This file was deleted.

templates/financial_retail_banking/outputs/retail_banking_dataset_a.json

Lines changed: 4265 additions & 0 deletions
Large diffs are not rendered by default.

templates/financial_retail_banking/outputs/retail_banking_dataset_b.json

Lines changed: 5700 additions & 0 deletions
Large diffs are not rendered by default.

0 commit comments

Comments
 (0)