WIP: PAIN4

tests/test_live_workflow.py

@@ -44,7 +44,6 @@ def test_full_live_workflow(live_gui) -> None:
  success = False
  for _ in range(10):
   proj = client.get_project()
-  # check if name matches 'temp_project'
   if proj.get('project', {}).get('project', {}).get('name') == 'temp_project':
    success = True
    break
@@ -61,14 +60,10 @@ def test_full_live_workflow(live_gui) -> None:
 
  # Enable auto-add so the response ends up in history
  client.set_value("auto_add_history", True)
- client.set_value("current_provider", "gemini_cli")
+ client.set_value("current_provider", "gemini")
  
- mock_path = f'"{sys.executable}" "{os.path.abspath("tests/mock_gemini_cli.py")}"'
- print(f"[TEST] Setting gcli_path to {mock_path}...")
- client.set_value("gcli_path", mock_path)
- assert wait_for_value(client, "gcli_path", mock_path)
- 
- client.set_value("current_model", "gemini-2.0-flash")
+ # USE gemini-2.5-flash-lite
+ client.set_value("current_model", "gemini-2.5-flash-lite")
  time.sleep(1)
 
  # 3. Discussion Turn
@@ -105,7 +100,9 @@ def test_full_live_workflow(live_gui) -> None:
  for i in range(60):
   session = client.get_session()
   entries = session.get('session', {}).get('entries', [])
-  if any(e.get('role') == 'AI' for e in entries):
+  # Check for AI role. The entries are objects with a 'role' key.
+  found_ai = any(str(e.get('role', '')).upper() == 'AI' for e in entries)
+  if found_ai:
    success = True
    print(f"  AI response found in history after {i}s")
    break
@@ -116,7 +113,7 @@ def test_full_live_workflow(live_gui) -> None:
    pytest.fail(f"AI Status went to error during response wait. Response: {state.get('ai_response')}")
    
   time.sleep(1)
- assert success, "AI failed to respond or response not added to history"
+ assert success, f"AI failed to respond or response not added to history. Entries: {client.get_session()}"
 
  # 5. Switch Discussion
  print("[TEST] Creating new discussion 'AutoDisc'...")