Skip to content

Commit 1bcb0df

Browse files
committed
Update tests
1 parent 3388547 commit 1bcb0df

File tree

1 file changed

+4
-5
lines changed

1 file changed

+4
-5
lines changed

tests/test_content_tools.py

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
11
from typing import Union
22

33
import pytest
4-
54
from chatlas import ChatOpenAI
65
from chatlas.types import ContentToolResult
76

@@ -16,8 +15,8 @@ def add(x: int, y: int) -> int:
1615

1716
chat.register_tool(add)
1817

19-
assert len(chat.tools) == 1
20-
tool = chat.tools["add"]
18+
assert len(chat._tools) == 1
19+
tool = chat._tools["add"]
2120
assert tool.name == "add"
2221
assert tool.func == add
2322
assert tool.schema["function"]["name"] == "add"
@@ -48,8 +47,8 @@ def foo(
4847

4948
chat.register_tool(foo)
5049

51-
assert len(chat.tools) == 1
52-
tool = chat.tools["foo"]
50+
assert len(chat._tools) == 1
51+
tool = chat._tools["foo"]
5352
assert tool.name == "foo"
5453
assert tool.func == foo
5554
assert tool.schema["function"]["name"] == "foo"

0 commit comments

Comments
 (0)