Johnny Lee commited on
Commit
fd1ab9f
·
1 Parent(s): 8aec4d4

bump models

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -294,7 +294,7 @@ class ChatSession(BaseModel):
294
  ) -> Tuple[Union[ChatOpenAI, ChatAnthropic], int, tiktoken.tokenizer]:
295
  if use_claude:
296
  llm = ChatAnthropic(
297
- model="claude-2.1",
298
  anthropic_api_key=ANTHROPIC_API_KEY,
299
  temperature=1,
300
  max_tokens_to_sample=2048,
@@ -305,7 +305,7 @@ class ChatSession(BaseModel):
305
  return llm, context_length, tokenizer
306
  else:
307
  llm = ChatOpenAI(
308
- model_name="gpt-4",
309
  temperature=1,
310
  openai_api_key=OPENAI_API_KEY,
311
  max_retries=6,
 
294
  ) -> Tuple[Union[ChatOpenAI, ChatAnthropic], int, tiktoken.tokenizer]:
295
  if use_claude:
296
  llm = ChatAnthropic(
297
+ model="claude-3-5-sonnet-20240620",
298
  anthropic_api_key=ANTHROPIC_API_KEY,
299
  temperature=1,
300
  max_tokens_to_sample=2048,
 
305
  return llm, context_length, tokenizer
306
  else:
307
  llm = ChatOpenAI(
308
+ model_name="gpt-4o",
309
  temperature=1,
310
  openai_api_key=OPENAI_API_KEY,
311
  max_retries=6,