21 lines
628 B
Python
21 lines
628 B
Python
|
#coding=utf8
|
||
|
|
|
||
|
|
import os
|
||
|
|
from langchain.utilities import BingSearchAPIWrapper
|
||
|
|
|
||
|
|
|
||
|
|
env_bing_key = os.environ.get("BING_SUBSCRIPTION_KEY")
|
||
|
|
env_bing_url = os.environ.get("BING_SEARCH_URL")
|
||
|
|
|
||
|
|
|
||
|
|
def search(text, result_len=3):
|
||
|
|
if not (env_bing_key and env_bing_url):
|
||
|
|
return [{"snippet":"please set BING_SUBSCRIPTION_KEY and BING_SEARCH_URL in os ENV",
|
||
|
|
"title": "env inof not fould", "link":"https://python.langchain.com/en/latest/modules/agents/tools/examples/bing_search.html"}]
|
||
|
|
search = BingSearchAPIWrapper()
|
||
|
|
return search.results(text, result_len)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
r = search('python')
|