Upload 6 files

mindsearch/agent/__init__.py

@@ -17,7 +17,7 @@ from mindsearch.agent.mindsearch_prompt import (
 LLM = {}
 
 
-def init_agent(lang='cn', model_format='internlm_server'):
+def init_agent(lang='cn', model_format='internlm_server',search_engine='DuckDuckGoSearch'):
     llm = LLM.get(model_format, None)
     if llm is None:
         llm_cfg = getattr(llm_factory, model_format)
@@ -43,7 +43,7 @@ def init_agent(lang='cn', model_format='internlm_server'):
         searcher_cfg=dict(
             llm=llm,
             plugin_executor=ActionExecutor(
-                BingBrowser(searcher_type='DuckDuckGoSearch',
+                BingBrowser(searcher_type=search_engine,
                             topk=6,
                             api_key=os.environ.get('BING_API_KEY',
                                                    'YOUR BING API'))),

mindsearch/agent/models.py

@@ -34,11 +34,11 @@ internlm_hf = dict(type=HFTransformerCasualLM,
                    max_new_tokens=8192,
                    repetition_penalty=1.02,
                    stop_words=['<|im_end|>'])
-
+# openai_api_base needs to fill in the complete chat api address, such as: https://api.openai.com/v1/chat/completions
 gpt4 = dict(type=GPTAPI,
             model_type='gpt-4-turbo',
             key=os.environ.get('OPENAI_API_KEY', 'YOUR OPENAI API KEY'),
-            openai_api_base=os.environ.get('OPENAI_API_BASE', 'https://api.openai.com/v1'),
+            openai_api_base=os.environ.get('OPENAI_API_BASE', 'https://api.openai.com/v1/chat/completions'),
             )
 
 url = 'https://dashscope.aliyuncs.com/api/v1/services/aigc/text-generation/generation'
@@ -58,3 +58,20 @@ qwen = dict(type=GPTAPI,
             max_new_tokens=4096,
             repetition_penalty=1.02,
             stop_words=['<|im_end|>'])
+
+internlm_silicon = dict(type=GPTAPI,
+                        model_type='internlm/internlm2_5-7b-chat',
+                        key=os.environ.get('SILICON_API_KEY', 'YOUR SILICON API KEY'),
+                        openai_api_base='https://api.siliconflow.cn/v1/chat/completions',
+                        meta_template=[
+                            dict(role='system', api_role='system'),
+                            dict(role='user', api_role='user'),
+                            dict(role='assistant', api_role='assistant'),
+                            dict(role='environment', api_role='system')
+                        ],
+                        top_p=0.8,
+                        top_k=1,
+                        temperature=0,
+                        max_new_tokens=8192,
+                        repetition_penalty=1.02,
+                        stop_words=['<|im_end|>'])

mindsearch/app.py

@@ -23,6 +23,10 @@ def parse_arguments():
                         default='internlm_server',
                         type=str,
                         help='Model format')
+    parser.add_argument('--search_engine',
+                       default='DuckDuckGoSearch',
+                       type=str,
+                       help='Search engine')
     return parser.parse_args()
 
 
@@ -123,7 +127,7 @@ async def run(request: GenerationParams):
             await queue.wait_closed()
 
     inputs = request.inputs
-    agent = init_agent(lang=args.lang, model_format=args.model_format)
+    agent = init_agent(lang=args.lang, model_format=args.model_format,search_engine=args.search_engine)
     return EventSourceResponse(generate())