Nelly-43 commited on
Commit
35a8be0
·
verified ·
1 Parent(s): b1b0945

Update src/llm/__init__.py

Browse files
Files changed (1) hide show
  1. src/llm/__init__.py +2 -2
src/llm/__init__.py CHANGED
@@ -2,14 +2,14 @@
2
  from src.llm import openai as _openai
3
  from src.llm import anthropic as _anthropic
4
  from src.llm import gemini as _gemini
5
- from src.llm import meta as _llama
6
  from src.config_loader import get_api_key
7
 
8
  PROVIDERS = {
9
  "openai": _openai,
10
  "anthropic": _anthropic,
11
  "gemini": _gemini,
12
- "meta-llama": _llama,
13
  }
14
 
15
  def generate(system_prompt: str, user_message: str, cfg: dict,
 
2
  from src.llm import openai as _openai
3
  from src.llm import anthropic as _anthropic
4
  from src.llm import gemini as _gemini
5
+ from src.llm import hugging_face as hf
6
  from src.config_loader import get_api_key
7
 
8
  PROVIDERS = {
9
  "openai": _openai,
10
  "anthropic": _anthropic,
11
  "gemini": _gemini,
12
+ "hugging-face": hf,
13
  }
14
 
15
  def generate(system_prompt: str, user_message: str, cfg: dict,