Skip to content

Commit

Permalink
Set up default using gpt-4o
Browse files Browse the repository at this point in the history
Signed-off-by: Toomore Chiang <[email protected]>
  • Loading branch information
toomore committed Jun 17, 2024
1 parent b63b742 commit fe153d6
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions toldwords/openai.py
Original file line number Diff line number Diff line change
@@ -1,8 +1,9 @@
''' OpenAI API Connect '''
from enum import Enum
from typing import List
from requests import Session

from pydantic import BaseModel
from requests import Session


class Role(str, Enum):
Expand Down Expand Up @@ -59,7 +60,7 @@ def __init__(self, token: str, organization: str | None) -> None:

def chat_completions(self, # pylint: disable=too-many-arguments
messages: List[Message],
model: str = 'gpt-3.5-turbo',
model: str = 'gpt-4o',
temperature: int = 1,
n: int = 1, # pylint: disable=invalid-name
user: str = 'api') -> RespCompletions:
Expand Down

0 comments on commit fe153d6

Please sign in to comment.