Skip to content

Commit fe69b13

Browse files
committed
add web search warning
1 parent bde0456 commit fe69b13

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

src/cleanlab_tlm/utils/chat.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -755,6 +755,14 @@ def _responses_messages_to_string(messages: list[dict[str, Any]]) -> str:
755755
)
756756

757757
if message["action"]["type"] == "search":
758+
if message["action"]["sources"] is None:
759+
warnings.warn(
760+
"Web search call returned no results. Please include include=['web_search_call.action.sources'] in your request.",
761+
UserWarning,
762+
stacklevel=2,
763+
)
764+
continue
765+
758766
urls = list({source["url"] for source in message["action"]["sources"] if source["type"] == "url"})
759767

760768
with ThreadPoolExecutor() as executor:

0 commit comments

Comments
 (0)