Class CodeChatModel (1.35.0)

  CodeChatModel 
 ( 
 model_id 
 : 
 str 
 , 
 endpoint_name 
 : 
 typing 
 . 
 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( context="I'm writing a large-scale enterprise application.", max_output_tokens=128, temperature=0.2, )

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

Methods

CodeChatModel

  CodeChatModel 
 ( 
 model_id 
 : 
 str 
 , 
 endpoint_name 
 : 
 typing 
 . 
 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 
 ) 
 - 
> vertexai 
 . 
 _model_garden 
 . 
 _model_garden_models 
 . 
 T

Loads a _ModelGardenModel.

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

start_chat

  start_chat 
 ( 
 * 
 , 
 context 
 : 
 typing 
 . 
 Optional 
 [ 
 str 
 ] 
 = 
 None 
 , 
 max_output_tokens 
 : 
 typing 
 . 
 Optional 
 [ 
 int 
 ] 
 = 
 None 
 , 
 temperature 
 : 
 typing 
 . 
 Optional 
 [ 
 float 
 ] 
 = 
 None 
 , 
 message_history 
 : 
 typing 
 . 
 Optional 
 [ 
 typing 
 . 
 List 
 [ 
 vertexai 
 . 
 language_models 
 . 
 ChatMessage 
 ] 
 ] 
 = 
 None 
 , 
 stop_sequences 
 : 
 typing 
 . 
 Optional 
 [ 
 typing 
 . 
 List 
 [ 
 str 
 ]] 
 = 
 None 
 ) 
 - 
> vertexai 
 . 
 language_models 
 . 
 CodeChatSession

Starts a chat session with the code chat model.