修复embedding失误导入成llm的问题

This commit is contained in:
YILING0013
2025-02-16 23:10:32 +08:00
parent ba10e6dd66
commit f5691f268f
4 changed files with 143 additions and 26 deletions
+6 -8
View File
@@ -6,6 +6,7 @@
import os
import logging
from llm_adapters import create_llm_adapter
from embedding_adapters import create_embedding_adapter
from prompt_definitions import summary_prompt, update_character_state_prompt
from novel_generator.common import invoke_with_cleaning
from utils import read_file, clear_file_content, save_string_to_txt
@@ -75,14 +76,11 @@ def finalize_chapter(
save_string_to_txt(new_char_state, character_state_file)
update_vector_store(
embedding_adapter=create_llm_adapter(
interface_format=embedding_interface_format,
base_url=embedding_url,
model_name=embedding_model_name,
api_key=embedding_api_key,
temperature=temperature,
max_tokens=max_tokens,
timeout=timeout
embedding_adapter=create_embedding_adapter(
embedding_interface_format,
embedding_api_key,
embedding_url,
embedding_model_name
),
new_chapter=chapter_text,
filepath=filepath