Spaces:
Sleeping
Sleeping
File size: 2,227 Bytes
7d80981 | 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 48 49 50 51 52 53 54 55 56 57 58 |
from client import CustomerEnv
from models import CustomerAction
def run_test():
print("🔌 Connecting to Local OpenEnv Server at http://127.0.0.1:8000...\n")
# Initialize the client, then call .sync() to use it in a standard 'with' block
client = CustomerEnv(base_url="http://127.0.0.1:8000")
with client.sync() as env:
# --- 1. RESET ---
print("--- NEW EPISODE ---")
result = env.reset()
print(f"Customer: {result.observation.customer_reply}")
print(f"Initial Reward: {result.reward}")
# --- 2. STEP 1: PROBING ---
print("\n--- AGENT ACTION 1: SPEAK ---")
action1 = CustomerAction(
action_type="speak",
content="I can help with that. Could I please get your account name?"
)
print(f"Agent: {action1.content}")
result = env.step(action1)
print(f"Customer Reply: {result.observation.customer_reply}")
print(f"Reward (Should be slightly negative for turns): {result.reward}")
# --- 3. STEP 2: TOOL USAGE ---
print("\n--- AGENT ACTION 2: TOOL CALL ---")
action2 = CustomerAction(
action_type="tool_call",
content="lookup_account",
tool_args={"name": "John Doe"}
)
print(f"Agent [Action]: Using tool '{action2.content}'")
result = env.step(action2)
print(f"System Response: {result.observation.tool_response}")
print(f"Reward (Should increase for tool usage): {result.reward}")
# --- 4. STEP 3: END CALL (Triggers Judge) ---
print("\n--- AGENT ACTION 3: END EPISODE ---")
action3 = CustomerAction(
action_type="end_call",
content="Thank you, your issue is resolved."
)
result = env.step(action3)
print(f"Episode Done: {result.done}")
print(f"Final Step Reward: {result.reward}")
print(f"Hidden Intent was: {result.observation.metadata.get('hidden_intent')}")
# Print the full transcript tracked by the environment
print("\n--- FULL CONVERSATION TRANSCRIPT ---")
print(result.observation.conversation_history)
if __name__ == "__main__":
run_test() |