feat(llm, prompt):增加日志输出并扩展实体类型
- 在 llm.py 中添加了日志输出,用于调试和记录 LLM 查询输入 - 在 prompt.py 中增加了 "category" 实体类型,扩展了实体提取的范围
This commit is contained in:
@@ -29,7 +29,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
|
||||
from .utils import (
|
||||
wrap_embedding_func_with_attrs,
|
||||
locate_json_string_body_from_string,
|
||||
safe_unicode_decode,
|
||||
safe_unicode_decode, logger,
|
||||
)
|
||||
|
||||
import sys
|
||||
@@ -69,6 +69,11 @@ async def openai_complete_if_cache(
|
||||
messages.extend(history_messages)
|
||||
messages.append({"role": "user", "content": prompt})
|
||||
|
||||
# 添加日志输出
|
||||
logger.debug("===== Query Input to LLM =====")
|
||||
logger.debug(f"Query: {prompt}")
|
||||
logger.debug(f"System prompt: {system_prompt}")
|
||||
logger.debug("Full context:")
|
||||
if "response_format" in kwargs:
|
||||
response = await openai_async_client.beta.chat.completions.parse(
|
||||
model=model, messages=messages, **kwargs
|
||||
|
Reference in New Issue
Block a user