gabykim commited on
Commit
093a34b
·
1 Parent(s): f6ba372

hide huggingface import inside

Browse files
src/know_lang_bot/utils/model_provider.py CHANGED
@@ -1,6 +1,5 @@
1
  from pydantic_ai.models import Model, KnownModelName
2
  from know_lang_bot.core.types import ModelProvider
3
- from know_lang_bot.models.huggingface import HuggingFaceModel
4
  from typing import get_args
5
 
6
  def create_pydantic_model(
@@ -12,6 +11,7 @@ def create_pydantic_model(
12
  if model_str in get_args(KnownModelName):
13
  return model_str
14
  elif model_provider == ModelProvider.HUGGINGFACE:
 
15
  return HuggingFaceModel(model_name=model_name)
16
  elif model_provider == ModelProvider.TESTING:
17
  # should be used for testing purposes only
 
1
  from pydantic_ai.models import Model, KnownModelName
2
  from know_lang_bot.core.types import ModelProvider
 
3
  from typing import get_args
4
 
5
  def create_pydantic_model(
 
11
  if model_str in get_args(KnownModelName):
12
  return model_str
13
  elif model_provider == ModelProvider.HUGGINGFACE:
14
+ from know_lang_bot.models.huggingface import HuggingFaceModel
15
  return HuggingFaceModel(model_name=model_name)
16
  elif model_provider == ModelProvider.TESTING:
17
  # should be used for testing purposes only