From 6168eabfb1b714747e65c9a10261c5797d6afbd1 Mon Sep 17 00:00:00 2001 From: Koki Ryu Date: Wed, 20 Nov 2024 03:31:27 +0000 Subject: [PATCH] Organize the import --- src/langcheck/metrics/eval_clients/_openai.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/langcheck/metrics/eval_clients/_openai.py b/src/langcheck/metrics/eval_clients/_openai.py index 7cddd424..4fa18c84 100644 --- a/src/langcheck/metrics/eval_clients/_openai.py +++ b/src/langcheck/metrics/eval_clients/_openai.py @@ -8,13 +8,14 @@ import torch from openai import AsyncAzureOpenAI, AsyncOpenAI, AzureOpenAI, OpenAI +from openai.types.create_embedding_response import CreateEmbeddingResponse from langcheck.utils.progress_bar import tqdm_wrapper from ..prompts._utils import get_template from ..scorer._base import BaseSimilarityScorer from ._base import EvalClient, TextResponseWithLogProbs -from openai.types.create_embedding_response import CreateEmbeddingResponse + class OpenAIEvalClient(EvalClient): """EvalClient defined for OpenAI API."""