diff --git a/src/writer/ai.py b/src/writer/ai.py index 6ce94e142..ce54ec368 100644 --- a/src/writer/ai.py +++ b/src/writer/ai.py @@ -155,7 +155,7 @@ def use_chat_model(cls) -> str: :returns: Name for the chat model. """ - return "palmyra-x-002-32k" + return "palmyra-x-004" @classmethod def use_completion_model(cls) -> str: @@ -164,7 +164,7 @@ def use_completion_model(cls) -> str: :returns: Name for the completion model. """ - return "palmyra-x-002-instruct" + return "palmyra-x-003-instruct" @classmethod def acquire_client(cls) -> Writer: @@ -1040,4 +1040,4 @@ def init(token: Optional[str] = None): """ return WriterAIManager(token=token) -apps = Apps() \ No newline at end of file +apps = Apps()