Class CodeChatModel (1.26.1)

CodeChatModel(model_id: str, endpoint_name: Optional[str] = None)

CodeChatModel represents a model that is capable of completing code.

.. rubric:: Examples

code_chat_model = CodeChatModel.from_pretrained("codechat-bison@001")

code_chat = code_chat_model.start_chat( max_output_tokens=128, temperature=0.2, )

code_chat.send_message("Please help write a function to calculate the min of two numbers")

Inheritance

builtins.object > vertexai._model_garden._model_garden_models._ModelGardenModel > vertexai.language_models._language_models._LanguageModel > vertexai.language_models._language_models._ChatModelBase > CodeChatModel

Methods

CodeChatModel

CodeChatModel(model_id: str, endpoint_name: Optional[str] = None)

Creates a LanguageModel.

This constructor should not be called directly. Use LanguageModel.from_pretrained(model_name=...) instead.

from_pretrained

from_pretrained(model_name: str)

Loads a _ModelGardenModel.

Exceptions
Type Description
ValueError If model_name is unknown.
ValueError If model does not support this class.

start_chat

start_chat(*, max_output_tokens: int = 128, temperature: float = 0.5)

Starts a chat session with the code chat model.