+1 -3

tests/integration/test_cortex.py

reviewed

··· 75 75 time.sleep(0.1) 76 76 77 77 # Create a request 78 78 - agent_id = cortex_request( 79 79 - prompt="Test prompt", name="default", provider="openai" 80 80 - ) 78 78 + agent_id = cortex_request(prompt="Test prompt", name="default", provider="openai") 81 79 82 80 time.sleep(0.2) 83 81

+1 -3

tests/test_get_insights.py

reviewed

··· 84 84 segment = utils.get_insights_by_schedule("segment") 85 85 assert len(segment) > 0 86 86 for key, meta in segment.items(): 87 87 - assert ( 88 88 - meta.get("schedule") == "segment" 89 89 - ), f"{key} should have schedule=segment" 87 87 + assert meta.get("schedule") == "segment", f"{key} should have schedule=segment" 90 88 91 89 # Verify no overlap 92 90 assert not set(daily.keys()) & set(

+1 -3

think/providers/google.py

reviewed

··· 653 653 async with create_mcp_client(str(mcp_url)) as mcp: 654 654 # Attach tool logging hooks to the MCP session 655 655 agent_id = config.get("agent_id") 656 656 - tool_hooks = ToolLoggingHooks( 657 657 - callback, agent_id=agent_id, name=name 658 658 - ) 656 656 + tool_hooks = ToolLoggingHooks(callback, agent_id=agent_id, name=name) 659 657 tool_hooks.attach(mcp.session) 660 658 661 659 # Extract allowed tools from config