Skip to content

Commit 47ab985

Browse files
committed
Fixed async browse and added type hints
1 parent 53fe821 commit 47ab985

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

src/multion/client.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -9,9 +9,9 @@
99
import os
1010

1111
from .wrappers import wraps_function
12+
from .types.browse_output import BrowseOutput
1213
from .types.retrieve_output import RetrieveOutput
1314

14-
1515
# this is used as the default value for optional parameters
1616
OMIT = typing.cast(typing.Any, ...)
1717

@@ -63,7 +63,7 @@ def __init__(
6363

6464
@agentops.record_function(event_name="browse") # type: ignore
6565
@wraps_function(BaseMultiOn.browse)
66-
def browse(self, *args, **kwargs):
66+
def browse(self, *args, **kwargs) -> BrowseOutput:
6767
agentops.start_session(tags=["multion-sdk"])
6868
return super().browse(*args, **kwargs)
6969

@@ -121,9 +121,9 @@ def __init__(
121121

122122
@agentops.record_function(event_name="browse") # type: ignore
123123
@wraps_function(AsyncBaseMultiOn.browse)
124-
async def browse(self, *args, **kwargs):
124+
async def browse(self, *args, **kwargs) -> BrowseOutput:
125125
agentops.start_session(tags=["multion-sdk"])
126-
return super().browse(*args, **kwargs)
126+
return await super().browse(*args, **kwargs)
127127

128128
@agentops.record_function(event_name="retrieve") # type: ignore
129129
@wraps_function(BaseMultiOn.retrieve)

0 commit comments

Comments
 (0)