Langchain-Chatchat/agent/bing_search.py
halfss dfcd6f6102
添加bing搜索agent (#378)
* 1: 基于langchain的bing search agent添加bing搜索支持(百度不靠谱,Google不可用)
2: 调整输入框交互模式,对话/知识库/搜索,三选一

* fixed bug of message have no text

---------

Co-authored-by: root <jv.liu@i1368.com>
2023-05-16 23:51:34 +08:00

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')