44 lines
1.5 KiB
Python
44 lines
1.5 KiB
Python
|
'''
|
|||
|
|
非聊天模型类,继承自 LLM_Model_Base
|
|||
|
|
|
|||
|
|
author: DrGraph
|
|||
|
|
date: 2025-11-20
|
|||
|
|
'''
|
|||
|
|
from loguru import logger
|
|||
|
|
from langchain_openai import OpenAI
|
|||
|
|
from langchain_core.messages import AIMessage
|
|||
|
|
from DrGraph.utils.Constant import Constant
|
|||
|
|
from LLM.llm_model_base import LLM_Model_Base
|
|||
|
|
|
|||
|
|
|
|||
|
|
class NonChat_LLM(LLM_Model_Base):
|
|||
|
|
'''
|
|||
|
|
非聊天模型类，继承自 LLM_Model_Base，调用这个非聊天模型OpenAI
|
|||
|
|
- 语言模型名称， 缺省为"gpt-4o-mini"
|
|||
|
|
- 温度，缺省为0.7
|
|||
|
|
- 语言模型名称 = "非聊天模型"， 在人机界面中显示
|
|||
|
|
'''
|
|||
|
|
def __init__(self, model_name: str = "gpt-4o-mini", temperature: float = 0.7):
|
|||
|
|
super().__init__(model_name, temperature)
|
|||
|
|
self.name = '非聊天模型'
|
|||
|
|
self.mode = Constant.LLM_MODE_NONCHAT
|
|||
|
|
self.llmModel = OpenAI(
|
|||
|
|
model_name=self.model_name,
|
|||
|
|
temperature=self.temperature,
|
|||
|
|
)
|
|||
|
|
# 返回消息格式，以便在chatbot中显示
|
|||
|
|
def invoke(self, prompt: str):
|
|||
|
|
'''
|
|||
|
|
调用非聊天模型，返回消息格式，以便在chatbot中显示
|
|||
|
|
prompt: 用户输入，为字符串类型
|
|||
|
|
return: 助手回复，为字符串类型
|
|||
|
|
'''
|
|||
|
|
logger.info(f"{self.name} >>> 1.1 用户输入: {type(prompt)}")
|
|||
|
|
try:
|
|||
|
|
response = self.llmModel.invoke(prompt)
|
|||
|
|
logger.info(f"{self.name} >>> 1.2 助手回复: {type(response)}")
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(e)
|
|||
|
|
return response
|
|||
|
|
|
|||
|
|
|