From d8dea2e8a839309a4c54ec9ef10f14a06584864e Mon Sep 17 00:00:00 2001
From: Cyber MacGeddon <cybermaggedon@gmail.com>
Date: Mon, 23 Feb 2026 22:01:20 +0000
Subject: [PATCH] Update tests

---
 tests/unit/test_cli/test_tool_commands.py | 28 +++++++++++++++--------
 1 file changed, 19 insertions(+), 9 deletions(-)

diff --git a/tests/unit/test_cli/test_tool_commands.py b/tests/unit/test_cli/test_tool_commands.py
index f7bdf6ca..9c204614 100644
--- a/tests/unit/test_cli/test_tool_commands.py
+++ b/tests/unit/test_cli/test_tool_commands.py
@@ -406,9 +406,9 @@ class TestShowToolsStructuredQuery:
 
     @patch('trustgraph.cli.show_tools.Api')
     def test_show_mixed_tool_types(self, mock_api_class, mock_api, capsys):
-        """Test displaying multiple tool types including structured-query."""
+        """Test displaying multiple tool types including structured-query and row-embeddings-query."""
         mock_api_class.return_value, mock_config = mock_api
-        
+
         tools = [
             {
                 "name": "ask_knowledge",
@@ -417,37 +417,47 @@ class TestShowToolsStructuredQuery:
                 "collection": "docs"
             },
             {
-                "name": "query_data", 
+                "name": "query_data",
                 "description": "Query structured data",
                 "type": "structured-query",
                 "collection": "sales"
             },
+            {
+                "name": "find_customer",
+                "description": "Find customers by semantic search",
+                "type": "row-embeddings-query",
+                "schema-name": "customers",
+                "collection": "crm"
+            },
             {
                 "name": "complete_text",
                 "description": "Generate text",
                 "type": "text-completion"
             }
         ]
-        
+
         config_values = [
             ConfigValue(type="tool", key=f"tool_{i}", value=json.dumps(tool))
             for i, tool in enumerate(tools)
         ]
         mock_config.get_values.return_value = config_values
-        
+
         show_config("http://test.com")
-        
+
         captured = capsys.readouterr()
         output = captured.out
-        
+
         # All tool types should be displayed
         assert "knowledge-query" in output
-        assert "structured-query" in output  
+        assert "structured-query" in output
+        assert "row-embeddings-query" in output
         assert "text-completion" in output
-        
+
         # Collections should be shown for appropriate tools
         assert "docs" in output  # knowledge-query collection
         assert "sales" in output  # structured-query collection
+        assert "crm" in output  # row-embeddings-query collection
+        assert "customers" in output  # row-embeddings-query schema-name
 
     def test_show_main_parses_args_correctly(self):
         """Test that show main() parses arguments correctly."""