74 lines
2.1 KiB
Python
74 lines
2.1 KiB
Python
# Copyright 2023 Google LLC
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
#
|
|
"""Classes for working with language models."""
|
|
|
|
from vertexai.language_models._language_models import (
|
|
_PreviewChatModel,
|
|
_PreviewChatSession,
|
|
_PreviewCodeChatModel,
|
|
_PreviewCodeChatSession,
|
|
_PreviewCodeGenerationModel,
|
|
_PreviewTextEmbeddingModel,
|
|
_PreviewTextGenerationModel,
|
|
ChatMessage,
|
|
CountTokensResponse,
|
|
InputOutputTextPair,
|
|
TextEmbedding,
|
|
TextEmbeddingInput,
|
|
TextGenerationResponse,
|
|
TuningEvaluationSpec,
|
|
)
|
|
|
|
from vertexai.language_models._evaluatable_language_models import (
|
|
EvaluationTextGenerationSpec,
|
|
EvaluationTextSummarizationSpec,
|
|
EvaluationQuestionAnsweringSpec,
|
|
EvaluationTextClassificationSpec,
|
|
EvaluationClassificationMetric,
|
|
EvaluationMetric,
|
|
)
|
|
|
|
|
|
ChatModel = _PreviewChatModel
|
|
ChatSession = _PreviewChatSession
|
|
CodeChatModel = _PreviewCodeChatModel
|
|
CodeChatSession = _PreviewCodeChatSession
|
|
CodeGenerationModel = _PreviewCodeGenerationModel
|
|
TextGenerationModel = _PreviewTextGenerationModel
|
|
TextEmbeddingModel = _PreviewTextEmbeddingModel
|
|
|
|
__all__ = [
|
|
"ChatMessage",
|
|
"ChatModel",
|
|
"ChatSession",
|
|
"CodeChatModel",
|
|
"CodeChatSession",
|
|
"CodeGenerationModel",
|
|
"CountTokensResponse",
|
|
"EvaluationClassificationMetric",
|
|
"EvaluationMetric",
|
|
"EvaluationTextGenerationSpec",
|
|
"EvaluationTextSummarizationSpec",
|
|
"EvaluationQuestionAnsweringSpec",
|
|
"EvaluationTextClassificationSpec",
|
|
"InputOutputTextPair",
|
|
"TextEmbedding",
|
|
"TextEmbeddingInput",
|
|
"TextEmbeddingModel",
|
|
"TextGenerationModel",
|
|
"TextGenerationResponse",
|
|
"TuningEvaluationSpec",
|
|
]
|