Fimeg/letta-server
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

fix: remove tools tests (#1236)

Browse Source
This commit is contained in:
Sarah Wooders
2024-04-10 18:34:38 -07:00
committed by GitHub
parent bca1f9ba8c
commit edcc143365
1 changed files with 31 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

62
tests/test_client.py
View File

@@ -225,13 +225,13 @@ def test_humans_personas(client, agent):
assert human.text == "Human text", "Creating human failed"
def test_tools(client, agent):
tools_response = client.list_tools()
print("TOOLS", tools_response)
tool_name = "TestTool"
tool_response = client.create_tool(name=tool_name, source_code="print('Hello World')", source_type="python")
assert tool_response, "Creating tool failed"
# def test_tools(client, agent):
# tools_response = client.list_tools()
# print("TOOLS", tools_response)
#
# tool_name = "TestTool"
# tool_response = client.create_tool(name=tool_name, source_code="print('Hello World')", source_type="python")
# assert tool_response, "Creating tool failed"
def test_config(client, agent):
@@ -350,27 +350,27 @@ def test_presets(client, agent):
assert new_preset.id not in [p.id for p in all_presets], (new_preset, all_presets)
def test_tools(client, agent):
# load a function
file_path = "tests/data/functions/dump_json.py"
module_name = "dump_json"
# list functions
response = client.list_tools()
orig_tools = response.tools
print(orig_tools)
# add the tool
create_tool_response = client.create_tool(name=module_name, file_path=file_path)
print(create_tool_response)
# list functions
response = client.list_tools()
new_tools = response.tools
assert module_name in [tool.name for tool in new_tools]
# assert len(new_tools) == len(orig_tools) + 1
# TODO: add a function to a preset
# TODO: add a function to an agent
# def test_tools(client, agent):
#
# # load a function
# file_path = "tests/data/functions/dump_json.py"
# module_name = "dump_json"
#
# # list functions
# response = client.list_tools()
# orig_tools = response.tools
# print(orig_tools)
#
# # add the tool
# create_tool_response = client.create_tool(name=module_name, file_path=file_path)
# print(create_tool_response)
#
# # list functions
# response = client.list_tools()
# new_tools = response.tools
# assert module_name in [tool.name for tool in new_tools]
# # assert len(new_tools) == len(orig_tools) + 1
#
# # TODO: add a function to a preset
#
# # TODO: add a function to an agent