From 1bcb0df0a1089cf4489edcc9bd2ef51e9040bf74 Mon Sep 17 00:00:00 2001 From: Carson Date: Wed, 11 Dec 2024 15:21:40 -0600 Subject: [PATCH] Update tests --- tests/test_content_tools.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/tests/test_content_tools.py b/tests/test_content_tools.py index 92a2720..ba1bd04 100644 --- a/tests/test_content_tools.py +++ b/tests/test_content_tools.py @@ -1,7 +1,6 @@ from typing import Union import pytest - from chatlas import ChatOpenAI from chatlas.types import ContentToolResult @@ -16,8 +15,8 @@ def add(x: int, y: int) -> int: chat.register_tool(add) - assert len(chat.tools) == 1 - tool = chat.tools["add"] + assert len(chat._tools) == 1 + tool = chat._tools["add"] assert tool.name == "add" assert tool.func == add assert tool.schema["function"]["name"] == "add" @@ -48,8 +47,8 @@ def foo( chat.register_tool(foo) - assert len(chat.tools) == 1 - tool = chat.tools["foo"] + assert len(chat._tools) == 1 + tool = chat._tools["foo"] assert tool.name == "foo" assert tool.func == foo assert tool.schema["function"]["name"] == "foo"