更新LlamaIndex版本库

2024-09-10 14:07:52 +08:00
parent 6f5548ee61
commit a165d55822
4 changed files with 80 additions and 7 deletions
@@ -5,7 +5,7 @@ from llama_index.core.callbacks.base import CallbackManager
 from llama_index.core.indices.prompt_helper import PromptHelper
 from llama_index.core.prompts import BasePromptTemplate
 from llama_index.core.service_context import ServiceContext
-from llama_index.core.service_context_elements.llm_predictor import LLMPredictorType
+from llama_index.core.llms import LLM
 from llama_index.core.types import BaseModel,RESPONSE_TEXT_TYPE
 from llama_index.core.async_utils import run_async_tasks
 from llama_index.core.utils import get_tokenizer
@@ -14,7 +14,7 @@ from llama_index.core.prompts.prompt_utils import get_empty_prompt_txt
 class CustomTreeResponse(TreeSummarize):
    def __init__(
        self,
-        llm: Optional[LLMPredictorType] = None,
+        llm: Optional[LLM] = None,
        callback_manager: Optional[CallbackManager] = None,
        prompt_helper: Optional[PromptHelper] = None,
        summary_template: Optional[BasePromptTemplate] = None,