refine code

This commit is contained in:
geekan 2024-01-10 16:17:48 +08:00 committed by 莘权 马
parent cd29edcc4f
commit 00a212b52b
4 changed files with 9 additions and 8 deletions

View file

@ -165,6 +165,7 @@ class ContextMixin(BaseModel):
@property
def llm(self) -> BaseLLM:
"""Role llm: role llm > context llm"""
# logger.info(f"class:{self.__class__.__name__}, llm: {self._llm}, llm_config: {self._llm_config}")
if self._llm_config and not self._llm:
self._llm = self.context.llm(self._llm_config.name, self._llm_config.provider)
return self._llm or self.context.llm()