File size: 1,860 Bytes
dc97fe1 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 | import json
import sys
import os
# Add parent directory to path to import backend
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
from backend.env import CustomerSupportEnv
from backend.models import Action
def test_kb_and_sentiment():
env = CustomerSupportEnv()
print("--- Testing Reset ---")
obs = env.reset()
ticket_text = obs.state["ticket_text"]
print(f"Initial Ticket: {ticket_text}")
print(f"Initial Sentiment: {obs.state['sentiment']}")
print("\n--- Testing KB Search ---")
action = Action(action_type="search_kb", payload={"query": "refund policy"})
obs, reward, done, info = env.step(action)
print(f"Message: {info['message']}")
print(f"KB Context in Obs: {obs.state.get('kb_context')}")
print("\n--- Testing Sentiment Decay ---")
# Take 3 more steps to trigger sentiment change
for i in range(2):
action = Action(action_type="generate_response", payload={"response": "Wait..."})
obs, reward, done, info = env.step(action)
print(f"Step {i+2} Sentiment: {obs.state['sentiment']}")
# 4th step should trigger decay from initial (which was likely ANGRY/NEUTRAL etc)
action = Action(action_type="generate_response", payload={"response": "Almost there..."})
obs, reward, done, info = env.step(action)
print(f"Step 4 Sentiment: {obs.state['sentiment']}")
print(f"Message: {info['message']}")
print("\n--- Testing Clarification ---")
# Force a vague scenario for testing if needed, or just test the action
action = Action(action_type="ask_clarification", payload={"question": "What is wrong?"})
obs, reward, done, info = env.step(action)
print(f"Is Clarified in Obs: {obs.state.get('is_clarified')}")
print(f"Message: {info['message']}")
if __name__ == "__main__":
test_kb_and_sentiment()
|