Skip to content

Commit bfe3ae5

Browse files
committed
Update model params
1 parent 76a072a commit bfe3ae5

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

banyabot/banyabot.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -14,20 +14,20 @@
1414
model_to_use = 3 # 1 for OpenAI, 2 for DeepSeek
1515

1616
if 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.")
2222
elif 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.")
2929
elif 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")

0 commit comments

Comments
 (0)