43 lines
1.7 KiB
Python
43 lines
1.7 KiB
Python
import sys
|
|
import os
|
|
import time
|
|
from simulation.sim_base import BaseSimulation, run_sim
|
|
|
|
class AISettingsSimulation(BaseSimulation):
|
|
def run(self):
|
|
print("\n--- Running AI Settings Simulation ---")
|
|
|
|
# 1. Verify initial model (Gemini by default)
|
|
provider = self.client.get_value("current_provider")
|
|
model = self.client.get_value("current_model")
|
|
print(f"[Sim] Initial Provider: {provider}, Model: {model}")
|
|
|
|
# 2. Switch to Anthropic
|
|
print("[Sim] Switching to Anthropic...")
|
|
self.client.set_value("current_provider", "anthropic")
|
|
# Need to set a valid model for Anthropic too
|
|
anthropic_model = "claude-3-5-sonnet-20241022"
|
|
self.client.set_value("current_model", anthropic_model)
|
|
time.sleep(1)
|
|
|
|
# Verify
|
|
new_provider = self.client.get_value("current_provider")
|
|
new_model = self.client.get_value("current_model")
|
|
print(f"[Sim] Updated Provider: {new_provider}, Model: {new_model}")
|
|
assert new_provider == "anthropic", f"Expected 'anthropic', got {new_provider}"
|
|
assert new_model == anthropic_model, f"Expected {anthropic_model}, got {new_model}"
|
|
|
|
# 3. Switch back to Gemini
|
|
print("[Sim] Switching back to Gemini...")
|
|
self.client.set_value("current_provider", "gemini")
|
|
gemini_model = "gemini-2.0-flash"
|
|
self.client.set_value("current_model", gemini_model)
|
|
time.sleep(1)
|
|
|
|
final_provider = self.client.get_value("current_provider")
|
|
print(f"[Sim] Final Provider: {final_provider}")
|
|
assert final_provider == "gemini", f"Expected 'gemini', got {final_provider}"
|
|
|
|
if __name__ == "__main__":
|
|
run_sim(AISettingsSimulation)
|