Skip to content
Snippets Groups Projects
Unverified Commit 719c094a authored by Andrei Fajardo's avatar Andrei Fajardo Committed by GitHub
Browse files

[FIX] faulty imports when using PydanticProgramMode.LM_FORMAT_ENFORCER (#11355)

parent be76419d
Branches
Tags
No related merge requests found
......@@ -64,7 +64,7 @@ def resolve_llm(
"llm must start with str 'local' or of type LLM or BaseLanguageModel"
)
try:
from llama_index.llms.llama.utils import (
from llama_index.llms.llama_cpp.llama_utils import (
completion_to_prompt,
messages_to_prompt,
) # pants: no-infer-dep
......
......@@ -90,9 +90,15 @@ def get_program_for_llm(
**kwargs,
)
elif pydantic_program_mode == PydanticProgramMode.LM_FORMAT_ENFORCER:
from llama_index.core.program.lmformatenforcer_program import (
LMFormatEnforcerPydanticProgram,
)
try:
from llama_index.program.lmformatenforcer import (
LMFormatEnforcerPydanticProgram,
) # pants: no-infer-dep
except ImportError:
raise ImportError(
"This mode requires the `llama-index-program-lmformatenforcer package. Please"
" install it by running `pip install llama-index-program-lmformatenforcer`."
)
return LMFormatEnforcerPydanticProgram.from_defaults(
output_cls=output_cls,
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment