Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Fuse stream and direct run calls #296

Merged
merged 9 commits into from
Jan 22, 2025
Merged

Fuse stream and direct run calls #296

Changes from 1 commit
Commits
Show all changes
9 commits
Select commit Hold shift + click to select a range
08bb8f5
fuse stream and non stream calls
clefourrier Jan 21, 2025
e2fe8a4
style
clefourrier Jan 21, 2025
a982c7e
update type hinting and doc
clefourrier Jan 21, 2025
f844792
fix type hinting
clefourrier Jan 21, 2025
b3edee6
fix style
clefourrier Jan 21, 2025
ff258fc
Merge branch 'main' into fix
clefourrier Jan 21, 2025
43e723a
correct type hinting since fn is now a generator
clefourrier Jan 21, 2025
612c5fb
make style"
clefourrier Jan 21, 2025
d290063
Fix type
aymeric-roucher Jan 22, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 1 addition & 2 deletions src/smolagents/types.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,6 @@
import tempfile
import uuid
from io import BytesIO
from typing import Union

import numpy as np
import requests
Expand Down Expand Up @@ -253,7 +252,7 @@ def handle_agent_input_types(*args, **kwargs):
return args, kwargs


def handle_agent_output_types(output, output_type=None) -> Union[str, AgentType]:
def handle_agent_output_types(output, output_type=None):
if output_type in _AGENT_TYPE_MAPPING:
# If the class has defined outputs, we can map directly according to the class definition
decoded_outputs = _AGENT_TYPE_MAPPING[output_type](output)
Expand Down
Loading