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

feat: rename block.name to block.template_name for clarity and add shared block tests (#1951)

Browse Source 
Co-authored-by: Matt Zhou <mattzh1314@gmail.com>
This commit is contained in:
Sarah Wooders
2024-11-04 11:49:16 -08:00
committed by GitHub
parent f0a6cb84de
commit 39999ce48c
15 changed files with 821 additions and 674 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
letta/schemas/tool.py
View File

@@ -8,7 +8,10 @@ from letta.functions.helpers import (
generate_crewai_tool_wrapper,
generate_langchain_tool_wrapper,
)
from letta.functions.schema_generator import generate_schema_from_args_schema
from letta.functions.schema_generator import (
generate_schema_from_args_schema_v1,
generate_schema_from_args_schema_v2,
)
from letta.schemas.letta_base import LettaBase
from letta.schemas.openai.chat_completions import ToolCall
@@ -97,7 +100,7 @@ class ToolCreate(LettaBase):
source_type = "python"
tags = ["composio"]
wrapper_func_name, wrapper_function_str = generate_composio_tool_wrapper(action)
json_schema = generate_schema_from_args_schema(composio_tool.args_schema, name=wrapper_func_name, description=description)
json_schema = generate_schema_from_args_schema_v2(composio_tool.args_schema, name=wrapper_func_name, description=description)
return cls(
name=wrapper_func_name,
@@ -129,7 +132,7 @@ class ToolCreate(LettaBase):
tags = ["langchain"]
# NOTE: langchain tools may come from different packages
wrapper_func_name, wrapper_function_str = generate_langchain_tool_wrapper(langchain_tool, additional_imports_module_attr_map)
json_schema = generate_schema_from_args_schema(langchain_tool.args_schema, name=wrapper_func_name, description=description)
json_schema = generate_schema_from_args_schema_v1(langchain_tool.args_schema, name=wrapper_func_name, description=description)
return cls(
name=wrapper_func_name,
@@ -159,7 +162,7 @@ class ToolCreate(LettaBase):
source_type = "python"
tags = ["crew-ai"]
wrapper_func_name, wrapper_function_str = generate_crewai_tool_wrapper(crewai_tool, additional_imports_module_attr_map)
json_schema = generate_schema_from_args_schema(crewai_tool.args_schema, name=wrapper_func_name, description=description)
json_schema = generate_schema_from_args_schema_v1(crewai_tool.args_schema, name=wrapper_func_name, description=description)
return cls(
name=wrapper_func_name,