From aa46a304720773b049114041e5e7c5ed5487ba6d Mon Sep 17 00:00:00 2001 From: Daniel Lee Date: Wed, 25 Dec 2024 14:47:05 -0800 Subject: [PATCH] patch - remove lazy import from client --- openpo/client.py | 5 +---- pyproject.toml | 2 +- 2 files changed, 2 insertions(+), 5 deletions(-) diff --git a/openpo/client.py b/openpo/client.py index 21f0baa..4ca683b 100644 --- a/openpo/client.py +++ b/openpo/client.py @@ -8,6 +8,7 @@ from .resources.batch.batch import Batch from .resources.completion.completion import Completion from .resources.eval.eval import Evaluation +from .resources.provider import Anthropic, HuggingFace, OpenAI, OpenRouter class OpenPO: @@ -51,28 +52,24 @@ def _get_provider_instance(self, provider: str): if provider == "huggingface": if not self.hf_api_key: raise AuthenticationError("HuggingFace") - from .resources.provider import HuggingFace return HuggingFace(api_key=self.hf_api_key) if provider == "openrouter": if not self.openrouter_api_key: raise AuthenticationError("OpenRouter") - from .resources.provider import OpenRouter return OpenRouter(api_key=self.openrouter_api_key) if provider == "openai": if not self.openai_api_key: raise AuthenticationError("OpenAI") - from .resources.provider import OpenAI return OpenAI(api_key=self.openai_api_key) if provider == "anthropic": if not self.anthropic_api_key: raise AuthenticationError("Anthropic") - from .resources.provider import Anthropic return Anthropic(api_key=self.anthropic_api_key) diff --git a/pyproject.toml b/pyproject.toml index 4d8bdd4..eb274af 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "openpo" -version = "0.7.3" +version = "0.7.4" description = "Build high quality synthetic datasets with AI feedback from 200+ LLMs" authors = ["Daniel Lee "] license = "Apache-2.0"