Jmaa/test-langgraph
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Configure wrapping
Some checks failed
Run Python tests (through Pytest) / Test (push) Failing after 30s
Details
Verify Python project can be installed, loaded and have version checked / Test (push) Failing after 28s
Details

Browse Source
This commit is contained in:
Jon Michael Aanes 2025-06-18 22:24:01 +02:00
parent a317d1aaef
commit f34a737d0c
1 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
test_langgraph/tools.py
View File

@ -60,10 +60,11 @@ RETURN_FORMATS = {
RETURN_FORMAT = 'json' RETURN_FORMAT = 'json'
MAX_TOOL_RESULT_LEN = 1000
APPEND_RESULT_TYPE_DOCS = True APPEND_RESULT_TYPE_DOCS = True
def wrap_method(class_, method): def wrap_method(class_, method, max_tool_result_len = 1_000_000_000,
provide_instructions: bool = False):
logger.info('Wrapping %s.%s', class_.__name__, method.__name__) logger.info('Wrapping %s.%s', class_.__name__, method.__name__)
return_type = method.__annotations__.get('return', '') return_type = method.__annotations__.get('return', '')
is_iterator = str(return_type).startswith( is_iterator = str(return_type).startswith(
@ -91,10 +92,13 @@ def wrap_method(class_, method):
repr(input_value), repr(input_value),
) )
raise raise
if len(result_str) > MAX_TOOL_RESULT_LEN: if len(result_str) > max_tool_result_len:
result_str = result_str[:MAX_TOOL_RESULT_LEN] + ' (remaining tool result elicited...)' result_str = result_str[:max_tool_result_len]
if provide_instructions:
result_str += '\nremaining tool result elicited...'
if provide_instructions:
if APPEND_RESULT_TYPE_DOCS and (return_docs := getattr(return_type, '__doc__', None)): if APPEND_RESULT_TYPE_DOCS and (return_docs := getattr(return_type, '__doc__', None)):
result_str = result_str+'\n'+return_docs result_str += '\n'+return_docs
return result_str return result_str
wrapper.__name__ = f'{class_.__name__}.{method.__name__}' wrapper.__name__ = f'{class_.__name__}.{method.__name__}'