Skip to content

Commit a6163a6

Browse files
committed
review comments
Signed-off-by: Akihiko Kuroda <akihikokuroda2020@gmail.com>
1 parent 0f9147c commit a6163a6

1 file changed

Lines changed: 5 additions & 5 deletions

File tree

mellea/backends/tools.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
import re
1313
from collections import defaultdict
1414
from collections.abc import Callable, Generator, Iterable, Mapping, Sequence
15-
from typing import Any, Generic, Literal, ParamSpec, TypeVar, overload
15+
from typing import Any, Literal, ParamSpec, TypeVar, overload
1616

1717
from pydantic import BaseModel, ConfigDict, Field
1818

@@ -77,14 +77,14 @@ def as_json_tool(self) -> dict[str, Any]:
7777
return self._as_json_tool.copy()
7878

7979
@classmethod
80-
def from_langchain(cls, tool: Any) -> "MelleaTool[P, Any]":
80+
def from_langchain(cls, tool: Any) -> "MelleaTool[..., Any]":
8181
"""Create a MelleaTool from a LangChain tool object.
8282
8383
Args:
8484
tool (Any): A ``langchain_core.tools.BaseTool`` instance to wrap.
8585
8686
Returns:
87-
MelleaTool[P, Any]: A Mellea tool wrapping the LangChain tool.
87+
MelleaTool[..., Any]: A Mellea tool wrapping the LangChain tool.
8888
8989
Raises:
9090
ImportError: If ``langchain-core`` is not installed.
@@ -124,14 +124,14 @@ def parameter_remapper(*args, **kwargs):
124124
) from e
125125

126126
@classmethod
127-
def from_smolagents(cls, tool: Any) -> "MelleaTool[P, Any]":
127+
def from_smolagents(cls, tool: Any) -> "MelleaTool[..., Any]":
128128
"""Create a Tool from a HuggingFace smolagents tool object.
129129
130130
Args:
131131
tool: A smolagents.Tool instance
132132
133133
Returns:
134-
MelleaTool[P, Any]: A Mellea tool wrapping the smolagents tool
134+
MelleaTool[..., Any]: A Mellea tool wrapping the smolagents tool
135135
136136
Raises:
137137
ImportError: If smolagents is not installed

0 commit comments

Comments
 (0)