File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 1414model_to_use = 3 # 1 for OpenAI, 2 for DeepSeek
1515
1616if model_to_use == 1 :
17- max_tokens = 250
17+ max_tokens = 512
1818 temperature = 0.7
1919 model = "gpt-4"
2020 client = openai .OpenAI (api_key = OPENAI_API_KEY , base_url = "https://api.openai.com/v1" )
2121 logger .info (f"Using OpenAI { model } model." )
2222elif model_to_use == 2 :
23- max_tokens = 250
23+ max_tokens = 512
2424 temperature = 0.9
2525 model = "deepseek-chat"
2626 # for DeepSeek backward compatibility, you can still use `https://api.deepseek.com/v1` as `base_url`.
2727 client = openai .OpenAI (api_key = DEEPSEEK_API_KEY , base_url = "https://api.deepseek.com" )
2828 logger .info (f"Using DeepSeek { model } model." )
2929elif model_to_use == 3 :
30- max_tokens = 250
30+ max_tokens = 512
3131 temperature = 0.7
3232 model = "grok-3-beta"
3333 client = openai .OpenAI (api_key = GROK_API_KEY , base_url = "https://api.x.ai/v1" )
You can’t perform that action at this time.
0 commit comments