fanqiNO1 commited on
Commit
1e5d6ca
·
1 Parent(s): 0082e2e

[Feature] Support SiliconFlow API (#169)

Browse files
Files changed (1) hide show
  1. mindsearch/agent/models.py +17 -0
mindsearch/agent/models.py CHANGED
@@ -58,3 +58,20 @@ qwen = dict(type=GPTAPI,
58
  max_new_tokens=4096,
59
  repetition_penalty=1.02,
60
  stop_words=['<|im_end|>'])
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
58
  max_new_tokens=4096,
59
  repetition_penalty=1.02,
60
  stop_words=['<|im_end|>'])
61
+
62
+ internlm_silicon = dict(type=GPTAPI,
63
+ model_type='internlm/internlm2_5-7b-chat',
64
+ key=os.environ.get('SILICON_API_KEY', 'YOUR SILICON API KEY'),
65
+ openai_api_base='https://api.siliconflow.cn/v1/chat/completions',
66
+ meta_template=[
67
+ dict(role='system', api_role='system'),
68
+ dict(role='user', api_role='user'),
69
+ dict(role='assistant', api_role='assistant'),
70
+ dict(role='environment', api_role='system')
71
+ ],
72
+ top_p=0.8,
73
+ top_k=1,
74
+ temperature=0,
75
+ max_new_tokens=8192,
76
+ repetition_penalty=1.02,
77
+ stop_words=['<|im_end|>'])