github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-26 22:05:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

x

Browse Source
This commit is contained in:
Eugene Yurtsev
2024-09-13 18:02:48 -04:00
parent d46ab19954
commit 3f6057df0d
1 changed files with 56 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

84
libs/partners/robocorp/tests/unit_tests/test_toolkits.py
View File

@@ -12,6 +12,10 @@ from langchain_core.utils.function_calling import (
from langchain_robocorp.toolkits import ActionServerToolkit
from ._fixtures import FakeChatLLMT
from pydantic import __version__
# Get the version tuple (it may contain a suffix like 'rc1')
VERSION_TUPLE = tuple(int(x) if x.isdigit() else x for x in __version__.split("."))
def test_initialization() -> None:
@@ -91,36 +95,60 @@ Strictly adhere to the schema."""
assert set(params["properties"].keys()) == {"sheet", "rows_to_add"}
desc = "The columns that make up the row"
expected = {
"description": "the rows to be added to the sheet",
"allOf": [
{
"title": "Rows To Add",
"type": "object",
"properties": {
"rows": {
"title": "Rows",
"description": "The rows that need to be added",
"type": "array",
"items": {
"title": "Row",
"type": "object",
"properties": {
"columns": {
"title": "Columns",
"description": desc,
"type": "array",
"items": {"type": "string"},
}
if VERSION_TUPLE < (2, 9):
expected = {
"description": "the rows to be added to the sheet",
"allOf": [
{
"title": "Rows To Add",
"type": "object",
"properties": {
"rows": {
"title": "Rows",
"description": "The rows that need to be added",
"type": "array",
"items": {
"title": "Row",
"type": "object",
"properties": {
"columns": {
"title": "Columns",
"description": desc,
"type": "array",
"items": {"type": "string"},
}
},
"required": ["columns"],
},
"required": ["columns"],
}
},
"required": ["rows"],
}
],
}
else:
expected = {
"properties": {
"rows": {
"description": "The rows that need to be added",
"items": {
"properties": {
"columns": {
"description": "The columns that make up the row",
"items": {"type": "string"},
"type": "array",
}
},
}
},
"required": ["rows"],
}
],
}
"required": ["columns"],
"type": "object",
},
"type": "array",
}
},
"required": ["rows"],
"type": "object",
}
assert params["properties"]["rows_to_add"] == expected