Skip to content

Commit 1c5fb70

Browse files
Fix agent tool wiring and vector index cleanup
1 parent d50e975 commit 1c5fb70

3 files changed

Lines changed: 10 additions & 5 deletions

File tree

‎tests/agents/test_3800_agente2e.py‎

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -408,7 +408,7 @@ def test_3800_agent_end_to_end(
408408
"3. If defective: "
409409
" a. Process the defective return."
410410
),
411-
tools=["Websearch", "Email"],
411+
tools=[websearch_tool_name, email_tool_name],
412412
enable_human_tool=False,
413413
),
414414
)
@@ -417,11 +417,16 @@ def test_3800_agent_end_to_end(
417417
log_object_details("create_task", "task", task)
418418

419419
assert task.task_name == task_name
420-
assert set(task.attributes.tools) == {"Websearch", "Email"}
420+
assert {
421+
tool.split("_", 1)[0] for tool in task.attributes.tools
422+
} == {"Websearch", "Email"}
421423
assert task.attributes.enable_human_tool is False
422424

423425
assert task.task_name == task_name
424-
assert set(task.attributes.tools) == {"Websearch", "Email"}
426+
assert {tool.split("_", 1)[0] for tool in task.attributes.tools} == {
427+
"Websearch",
428+
"Email",
429+
}
425430

426431
# -------------------------------
427432
# TEAM

‎tests/vector_index/test_5100_async_drop_index.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -341,7 +341,7 @@ async def test_5104(self):
341341
await self.async_vector_index.delete(include_data=True, force=True)
342342
await asyncio.sleep(1)
343343
await self.assert_index_count(f"^{self.index_name}", 0)
344-
table_name = "TEST_VECTOR_INDEX$VECTAB"
344+
table_name = f"{self.index_name}$VECTAB".upper()
345345
async with select_ai.async_cursor() as cursor:
346346
await cursor.execute(
347347
"""

‎tests/vector_index/test_5100_drop_index.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -358,7 +358,7 @@ def test_5104(self):
358358
self.vector_index.delete(include_data=True, force=True)
359359
time.sleep(1)
360360
self.assert_index_count(f"^{self.index_name}", 0)
361-
table_name = "TEST_VECTOR_INDEX$VECTAB"
361+
table_name = f"{self.index_name}$VECTAB".upper()
362362
with select_ai.cursor() as cursor:
363363
cursor.execute(
364364
"""

0 commit comments

Comments
 (0)