Skip to content

Commit 674ccee

Browse files
authored
chore(langchain-sdk): Fix type checks (#70)
1 parent 956591d commit 674ccee

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/toolbox_langchain_sdk/client.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
import asyncio
2-
from typing import Optional
2+
from typing import Optional, Type
33

44
from aiohttp import ClientSession
55
from langchain_core.tools import StructuredTool
@@ -92,7 +92,7 @@ def _generate_tool(
9292
The generated tool.
9393
"""
9494
tool_schema = manifest.tools[tool_name]
95-
tool_model: BaseModel = _schema_to_model(
95+
tool_model: Type[BaseModel] = _schema_to_model(
9696
model_name=tool_name, schema=tool_schema.parameters
9797
)
9898

0 commit comments

Comments
 (0)