- Move to Mistral iso OpenAI as primary choice
This commit is contained in:
@@ -8,7 +8,7 @@ import re
|
||||
from langchain_core.runnables import RunnablePassthrough
|
||||
|
||||
from common.extensions import minio_client
|
||||
from common.utils.model_utils import create_language_template
|
||||
from common.utils.model_utils import create_language_template, get_embedding_llm
|
||||
from .base_processor import BaseProcessor
|
||||
from common.utils.business_event_context import current_event
|
||||
from .processor_registry import ProcessorRegistry
|
||||
@@ -210,7 +210,7 @@ class PDFProcessor(BaseProcessor):
|
||||
return text_splitter.split_text(content)
|
||||
|
||||
def _process_chunks_with_llm(self, chunks):
|
||||
llm = self.model_variables.get_llm()
|
||||
llm = get_embedding_llm()
|
||||
template = self.model_variables.get_template('pdf_parse')
|
||||
pdf_prompt = ChatPromptTemplate.from_template(template)
|
||||
setup = RunnablePassthrough()
|
||||
|
||||
Reference in New Issue
Block a user